使用环境: ubuntu device: Tesla T4
docker: nvcr.io/nvidia/pytorch:20.03-py3
repo: https://gitlab.com/StrangeAI/centernet
复现步骤:
- 参考readme进行pip install等配置工作
- 从https://github.com/xingyizhou/CenterNet/blob/master/readme/MODEL_ZOO.md 上下载权重,使用的是’ctdet_coco_resdcn101‘
- python demo_det_r101.py
- 将第三步的结果使用opencv保存到文件, cv2.imwrite("test.jpg", res) #插入文件 #line151 后
- python demo_det_dla.py 可以检测出图像。