def show_ori_image(gori,
weights,
patch,
gradx=None,
grady=None,
gauss=None,
fnum=None):
"""
python -m pyhesaff._pyhesaff --test-test_rot_invar --show --nocpp
"""
import plottool as pt
if fnum is None:
fnum = pt.next_fnum()
print('gori.max = %r' % gori.max())
assert gori.max() <= TAU
assert gori.min() >= 0
bgr_ori = pt.color_orimag(gori, weights, False, encoding='bgr')
print('bgr_ori.max = %r' % bgr_ori.max())
#ut.embed()
bgr_ori = (255 * bgr_ori).astype(np.uint8)
print('bgr_ori.max = %r' % bgr_ori.max())
#bgr_ori = np.array(bgr_ori, dtype=np.uint8)
legend = pt.make_ori_legend_img()
#gorimag_, woff, hoff = vt.stack_images(bgr_ori, legend, vert=False, modifysize=True)
import vtool as vt
gorimag_, offsets, sftup = vt.stack_images(bgr_ori,
legend,
vert=False,
modifysize=True,
return_offset=True,
return_sf=True)
(woff, hoff) = offsets[1]
if patch is None:
pt.imshow(gorimag_, fnum=fnum)
else:
pt.imshow(gorimag_,
fnum=fnum,
pnum=(3, 1, 1),
title='colored by orientation')
#pt.imshow(patch, fnum=fnum, pnum=(2, 2, 1))
#gradx, grady = np.cos(gori + TAU / 4.0), np.sin(gori + TAU / 4.0)
if gradx is not None and grady is not None:
if weights is not None:
gradx *= weights
grady *= weights
pt.imshow(np.array(gradx * 255, dtype=np.uint8),
fnum=fnum,
pnum=(3, 3, 4))
pt.imshow(np.array(grady * 255, dtype=np.uint8),
fnum=fnum,
pnum=(3, 3, 5))
#pt.imshow(bgr_ori, pnum=(2, 2, 4))
pt.draw_vector_field(gradx, grady, pnum=(3, 3, 6), invert=True)
pt.imshow(patch, fnum=fnum, pnum=(3, 1, 3))
def show_ori_image(gori, weights, patch, gradx=None, grady=None, gauss=None, fnum=None):
"""
python -m pyhesaff._pyhesaff --test-test_rot_invar --show --nocpp
"""
import plottool as pt
if fnum is None:
fnum = pt.next_fnum()
print('gori.max = %r' % gori.max())
assert gori.max() <= TAU
assert gori.min() >= 0
bgr_ori = pt.color_orimag(gori, weights, False, encoding='bgr')
print('bgr_ori.max = %r' % bgr_ori.max())
#ut.embed()
bgr_ori = (255 * bgr_ori).astype(np.uint8)
print('bgr_ori.max = %r' % bgr_ori.max())
#bgr_ori = np.array(bgr_ori, dtype=np.uint8)
legend = pt.make_ori_legend_img()
#gorimag_, woff, hoff = vt.stack_images(bgr_ori, legend, vert=False, modifysize=True)
import vtool as vt
gorimag_, offsets, sftup = vt.stack_images(bgr_ori, legend, vert=False,
modifysize=True,
return_offset=True,
return_sf=True)
(woff, hoff) = offsets[1]
if patch is None:
pt.imshow(gorimag_, fnum=fnum)
else:
pt.imshow(gorimag_, fnum=fnum, pnum=(3, 1, 1), title='colored by orientation')
#pt.imshow(patch, fnum=fnum, pnum=(2, 2, 1))
#gradx, grady = np.cos(gori + TAU / 4.0), np.sin(gori + TAU / 4.0)
if gradx is not None and grady is not None:
if weights is not None:
gradx *= weights
grady *= weights
pt.imshow(np.array(gradx * 255, dtype=np.uint8), fnum=fnum, pnum=(3, 3, 4))
pt.imshow(np.array(grady * 255, dtype=np.uint8), fnum=fnum, pnum=(3, 3, 5))
#pt.imshow(bgr_ori, pnum=(2, 2, 4))
pt.draw_vector_field(gradx, grady, pnum=(3, 3, 6), invert=True)
pt.imshow(patch, fnum=fnum, pnum=(3, 1, 3))
