24 lines
894 B
Python
24 lines
894 B
Python
import os
|
|
import sys
|
|
from collections import defaultdict
|
|
|
|
from PIL import Image
|
|
|
|
# Python does not consider the current directory to be a package
|
|
sys.path.insert(0, os.path.join('..', 'monoloco'))
|
|
|
|
|
|
def test_printer():
|
|
"""Draw a fake figure"""
|
|
from stereoloco.visuals.printer import Printer
|
|
test_list = [[718.3351, 0., 600.3891], [0., 718.3351, 181.5122], [0., 0., 1.]]
|
|
boxes = [xx + [0] for xx in test_list]
|
|
kk = test_list
|
|
dict_ann = defaultdict(lambda: [1., 2., 3.], xyz_real=test_list, xyz_pred=test_list, uv_shoulders=test_list,
|
|
boxes=boxes, boxes_gt=boxes)
|
|
with open('docs/002282.png', 'rb') as f:
|
|
pil_image = Image.open(f).convert('RGB')
|
|
printer = Printer(image=pil_image, output_path=None, kk=kk, output_types=['combined'])
|
|
figures, axes = printer.factory_axes()
|
|
printer.draw(figures, axes, dict_ann, pil_image)
|