Slow inference (segmentation task) #5233
Unanswered
grudzinskap
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Hey all, I have used detectron2 with RCNN R50-FPN model for one-class segmentation task. I have a problem with slow inference - I get 0.2s/image on gpu, but the documentation says 0.043s/image or even less is possible. My images are big (about 8000x2000), but there is resize in predict function. I tried to add some power to my gpu, but results are the same. I also changed predict function to work on image as tensor instead of numpy and then pytorch conversion and it has helped, but I still can't get results like from documentation. Could you give me some advice on that? Maybe there are some parameters in config I should change? Now I have default settings.
Beta Was this translation helpful? Give feedback.
All reactions