How to draw a detection box and display confidence on the test data

chongruo / detectron2-ResNeSt

A fork of Detectron2 with ResNeSt backbone

https://arxiv.org/abs/2004.08955

Apache License 2.0

385 stars 73 forks source link

How to draw a detection box and display confidence on the test data #23

Closed dongzhenguo2016 closed 4 years ago

dongzhenguo2016 commented 4 years ago

Now I can run through the training program. Not only do I want to calculate AP and other indicators on the test data set, How do I draw a detection box and display confidence on the test data? thanks a lot

chongruo commented 4 years ago

please see https://github.com/facebookresearch/detectron2/blob/515c53ad645275dceec15c785c0eb0005bb95b63/detectron2/utils/visualizer.py#L308