基于MMDetection检测目标
感谢大佬 何小有 的文章 [https://www.jianshu.com/p/ae72d7e937ff]
接口实现参考 gen_target.py
代码粗陋请忽略
安装pytorch : pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
下载模型到 checkpoints/scnet 目录下: wget https://download.openmmlab.com/mmdetection/v2.0/scnet/scnet_r50_fpn_1x_coco/scnet_r50_fpn_1x_coco-c3f09857.pth