def _image_view(sel_aids=sel_aids, **_kwargs):
try:
viz.show_image(ibs, gid, sel_aids=sel_aids, fnum=self.fnum, **_kwargs)
df2.set_figtitle('Image View')
except TypeError as ex:
ut.printex(ex, ut.dict_str(_kwargs))
raise
def TEST_IBS(ibs):
gpath_list = grabdata.get_test_gpaths(ndata=2)
print('[TEST] 1.ibs.add_images(gpath_list=%r)' % gpath_list)
gid_list = ibs.add_images(gpath_list)
print(' * gid_list=%r' % gid_list)
print('[TEST] 2. ibs.get_image_paths(gid_list)')
gpaths_list = ibs.get_image_paths(gid_list)
print(' * gpaths_list=%r' % gpaths_list)
print('[TEST] 3. get_image_props')
uri_list = ibs.get_image_uris(gid_list)
path_list = ibs.get_image_paths(gid_list)
gsize_list = ibs.get_image_sizes(gid_list)
time_list = ibs.get_image_unixtime(gid_list)
gps_list = ibs.get_image_gps(gid_list)
print(' * uri_list=%r' % uri_list)
print(' * path_list=%r' % path_list)
print(' * gsize_list=%r' % gsize_list)
print(' * time_list=%r' % time_list)
print(' * gps_list=%r' % gps_list)
print('[TEST] 4. get_image_props')
mult_cols_list = ibs.db.get('images',
('image_uri', 'image_width', 'image_height'),
gid_list)
print(' * gps_list=%r' % mult_cols_list)
print('[TEST] 5. add_annots')
gid = gid_list[0]
gid_list = [gid, gid]
bbox_list = [(50, 50, 100, 100), (75, 75, 102, 101)]
theta_list = [0, 1.1]
nid_list = None
name_list = None
notes_list = None
aid_list = ibs.add_annots(gid_list,
bbox_list,
theta_list=theta_list,
nid_list=nid_list,
name_list=name_list,
notes_list=notes_list)
print(' * aid_list=%r' % aid_list)
print('[TEST] 6. get_annot_props')
gid_list = ibs.get_annot_gids(aid_list)
bbox_list = ibs.get_annot_bboxes(aid_list)
theta_list = ibs.get_annot_thetas(aid_list)
aids_list = ibs.get_image_aids(gid)
print(' * gid_list=%r' % gid_list)
print(' * bbox_list=%r' % bbox_list)
print(' * theta_list=%r' % theta_list)
print(' * aids_list=%r' % aids_list)
assert len(bbox_list) == len(
theta_list), "len(bbox_list) != len(theta_list)"
assert len(theta_list) == len(aid_list), "len(theta_list) != len(aid_list)"
viz.show_image(ibs, gid)
return locals()
def iselect_bbox(ibs, gid, fnum=1,
figtitle='Image View - Select ANNOTATION (click two points)',
**kwargs):
#from matplotlib.backend_bases import mplDeprecation
print('[*interact] select_bbox(gid=%r, fnum=%r)' % (gid, fnum))
print('[*interact] Define a Rectanglular ANNOTATION by clicking two points.')
# Show the image
fig = ih.begin_interaction('select_bbox', fnum)
fig_presenter.bring_to_front(fig)
viz.show_image(ibs, gid, **kwargs)
try:
viz.draw()
fig = df2.gcf()
pts = fig.ginput(2)
print('[*guitools] ginput(2) = %r' % (pts,))
[(x1, y1), (x2, y2)] = pts
xm = min(x1, x2)
xM = max(x1, x2)
ym = min(y1, y2)
yM = max(y1, y2)
bbox = tuple(map(int, map(round, (xm, ym, xM - xm, yM - ym))))
# Reconnect the old button press events
print('[*interact] bbox = %r ' % (bbox,))
return bbox
except Exception as ex:
print('<!!!>')
print('[*interact] Caught: %s %s' % (type(ex), ex))
print('[*interact] ANNOTATION selection Failed:')
print('</!!!>')
raise
def iselect_bbox(ibs, gid, fnum=1,
figtitle='Image View - Select ANNOTATION (click two points)',
**kwargs):
#from matplotlib.backend_bases import mplDeprecation
print('[*interact] select_bbox(gid=%r, fnum=%r)' % (gid, fnum))
print('[*interact] Define a Rectanglular ANNOTATION by clicking two points.')
# Show the image
fig = ih.begin_interaction('select_bbox', fnum)
fig_presenter.bring_to_front(fig)
viz.show_image(ibs, gid, **kwargs)
try:
viz.draw()
fig = df2.gcf()
pts = fig.ginput(2)
print('[*guitools] ginput(2) = %r' % (pts,))
[(x1, y1), (x2, y2)] = pts
xm = min(x1, x2)
xM = max(x1, x2)
ym = min(y1, y2)
yM = max(y1, y2)
bbox = tuple(map(int, map(round, (xm, ym, xM - xm, yM - ym))))
# Reconnect the old button press events
print('[*interact] bbox = %r ' % (bbox,))
return bbox
except Exception as ex:
print('<!!!>')
print('[*interact] Caught: %s %s' % (type(ex), ex))
print('[*interact] ANNOTATION selection Failed:')
print('</!!!>')
raise
def TEST_VIZ(ibs):
valid_gids = ibs.get_valid_gids()
valid_aids = ibs.get_valid_aids()
print('len(valid_aids) = %r' % len(valid_aids))
print('len(valid_gids) = %r' % len(valid_gids))
assert len(valid_gids) > 0, 'database images cannot be empty for test'
gindex = 1
gid = valid_gids[gindex]
aid_list = ibs.get_image_aids(gid)
rindex = 0
aid = aid_list[rindex]
qaid = aid
aids = aid_list[1:3]
aid = aid_list[-1]
try:
qres = ibs._query_chips4([qaid], valid_aids)[qaid]
print(qres)
top_aids = qres.get_top_aids(ibs)
assert len(top_aids) > 0, 'Results seems to be empty'
aid2 = top_aids[0] # 294
query_failed = False
except Exception as ex:
query_failed = True
utool.printex(ex, 'QUERY FAILED!')
raise
#----------------------
#print('Show Image')
viz.show_image(ibs, gid, aids=aids, fnum=1)
df2.set_figtitle('Show Image')
#----------------------
#print('Show Chip')
kpts_kwgs = dict(ell=True,
ori=True,
rect=True,
eig=True,
pts=False,
kpts_subset=10)
viz.show_chip(ibs, aid, in_image=False, fnum=2, **kpts_kwgs)
df2.set_figtitle('Show Chip (normal)')
viz.show_chip(ibs, aid, in_image=True, fnum=3, **kpts_kwgs)
df2.set_figtitle('Show Chip (in_image)')
#----------------------
if not query_failed:
print('Show Query')
viz.show_matches(ibs, qres, aid2, fnum=4)
df2.set_figtitle('Show Chipres')
viz.show_qres(ibs, qres, fnum=5)
df2.set_figtitle('Show QRes')
##----------------------
print('return test locals')
return locals()
def _image_view(sel_aids=sel_aids, **_kwargs):
try:
viz.show_image(ibs,
gid,
sel_aids=sel_aids,
fnum=self.fnum,
**_kwargs)
df2.set_figtitle('Image View')
except TypeError as ex:
ut.printex(ex, ut.repr2(_kwargs))
raise
def TEST_IBS(ibs):
gpath_list = grabdata.get_test_gpaths(ndata=2)
print('[TEST] 1.ibs.add_images(gpath_list=%r)' % gpath_list)
gid_list = ibs.add_images(gpath_list)
print(' * gid_list=%r' % gid_list)
print('[TEST] 2. ibs.get_image_paths(gid_list)')
gpaths_list = ibs.get_image_paths(gid_list)
print(' * gpaths_list=%r' % gpaths_list)
print('[TEST] 3. get_image_props')
uri_list = ibs.get_image_uris(gid_list)
path_list = ibs.get_image_paths(gid_list)
gsize_list = ibs.get_image_sizes(gid_list)
time_list = ibs.get_image_unixtime(gid_list)
gps_list = ibs.get_image_gps(gid_list)
print(' * uri_list=%r' % uri_list)
print(' * path_list=%r' % path_list)
print(' * gsize_list=%r' % gsize_list)
print(' * time_list=%r' % time_list)
print(' * gps_list=%r' % gps_list)
print('[TEST] 4. get_image_props')
mult_cols_list = ibs.db.get('images', ('image_uri', 'image_width', 'image_height'), gid_list)
print(' * gps_list=%r' % mult_cols_list)
print('[TEST] 5. add_annots')
gid = gid_list[0]
gid_list = [gid, gid]
bbox_list = [(50, 50, 100, 100), (75, 75, 102, 101)]
theta_list = [0, 1.1]
nid_list = None
name_list = None
notes_list = None
aid_list = ibs.add_annots(gid_list, bbox_list, theta_list=theta_list,
nid_list=nid_list, name_list=name_list,
notes_list=notes_list)
print(' * aid_list=%r' % aid_list)
print('[TEST] 6. get_annot_props')
gid_list = ibs.get_annot_gids(aid_list)
bbox_list = ibs.get_annot_bboxes(aid_list)
theta_list = ibs.get_annot_thetas(aid_list)
aids_list = ibs.get_image_aids(gid)
print(' * gid_list=%r' % gid_list)
print(' * bbox_list=%r' % bbox_list)
print(' * theta_list=%r' % theta_list)
print(' * aids_list=%r' % aids_list)
assert len(bbox_list) == len(theta_list), "len(bbox_list) != len(theta_list)"
assert len(theta_list) == len(aid_list), "len(theta_list) != len(aid_list)"
viz.show_image(ibs, gid)
return locals()
def TEST_VIZ(ibs):
valid_gids = ibs.get_valid_gids()
valid_aids = ibs.get_valid_aids()
print('len(valid_aids) = %r' % len(valid_aids))
print('len(valid_gids) = %r' % len(valid_gids))
assert len(valid_gids) > 0, 'database images cannot be empty for test'
gindex = 1
gid = valid_gids[gindex]
aid_list = ibs.get_image_aids(gid)
rindex = 0
aid = aid_list[rindex]
qaid = aid
aids = aid_list[1:3]
aid = aid_list[-1]
try:
qres = ibs._query_chips4([qaid], valid_aids)[qaid]
print(qres)
top_aids = qres.get_top_aids(ibs)
assert len(top_aids) > 0, 'Results seems to be empty'
aid2 = top_aids[0] # 294
query_failed = False
except Exception as ex:
query_failed = True
utool.printex(ex, 'QUERY FAILED!')
raise
#----------------------
#print('Show Image')
viz.show_image(ibs, gid, aids=aids, fnum=1)
df2.set_figtitle('Show Image')
#----------------------
#print('Show Chip')
kpts_kwgs = dict(ell=True, ori=True, rect=True,
eig=True, pts=False, kpts_subset=10)
viz.show_chip(ibs, aid, in_image=False, fnum=2, **kpts_kwgs)
df2.set_figtitle('Show Chip (normal)')
viz.show_chip(ibs, aid, in_image=True, fnum=3, **kpts_kwgs)
df2.set_figtitle('Show Chip (in_image)')
#----------------------
if not query_failed:
print('Show Query')
viz.show_matches(ibs, qres, aid2, fnum=4)
df2.set_figtitle('Show Chipres')
viz.show_qres(ibs, qres, fnum=5)
df2.set_figtitle('Show QRes')
##----------------------
print('return test locals')
return locals()
