Vizer

Boxes and masks visualization tools.


Keywords
object-detection, pytorch, visualization
Install
pip install Vizer==0.1.5

Documentation

Vizer

Boxes and masks visualization tools.

Install

pip install Vizer

Usage

img = Image.open(path)
# draw masks
img = draw_masks(img, data['masks'], data['labels'])
# draw boxes
img = draw_boxes(img, boxes=data['boxes'], labels=data['labels'], scores=data['scores'])
# show
plt.imshow(img)
plt.show()

Example

Bear