Source code for wbia.plottool.tests.test_viz_image2

#!/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
import cv2
import utool
import numpy as np
from wbia.plottool.tests.test_helpers import dummy_bbox


[docs]def test_viz_image(img_fpath): # Read image img = cv2.imread(img_fpath) tau = np.pi * 2 # References: tauday.com # Create figure fig = df2.figure(fnum=42, pnum=(1, 1, 1)) # Clear figure fig.clf() # Build parameters bbox_list = [dummy_bbox(img), dummy_bbox(img, (-0.25, -0.25), 0.1)] showkw = { 'title': 'test axis title', # The list of bounding boxes to be drawn on the image 'bbox_list': bbox_list, 'theta_list': [tau * 0.7, tau * 0.9], 'sel_list': [True, False], 'label_list': ['test label', 'lbl2'], } # Print the keyword arguments to illustrate their format print('showkw = ' + utool.repr2(showkw)) # Display the image in figure-num 42, using a 1x1 axis grid in the first # axis. Pass showkw as keyword arguments. viz_image2.show_image(img, fnum=42, pnum=(1, 1, 1), **showkw) df2.set_figtitle('Test figure title')
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, recursive=False ) # test image paths # Get one image filepath to load and display img_fpath = imgpaths[0] # Run Test test_viz_image(img_fpath) # Magic exec which displays or puts you into IPython with --cmd flag exec(df2.present())