search

hikvision-research / opera

A Unified Toolbox for Object Perception & Application
Apache License 2.0
139 stars 12 forks source link

How to visualize results of PETR #19

Open STRUGGLE1999 opened 1 year ago

STRUGGLE1999 commented 1 year ago

Hello, I want to know how to get the visualization results of PETR. like Figure 5 in the paper.

marsgravity commented 1 year ago

We have provided image_demo to infer your own image, and you can use it like python /opera/demo/image_demo.py ${img_path} /opera/config/petr/petr_r50_16x2_100e_coco.py petr_r50_16x2_100e_coco.pth --out-file ${new_path}