junjiehe96 / FastInst

[CVPR2023] FastInst: A Simple Query-Based Model for Real-Time Instance Segmentation
MIT License
178 stars 16 forks source link

About Visualizing the results #3

Closed GTrui6 closed 1 year ago

GTrui6 commented 1 year ago

Hi, thanks for your wonderful work, i have trained FastInst on customized datasets, and now i want to visualize the prediction results, so i used the demo.py, but i failed to visualize it. Is there an another way i can try to visualize the results?

junjiehe96 commented 1 year ago

You can run

python demo/demo.py \
    --config-file configs/coco/instance-segmentation/fastinst_R50_ppm-fpn_x1_576.yaml \
    --input xxx.jpg \
    --output result.jpg \
    --opts MODEL.WEIGHTS /path/to/fastinst_R50_ppm-fpn_x1_576_34.9.pth

It will visualize top 100 scored predictions.

GTrui6 commented 1 year ago

Appreciated your help, i have visualized the results.