def show_single_coverage_mask(qreq_, cm, weight_mask_m, weight_mask, daids, fnum=None):
import plottool as pt
from ibeis import viz
fnum = pt.ensure_fnum(fnum)
idx_list = ut.dict_take(cm.daid2_idx, daids)
nPlots = len(idx_list) + 1
nRows, nCols = pt.get_square_row_cols(nPlots)
pnum_ = pt.make_pnum_nextgen(nRows, nCols)
pt.figure(fnum=fnum, pnum=(1, 2, 1))
# Draw coverage masks with bbox
# <FlipHack>
#weight_mask_m = np.fliplr(np.flipud(weight_mask_m))
#weight_mask = np.fliplr(np.flipud(weight_mask))
# </FlipHack>
stacked_weights, offset_tup, sf_tup = vt.stack_images(weight_mask_m, weight_mask, return_sf=True)
(woff, hoff) = offset_tup[1]
wh1 = weight_mask_m.shape[0:2][::-1]
wh2 = weight_mask.shape[0:2][::-1]
pt.imshow(255 * (stacked_weights), fnum=fnum, pnum=pnum_(0), title='(query image) What did match vs what should match')
pt.draw_bbox(( 0, 0) + wh1, bbox_color=(0, 0, 1))
pt.draw_bbox((woff, hoff) + wh2, bbox_color=(0, 0, 1))
# Get contributing matches
qaid = cm.qaid
daid_list = daids
fm_list = ut.take(cm.fm_list, idx_list)
fs_list = ut.take(cm.fs_list, idx_list)
# Draw matches
for px, (daid, fm, fs) in enumerate(zip(daid_list, fm_list, fs_list), start=1):
viz.viz_matches.show_matches2(qreq_.ibs, qaid, daid, fm, fs,
draw_pts=False, draw_lines=True,
draw_ell=False, fnum=fnum, pnum=pnum_(px),
darken=.5)
coverage_score = score_matching_mask(weight_mask_m, weight_mask)
pt.set_figtitle('score=%.4f' % (coverage_score,))
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_single_coverage_mask(qreq_,
cm,
weight_mask_m,
weight_mask,
daids,
fnum=None):
import plottool as pt
from ibeis import viz
fnum = pt.ensure_fnum(fnum)
idx_list = ut.dict_take(cm.daid2_idx, daids)
nPlots = len(idx_list) + 1
nRows, nCols = pt.get_square_row_cols(nPlots)
pnum_ = pt.make_pnum_nextgen(nRows, nCols)
pt.figure(fnum=fnum, pnum=(1, 2, 1))
# Draw coverage masks with bbox
# <FlipHack>
#weight_mask_m = np.fliplr(np.flipud(weight_mask_m))
#weight_mask = np.fliplr(np.flipud(weight_mask))
# </FlipHack>
stacked_weights, offset_tup, sf_tup = vt.stack_images(weight_mask_m,
weight_mask,
return_sf=True)
(woff, hoff) = offset_tup[1]
wh1 = weight_mask_m.shape[0:2][::-1]
wh2 = weight_mask.shape[0:2][::-1]
pt.imshow(255 * (stacked_weights),
fnum=fnum,
pnum=pnum_(0),
title='(query image) What did match vs what should match')
pt.draw_bbox((0, 0) + wh1, bbox_color=(0, 0, 1))
pt.draw_bbox((woff, hoff) + wh2, bbox_color=(0, 0, 1))
# Get contributing matches
qaid = cm.qaid
daid_list = daids
fm_list = ut.take(cm.fm_list, idx_list)
fs_list = ut.take(cm.fs_list, idx_list)
# Draw matches
for px, (daid, fm, fs) in enumerate(zip(daid_list, fm_list, fs_list),
start=1):
viz.viz_matches.show_matches2(qreq_.ibs,
qaid,
daid,
fm,
fs,
draw_pts=False,
draw_lines=True,
draw_ell=False,
fnum=fnum,
pnum=pnum_(px),
darken=.5)
coverage_score = score_matching_mask(weight_mask_m, weight_mask)
pt.set_figtitle('score=%.4f' % (coverage_score, ))
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))
