def render_single_result(request, cm, aid, **kwargs):
# HACK FOR WEB VIEWER
chips = request.get_fmatch_overlayed_chip([cm.qaid, aid],
config=request.config)
import vtool as vt
out_img = vt.stack_image_list(chips)
return out_img
def render_single_result(request, cm, aid, **kwargs):
# HACK FOR WEB VIEWER
overlay = kwargs.get('draw_fmatches')
chips = request.get_fmatch_overlayed_chip(
[cm.qaid, aid], overlay=overlay, config=request.config
)
out_img = vt.stack_image_list(chips)
return out_img
def visualize_vocab_word(ibs, invassign, wx, fnum=None):
"""
Example:
>>> from ibeis.new_annots import * # NOQA
>>> import plottool as pt
>>> pt.qt4ensure()
>>> ibs, aid_list, vocab = testdata_vocab()
>>> #aid_list = aid_list[0:1]
>>> fstack = StackedFeatures(ibs, aid_list)
>>> nAssign = 2
>>> invassign = fstack.inverted_assignment(vocab, nAssign)
>>> sortx = ut.argsort(invassign.num_list)[::-1]
>>> wx_list = ut.take(invassign.wx_list, sortx)
>>> wx = wx_list[0]
"""
import plottool as pt
pt.qt4ensure()
vecs = invassign.get_vecs(wx)
word = invassign.vocab.wx2_word[wx]
word_patches = invassign.get_patches(wx)
average_patch = np.mean(word_patches, axis=0)
average_vec = vecs.mean(axis=0)
average_vec = word
word
with_sift = True
fnum = 2
fnum = pt.ensure_fnum(fnum)
if with_sift:
patch_img = pt.render_sift_on_patch(average_patch, average_vec)
#sift_word_patches = [pt.render_sift_on_patch(patch, vec) for patch, vec in ut.ProgIter(list(zip(word_patches, vecs)))]
#stacked_patches = vt.stack_square_images(word_patches)
#stacked_patches = vt.stack_square_images(sift_word_patches)
else:
patch_img = average_patch
stacked_patches = vt.stack_square_images(word_patches)
solidbar = np.zeros((patch_img.shape[0], int(patch_img.shape[1] * .1), 3),
dtype=patch_img.dtype)
border_color = (100, 10, 10) # bgr, darkblue
if ut.is_float(solidbar):
solidbar[:, :, :] = (np.array(border_color) / 255)[None, None]
else:
solidbar[:, :, :] = np.array(border_color)[None, None]
word_img = vt.stack_image_list([patch_img, solidbar, stacked_patches],
vert=False,
modifysize=True)
pt.imshow(word_img, fnum=fnum)
#pt.imshow(patch_img, pnum=(1, 2, 1), fnum=fnum)
#patch_size = 64
#half_size = patch_size / 2
#pt.imshow(stacked_patches, pnum=(1, 2, 2), fnum=fnum)
pt.iup()
def render_single_result(request, cm, aid, **kwargs):
# HACK FOR WEB VIEWER
if kwargs.get('draw_fmatches'):
chips = request.get_fmatch_overlayed_chip([cm.qaid, aid], request.config)
else:
depc = request.depc
chips = depc.get('Cropped_Chips', [cm.qaid, aid], 'img', config=request.config)
import vtool as vt
out_img = vt.stack_image_list(chips)
return out_img
def visualize_vocab_word(ibs, invassign, wx, fnum=None):
"""
Example:
>>> from ibeis.new_annots import * # NOQA
>>> import plottool as pt
>>> pt.qt4ensure()
>>> ibs, aid_list, vocab = testdata_vocab()
>>> #aid_list = aid_list[0:1]
>>> fstack = StackedFeatures(ibs, aid_list)
>>> nAssign = 2
>>> invassign = fstack.inverted_assignment(vocab, nAssign)
>>> sortx = ut.argsort(invassign.num_list)[::-1]
>>> wx_list = ut.take(invassign.wx_list, sortx)
>>> wx = wx_list[0]
"""
import plottool as pt
pt.qt4ensure()
vecs = invassign.get_vecs(wx)
word = invassign.vocab.wx2_word[wx]
word_patches = invassign.get_patches(wx)
average_patch = np.mean(word_patches, axis=0)
average_vec = vecs.mean(axis=0)
average_vec = word
word
with_sift = True
fnum = 2
fnum = pt.ensure_fnum(fnum)
if with_sift:
patch_img = pt.render_sift_on_patch(average_patch, average_vec)
#sift_word_patches = [pt.render_sift_on_patch(patch, vec) for patch, vec in ut.ProgIter(list(zip(word_patches, vecs)))]
#stacked_patches = vt.stack_square_images(word_patches)
#stacked_patches = vt.stack_square_images(sift_word_patches)
else:
patch_img = average_patch
stacked_patches = vt.stack_square_images(word_patches)
solidbar = np.zeros((patch_img.shape[0], int(patch_img.shape[1] * .1), 3), dtype=patch_img.dtype)
border_color = (100, 10, 10) # bgr, darkblue
if ut.is_float(solidbar):
solidbar[:, :, :] = (np.array(border_color) / 255)[None, None]
else:
solidbar[:, :, :] = np.array(border_color)[None, None]
word_img = vt.stack_image_list([patch_img, solidbar, stacked_patches], vert=False, modifysize=True)
pt.imshow(word_img, fnum=fnum)
#pt.imshow(patch_img, pnum=(1, 2, 1), fnum=fnum)
#patch_size = 64
#half_size = patch_size / 2
#pt.imshow(stacked_patches, pnum=(1, 2, 2), fnum=fnum)
pt.iup()
def testshow_colors(rgb_list, gray=ut.get_argflag('--gray')):
"""
colors = ['r', 'b', 'purple', 'orange', 'deeppink', 'g']
colors = list(mcolors.CSS4_COLORS.keys())
CommandLine:
python -m wbia.plottool.color_funcs testshow_colors --show
Example:
>>> # ENABLE_DOCTEST
>>> from wbia.plottool.color_funcs import * # NOQA
>>> colors = ut.get_argval('--colors', type_=list, default=['k', 'r'])
>>> ut.quit_if_noshow()
>>> rgb_list = ut.emap(ensure_base01, colors)
>>> testshow_colors(rgb_list)
>>> import wbia.plottool as pt
>>> pt.show_if_requested()
"""
import wbia.plottool as pt
import vtool as vt
block = np.zeros((5, 5, 3))
block_list = [block + color[0:3] for color in rgb_list]
# print(ut.repr2(block_list))
# print(ut.repr2(rgb_list))
chunks = ut.ichunks(block_list, 10)
stacked_chunk = []
for chunk in chunks:
stacked_chunk.append(vt.stack_image_list(chunk, vert=False))
stacked_block = vt.stack_image_list(stacked_chunk, vert=True)
# convert to bgr
stacked_block = stacked_block[:, :, ::-1]
uint8_img = (255 * stacked_block).astype(np.uint8)
if gray:
import cv2
uint8_img = cv2.cvtColor(uint8_img, cv2.COLOR_RGB2GRAY)
pt.imshow(uint8_img)
def render_single_result(request, cm, aid, **kwargs):
# HACK FOR WEB VIEWER
if kwargs.get('draw_fmatches'):
chips = request.get_fmatch_overlayed_chip([cm.qaid, aid],
request.config)
else:
depc = request.depc
chips = depc.get('Cropped_Chips', [cm.qaid, aid],
'img',
config=request.config)
import vtool as vt
out_img = vt.stack_image_list(chips)
return out_img
def testshow_colors(rgb_list, gray=ut.get_argflag('--gray')):
import plottool as pt
import vtool as vt
block = np.zeros((5, 5, 3))
block_list = [block + color[0:3] for color in rgb_list]
#print(ut.list_str(block_list))
#print(ut.list_str(rgb_list))
stacked_block = vt.stack_image_list(block_list, vert=False)
# convert to bgr
stacked_block = stacked_block[:, :, ::-1]
uint8_img = (255 * stacked_block).astype(np.uint8)
if gray:
import cv2
uint8_img = cv2.cvtColor(uint8_img, cv2.COLOR_RGB2GRAY)
pt.imshow(uint8_img)
pt.show_if_requested()
def render_inverted_vocab_word(inva, wx, ibs, fnum=None):
"""
Creates a visualization of a visual word. This includes the average patch,
the SIFT-like representation of the centroid, and some of the patches that
were assigned to it.
CommandLine:
python -m wbia.algo.smk.inverted_index render_inverted_vocab_word --show
Example:
>>> # DISABLE_DOCTEST
>>> from wbia.algo.smk.inverted_index import * # NOQA
>>> import wbia.plottool as pt
>>> qreq_, inva = testdata_inva()
>>> ibs = qreq_.ibs
>>> wx_list = list(inva.wx_to_aids.keys())
>>> wx = wx_list[0]
>>> ut.qtensure()
>>> fnum = 2
>>> fnum = pt.ensure_fnum(fnum)
>>> # Interactive visualization of many words
>>> for wx in ut.InteractiveIter(wx_list):
>>> word_img = inva.render_inverted_vocab_word(wx, ibs, fnum)
>>> pt.imshow(word_img, fnum=fnum, title='Word %r/%r' % (wx, '?'))
>>> pt.update()
"""
import wbia.plottool as pt
# Create the contributing patch image
word_patches = inva.get_patches(wx, ibs)
word_patches_ = ut.strided_sample(word_patches, 64)
stacked_patches = vt.stack_square_images(word_patches_)
# Create the average word image
vocab = ibs.depc['vocab'].get_row_data([inva.vocab_rowid], 'words')[0]
word = vocab.wx_to_word[wx]
average_patch = np.mean(word_patches, axis=0)
# vecs = inva.get_vecs(wx)
# assert np.allclose(word, vecs.mean(axis=0))
with_sift = True
if with_sift:
patch_img = pt.render_sift_on_patch(average_patch, word)
else:
patch_img = average_patch
# Stack them together
solidbar = np.zeros(
(patch_img.shape[0], int(patch_img.shape[1] * 0.1), 3), dtype=patch_img.dtype,
)
border_color = (100, 10, 10) # bgr, darkblue
if ut.is_float(solidbar):
solidbar[:, :, :] = (np.array(border_color) / 255)[None, None]
else:
solidbar[:, :, :] = np.array(border_color)[None, None]
patch_img2 = vt.inverted_sift_patch(word)
# Fix types
patch_img = vt.rectify_to_uint8(patch_img)
patch_img2 = vt.rectify_to_uint8(patch_img2)
solidbar = vt.rectify_to_uint8(solidbar)
stacked_patches = vt.rectify_to_uint8(stacked_patches)
# Stack everything together
patch_img2, patch_img = vt.make_channels_comparable(patch_img2, patch_img)
img_list = [patch_img, solidbar, patch_img2, solidbar, stacked_patches]
word_img = vt.stack_image_list(img_list, vert=False, modifysize=True)
return word_img
