def get_data_annot_pair_info(ibs, aid_list, qreq_, draw_fmatches, scale_down=False, kpts2_list=None): data_config2_ = (None if qreq_ is None else qreq_.get_external_data_config2()) #print('!!! data_config2_ = %r' % (data_config2_,)) #print('!!! dqreq_ = %r' % (qreq_,)) tblhack = getattr(qreq_, 'tablename', None) if (not tblhack or tblhack == 'BC_DTW') and getattr(qreq_, '_isnewreq', None): if hasattr(qreq_, 'get_fmatch_overlayed_chip') and draw_fmatches and draw_fmatches != 'hackoff': rchip2_list = qreq_.get_fmatch_overlayed_chip(aid_list, config=data_config2_) #rchip2_list = ibs.depc_annot.get_property('chips', aid_list, 'img', config=data_config2_) draw_fmatches = False else: rchip2_list = ibs.depc_annot.get_property('chips', aid_list, 'img', config=data_config2_) draw_fmatches = False #vh.get_chips(ibs, aid_list, config2_=data_config2_) else: rchip2_list = vh.get_chips(ibs, aid_list, config2_=data_config2_) if draw_fmatches: if kpts2_list is None: kpts2_list = vh.get_kpts(ibs, aid_list, config2_=data_config2_) else: kpts2_list = [None] * len(aid_list) if scale_down: pass return rchip2_list, kpts2_list
def get_query_annot_pair_info(ibs, qaid, qreq_, draw_fmatches, kpts1=None): #print('!!! qqreq_ = %r' % (qreq_,)) query_config2_ = (None if qreq_ is None else qreq_.get_external_query_config2()) tblhack = getattr(qreq_, 'tablename', None) #print('!!! query_config2_ = %r' % (query_config2_,)) if (not tblhack or tblhack == 'BC_DTW') and getattr( qreq_, '_isnewreq', None): if hasattr(qreq_, 'get_fmatch_overlayed_chip' ) and draw_fmatches and draw_fmatches != 'hackoff': rchip1 = qreq_.get_fmatch_overlayed_chip(qaid, config=query_config2_) draw_fmatches = False else: rchip1 = ibs.depc_annot.get_property('chips', qaid, 'img', config=query_config2_) draw_fmatches = False else: rchip1 = vh.get_chips(ibs, [qaid], config2_=query_config2_)[0] if draw_fmatches: if kpts1 is None: kpts1 = vh.get_kpts(ibs, [qaid], config2_=query_config2_)[0] else: kpts1 = None return rchip1, kpts1
def get_query_annot_pair_info(ibs, qaid, qreq_, draw_fmatches, kpts1=None, as_fpath=False): #print('!!! qqreq_ = %r' % (qreq_,)) query_config2_ = (None if qreq_ is None else qreq_.extern_query_config2) tblhack = getattr(qreq_, 'tablename', None) #print('!!! query_config2_ = %r' % (query_config2_,)) if (not tblhack or tblhack.lower() in [ 'bc_dtw', 'oc_wdtw', 'curvrankdorsal', 'curvrankfluke' ]) and getattr(qreq_, '_isnewreq', None): if hasattr(qreq_, 'get_fmatch_overlayed_chip' ) and draw_fmatches and draw_fmatches != 'hackoff': rchip1 = qreq_.get_fmatch_overlayed_chip(qaid, config=query_config2_) draw_fmatches = False else: rchip1 = ibs.depc_annot.get_property('chips', qaid, 'img', config=query_config2_) draw_fmatches = False else: rchip1 = vh.get_chips(ibs, [qaid], config2_=query_config2_, as_fpath=as_fpath)[0] if draw_fmatches: if kpts1 is None: kpts1 = vh.get_kpts(ibs, [qaid], config2_=query_config2_)[0] else: kpts1 = None return rchip1, kpts1
def get_query_annot_pair_info(ibs, qaid, qreq_, draw_fmatches, kpts1=None): #print('!!! qqreq_ = %r' % (qreq_,)) query_config2_ = (None if qreq_ is None else qreq_.get_external_query_config2()) tblhack = getattr(qreq_, 'tablename', None) #print('!!! query_config2_ = %r' % (query_config2_,)) if (not tblhack or tblhack == 'BC_DTW') and getattr(qreq_, '_isnewreq', None): if hasattr(qreq_, 'get_fmatch_overlayed_chip') and draw_fmatches and draw_fmatches != 'hackoff': rchip1 = qreq_.get_fmatch_overlayed_chip(qaid, config=query_config2_) draw_fmatches = False else: rchip1 = ibs.depc_annot.get_property('chips', qaid, 'img', config=query_config2_) draw_fmatches = False else: rchip1 = vh.get_chips(ibs, [qaid], config2_=query_config2_)[0] if draw_fmatches: if kpts1 is None: kpts1 = vh.get_kpts(ibs, [qaid], config2_=query_config2_)[0] else: kpts1 = None return rchip1, kpts1
def get_data_annot_pair_info(ibs, aid_list, qreq_, draw_fmatches, scale_down=False, kpts2_list=None, as_fpath=False): data_config2_ = (None if qreq_ is None else qreq_.extern_data_config2) #print('!!! data_config2_ = %r' % (data_config2_,)) #print('!!! dqreq_ = %r' % (qreq_,)) tblhack = getattr(qreq_, 'tablename', None) if (not tblhack or tblhack.lower() in [ 'bc_dtw', 'oc_wdtw', 'curvrankdorsal', 'curvrankfluke' ]) and getattr(qreq_, '_isnewreq', None): if hasattr(qreq_, 'get_fmatch_overlayed_chip' ) and draw_fmatches and draw_fmatches != 'hackoff': rchip2_list = qreq_.get_fmatch_overlayed_chip(aid_list, config=data_config2_) #rchip2_list = ibs.depc_annot.get_property('chips', aid_list, 'img', config=data_config2_) draw_fmatches = False else: rchip2_list = ibs.depc_annot.get_property('chips', aid_list, 'img', config=data_config2_) draw_fmatches = False #vh.get_chips(ibs, aid_list, config2_=data_config2_) else: rchip2_list = vh.get_chips(ibs, aid_list, config2_=data_config2_, as_fpath=as_fpath) if draw_fmatches: if kpts2_list is None: kpts2_list = vh.get_kpts(ibs, aid_list, config2_=data_config2_) else: kpts2_list = [None] * len(aid_list) if scale_down: pass return rchip2_list, kpts2_list
def get_data_annot_pair_info(ibs, aid_list, qreq_, draw_fmatches, scale_down=False, kpts2_list=None): data_config2_ = (None if qreq_ is None else qreq_.get_external_data_config2()) #print('!!! data_config2_ = %r' % (data_config2_,)) #print('!!! dqreq_ = %r' % (qreq_,)) tblhack = getattr(qreq_, 'tablename', None) if (not tblhack or tblhack == 'BC_DTW') and getattr( qreq_, '_isnewreq', None): if hasattr(qreq_, 'get_fmatch_overlayed_chip' ) and draw_fmatches and draw_fmatches != 'hackoff': rchip2_list = qreq_.get_fmatch_overlayed_chip(aid_list, config=data_config2_) #rchip2_list = ibs.depc_annot.get_property('chips', aid_list, 'img', config=data_config2_) draw_fmatches = False else: rchip2_list = ibs.depc_annot.get_property('chips', aid_list, 'img', config=data_config2_) draw_fmatches = False #vh.get_chips(ibs, aid_list, config2_=data_config2_) else: rchip2_list = vh.get_chips(ibs, aid_list, config2_=data_config2_) if draw_fmatches: if kpts2_list is None: kpts2_list = vh.get_kpts(ibs, aid_list, config2_=data_config2_) else: kpts2_list = [None] * len(aid_list) if scale_down: pass return rchip2_list, kpts2_list
def show_chip(ibs, aid, in_image=False, annote=True, title_suffix='', weight_label=None, weights=None, config2_=None, **kwargs): r""" Driver function to show chips Args: ibs (ibeis.IBEISController): aid (int): annotation rowid in_image (bool): displays annotation with the context of its source image annote (bool): enables overlay annoations title_suffix (str): weight_label (None): (default = None) weights (None): (default = None) config2_ (dict): (default = None) Kwargs: enable_chip_title_prefix, nokpts, kpts_subset, kpts, text_color, notitle, draw_lbls, show_aidstr, show_gname, show_name, show_nid, show_exemplar, show_num_gt, show_quality_text, show_viewcode, fnum, title, figtitle, pnum, interpolation, cmap, heatmap, data_colorbar, darken, update, xlabel, redraw_image, ax, alpha, docla, doclf, projection, pts, ell color (3/4-tuple, ndarray, or str): colors for keypoints CommandLine: python -m ibeis.viz.viz_chip show_chip --show --ecc python -c "import utool as ut; ut.print_auto_docstr('ibeis.viz.viz_chip', 'show_chip')" python -m ibeis.viz.viz_chip show_chip --show --db NNP_Master3 --aids 14047 --no-annote python -m ibeis.viz.viz_chip show_chip --show --db NNP_Master3 --aids 14047 --no-annote python -m ibeis.viz.viz_chip show_chip --show --db PZ_MTEST --aid 1 --bgmethod=cnn python -m ibeis.viz.viz_chip show_chip --show --db PZ_MTEST --aid 1 --bgmethod=cnn --scale_max=30 python -m ibeis.viz.viz_chip show_chip --show --db PZ_MTEST --aid 1 --ecc --draw_lbls=False --notitle --save=~/slides/lnbnn_query.jpg --dpi=300 Example: >>> # VIZ_TEST >>> from ibeis.viz.viz_chip import * # NOQA >>> import numpy as np >>> import vtool_ibeis as vt >>> in_image = False >>> ibs, aid_list, kwargs, config2_ = testdata_showchip() >>> aid = aid_list[0] >>> if True: >>> import matplotlib as mpl >>> from ibeis.scripts.thesis import TMP_RC >>> mpl.rcParams.update(TMP_RC) >>> if ut.get_argflag('--ecc'): >>> kpts = ibs.get_annot_kpts(aid, config2_=config2_) >>> weights = ibs.get_annot_fgweights([aid], ensure=True, config2_=config2_)[0] >>> kpts = ut.random_sample(kpts[weights > .9], 200, seed=0) >>> ecc = vt.get_kpts_eccentricity(kpts) >>> scale = 1 / vt.get_scales(kpts) >>> #s = ecc if config2_.affine_invariance else scale >>> s = scale >>> colors = pt.scores_to_color(s, cmap_='jet') >>> kwargs['color'] = colors >>> kwargs['kpts'] = kpts >>> kwargs['ell_linewidth'] = 3 >>> kwargs['ell_alpha'] = .7 >>> show_chip(ibs, aid, in_image=in_image, config2_=config2_, **kwargs) >>> pt.show_if_requested() """ if ut.VERBOSE: print('[viz] show_chip(aid=%r)' % (aid, )) #ibs.assert_valid_aids((aid,)) # Get chip #print('in_image = %r' % (in_image,)) chip = vh.get_chips(ibs, aid, in_image=in_image, config2_=config2_) # Create chip title chip_text = vh.get_annot_texts(ibs, [aid], **kwargs)[0] if kwargs.get('enable_chip_title_prefix', True): chip_title_text = chip_text + title_suffix else: chip_title_text = title_suffix chip_title_text = chip_title_text.strip('\n') # Draw chip fig, ax = pt.imshow(chip, **kwargs) # Populate axis user data vh.set_ibsdat(ax, 'viztype', 'chip') vh.set_ibsdat(ax, 'aid', aid) if annote and not kwargs.get('nokpts', False): # Get and draw keypoints if 'color' not in kwargs: if weight_label == 'fg_weights': if weights is None and ibs.has_species_detector( ibs.get_annot_species_texts(aid)): weight_label = 'fg_weights' weights = ibs.get_annot_fgweights([aid], ensure=True, config2_=config2_)[0] if weights is not None: cmap_ = 'hot' #if weight_label == 'dstncvs': # cmap_ = 'rainbow' color = pt.scores_to_color(weights, cmap_=cmap_, reverse_cmap=False) kwargs['color'] = color kwargs['ell_color'] = color kwargs['pts_color'] = color kpts_ = vh.get_kpts(ibs, aid, in_image, config2_=config2_, kpts_subset=kwargs.get('kpts_subset', None), kpts=kwargs.pop('kpts', None)) pt.viz_keypoints._annotate_kpts(kpts_, **kwargs) if kwargs.get('draw_lbls', True): pt.upperleft_text(chip_text, color=kwargs.get('text_color', None)) use_title = not kwargs.get('notitle', False) if use_title: pt.set_title(chip_title_text) if in_image: gid = ibs.get_annot_gids(aid) aid_list = ibs.get_image_aids(gid) annotekw = viz_image.get_annot_annotations(ibs, aid_list, sel_aids=[aid], draw_lbls=kwargs.get( 'draw_lbls', True)) # Put annotation centers in the axis ph.set_plotdat(ax, 'annotation_bbox_list', annotekw['bbox_list']) ph.set_plotdat(ax, 'aid_list', aid_list) pt.viz_image2.draw_image_overlay(ax, **annotekw) zoom_ = ut.get_argval('--zoom', type_=float, default=None) if zoom_ is not None: import vtool_ibeis as vt # Zoom into the chip for some image context rotated_verts = ibs.get_annot_rotated_verts(aid) bbox = ibs.get_annot_bboxes(aid) #print(bbox) #print(rotated_verts) rotated_bbox = vt.bbox_from_verts(rotated_verts) imgw, imgh = ibs.get_image_sizes(gid) pad_factor = zoom_ pad_length = min(bbox[2], bbox[3]) * pad_factor minx = max(rotated_bbox[0] - pad_length, 0) miny = max(rotated_bbox[1] - pad_length, 0) maxx = min((rotated_bbox[0] + rotated_bbox[2]) + pad_length, imgw) maxy = min((rotated_bbox[1] + rotated_bbox[3]) + pad_length, imgh) #maxy = imgh - maxy #miny = imgh - miny ax = pt.gca() ax.set_xlim(minx, maxx) ax.set_ylim(miny, maxy) ax.invert_yaxis() else: ph.set_plotdat(ax, 'chipshape', chip.shape) #if 'featweights' in vars() and 'color' in kwargs: if weights is not None and weight_label is not None: ## HACK HACK HACK if len(weights) > 0: cb = pt.colorbar(weights, kwargs['color']) cb.set_label(weight_label) return fig, ax
def show_chip(ibs, aid, in_image=False, annote=True, title_suffix='', weight_label=None, weights=None, config2_=None, **kwargs): r""" Driver function to show chips Args: ibs (ibeis.IBEISController): aid (int): annotation rowid in_image (bool): displays annotation with the context of its source image annote (bool): enables overlay annoations title_suffix (str): weight_label (None): (default = None) weights (None): (default = None) config2_ (dict): (default = None) Kwargs: enable_chip_title_prefix, nokpts, kpts_subset, kpts, text_color, notitle, draw_lbls, show_aidstr, show_gname, show_name, show_nid, show_exemplar, show_num_gt, show_quality_text, show_yawtext, fnum, title, figtitle, pnum, interpolation, cmap, heatmap, data_colorbar, darken, update, xlabel, redraw_image, ax, alpha, docla, doclf, projection, use_gridspec, pts, ell color (3/4-tuple, ndarray, or str): colors for keypoints CommandLine: python -m ibeis.viz.viz_chip --test-show_chip --show --ecc python -c "import utool as ut; ut.print_auto_docstr('ibeis.viz.viz_chip', 'show_chip')" python -m ibeis.viz.viz_chip --test-show_chip --show --db NNP_Master3 --aids 14047 --no-annote python -m ibeis.viz.viz_chip --test-show_chip --show --db NNP_Master3 --aids 14047 --no-annote python -m ibeis.viz.viz_chip --test-show_chip --show --db PZ_MTEST --aid 1 --bgmethod=cnn python -m ibeis.viz.viz_chip --test-show_chip --show --db PZ_MTEST --aid 1 --bgmethod=cnn --scale_max=30 Example: >>> # VIZ_TEST >>> from ibeis.viz.viz_chip import * # NOQA >>> import numpy as np >>> import vtool as vt >>> in_image = False >>> ibs, aid_list, kwargs, config2_ = testdata_showchip() >>> aid = aid_list[0] >>> if ut.get_argflag('--ecc'): >>> kpts = ibs.get_annot_kpts(aid, config2_=config2_) >>> weights = ibs.get_annot_fgweights([aid], ensure=True, config2_=config2_)[0] >>> kpts = ut.random_sample(kpts[weights > .9], 200, seed=0) >>> ecc = vt.get_kpts_eccentricity(kpts) >>> scale = 1 / vt.get_scales(kpts) >>> s = ecc if config2_.affine_invariance else scale >>> colors = pt.scores_to_color(s, cmap_='jet') >>> kwargs['color'] = colors >>> kwargs['kpts'] = kpts >>> show_chip(ibs, aid, in_image=in_image, config2_=config2_, **kwargs) >>> pt.show_if_requested() """ if ut.VERBOSE: print('[viz] show_chip(aid=%r)' % (aid,)) #ibs.assert_valid_aids((aid,)) # Get chip #print('in_image = %r' % (in_image,)) chip = vh.get_chips(ibs, aid, in_image=in_image, config2_=config2_) # Create chip title chip_text = vh.get_annot_texts(ibs, [aid], **kwargs)[0] if kwargs.get('enable_chip_title_prefix', True): chip_title_text = chip_text + title_suffix else: chip_title_text = title_suffix chip_title_text = chip_title_text.strip('\n') # Draw chip fig, ax = pt.imshow(chip, **kwargs) # Populate axis user data vh.set_ibsdat(ax, 'viztype', 'chip') vh.set_ibsdat(ax, 'aid', aid) if annote and not kwargs.get('nokpts', False): # Get and draw keypoints if 'color' not in kwargs: if weight_label == 'fg_weights': if weights is None and ibs.has_species_detector(ibs.get_annot_species_texts(aid)): weight_label = 'fg_weights' weights = ibs.get_annot_fgweights([aid], ensure=True, config2_=config2_)[0] if weights is not None: cmap_ = 'hot' #if weight_label == 'dstncvs': # cmap_ = 'rainbow' color = pt.scores_to_color(weights, cmap_=cmap_, reverse_cmap=False) kwargs['color'] = color kwargs['ell_color'] = color kwargs['pts_color'] = color kpts_ = vh.get_kpts(ibs, aid, in_image, config2_=config2_, kpts_subset=kwargs.get('kpts_subset', None), kpts=kwargs.get('kpts', None)) try: del kwargs['kpts'] except KeyError: pass pt.viz_keypoints._annotate_kpts(kpts_, **kwargs) if not ut.get_argflag('--noaidlabel'): pt.upperleft_text(chip_text, color=kwargs.get('text_color', None)) use_title = not kwargs.get('notitle', False) if use_title: pt.set_title(chip_title_text) if in_image: gid = ibs.get_annot_gids(aid) aid_list = ibs.get_image_aids(gid) annotekw = viz_image.get_annot_annotations( ibs, aid_list, sel_aids=[aid], draw_lbls=kwargs.get('draw_lbls', True)) # Put annotation centers in the axis ph.set_plotdat(ax, 'annotation_bbox_list', annotekw['bbox_list']) ph.set_plotdat(ax, 'aid_list', aid_list) pt.viz_image2.draw_image_overlay(ax, **annotekw) zoom_ = ut.get_argval('--zoom', type_=float, default=None) if zoom_ is not None: import vtool as vt # Zoom into the chip for some image context rotated_verts = ibs.get_annot_rotated_verts(aid) bbox = ibs.get_annot_bboxes(aid) #print(bbox) #print(rotated_verts) rotated_bbox = vt.bbox_from_verts(rotated_verts) imgw, imgh = ibs.get_image_sizes(gid) pad_factor = zoom_ pad_length = min(bbox[2], bbox[3]) * pad_factor minx = max(rotated_bbox[0] - pad_length, 0) miny = max(rotated_bbox[1] - pad_length, 0) maxx = min((rotated_bbox[0] + rotated_bbox[2]) + pad_length, imgw) maxy = min((rotated_bbox[1] + rotated_bbox[3]) + pad_length, imgh) #maxy = imgh - maxy #miny = imgh - miny ax = pt.gca() ax.set_xlim(minx, maxx) ax.set_ylim(miny, maxy) ax.invert_yaxis() else: ph.set_plotdat(ax, 'chipshape', chip.shape) #if 'featweights' in vars() and 'color' in kwargs: if weights is not None and weight_label is not None: ## HACK HACK HACK if len(weights) > 0: cb = pt.colorbar(weights, kwargs['color']) cb.set_label(weight_label) return fig, ax