facebookresearch / facebookresearch/detectron2
Hamer bounding box detection very slow
Open
- Dominant language
- Python
- Stars
- 34.7k
- Forks
- 7.9k
- PR merge metrics
- No merged PRs in 30d
Description
Hi,
I am using the detectron2 cascade mask rcnn vitdet h checkpoint in the demo for hamer and it is very slow for inference on one image compared to detectron2 model zoo numbers. I get .78s/im vs the model zoo ~.2s/im
This is the line I timed: https://github.com/geopavlakos/hamer/blob/dc19e5686198a7c3fc3938bff3951f238a85fd11/demo.py#L81
I have an RTX A5000
Is there a setting in the config file that can speed things up?
Contributor guide
Assessment
This issue has not been assessed yet.