facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
Apache License 2.0
26.25k stars 5.45k forks source link

caffe2 converted model performance #1012

Open juvebogdan opened 3 years ago

juvebogdan commented 3 years ago

Hello,

I trained a model that works fine when I run it in pytorch. But caffe2 converted model makes mistakes on some example images. It just misses detecting some categories. When I run the same example in jupyter notebook using predictor all works fine.

How can I fix this?