Source code for wbia.plottool.tests.test_viz_images

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, print_function
from wbia.plottool import viz_image2
from wbia.plottool import draw_func2 as df2
from wbia.plottool import plot_helpers as ph
import utool
import numpy as np
from wbia.plottool.tests.test_helpers import dummy_bbox, imread_many


[docs]def test_viz_image(imgpaths): nImgs = len(imgpaths) assert len(imgpaths) < 20, '%d > 20 out of scope of this test' % nImgs tau = np.pi * 2 fnum = 1 img_list = imread_many(imgpaths) nRows, nCols = ph.get_square_row_cols(nImgs) print('[viz*] r=%r, c=%r' % (nRows, nCols)) # gs2 = gridspec.GridSpec(nRows, nCols) pnum_ = df2.get_pnum_func(nRows, nCols) fig = df2.figure(fnum=fnum, pnum=pnum_(0)) fig.clf() for px, img in enumerate(img_list): title = 'test title' bbox_list = [dummy_bbox(img), dummy_bbox(img, (-0.25, -0.25), 0.1)] theta_list = [tau * 0.7, tau * 0.9] sel_list = [True, False] label_list = ['test label', 'lbl2'] viz_image2.show_image( img, bbox_list=bbox_list, title=title, sel_list=sel_list, label_list=label_list, theta_list=theta_list, fnum=fnum, pnum=pnum_(px), )
if __name__ == '__main__': TEST_IMAGES_URL = 'https://wildbookiarepository.azureedge.net/data/testdata.zip' test_image_dir = utool.grab_zipped_url(TEST_IMAGES_URL, appname='utool') imgpaths = utool.list_images(test_image_dir, fullpath=True) # test image paths test_viz_image(imgpaths) exec(df2.present())