使用MMDetection进行目标检测

使用MMDetection进行目标检测

首先需要按照官方文档在本机的电脑上安装好MMDetection，不熟悉的小伙伴可以看我下面的这篇教程。​​​mmdetection的使用_dejahu的博客-​​ 安装好mmdetection后，对图片和视频进行目标检测非常简单，只需要3个文件即可

模型的配置文件模型文件需要进行推理的图片或者视频

详细的代码如下：

from mmdet.apis import init_detector, inference_detector, show_result_pyplotimport mmcvconfig_file = '../configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'# download the checkpoint from model zoo and put it in `checkpoints/`# url: = '../checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'# build the model from a config file and a checkpoint filemodel = init_detector(config_file, checkpoint_file, device='cuda:0')# test a single imageimg = 'demo.jpg'result = inference_detector(model, img)# show the resultsshow_result_pyplot(model, img, result)

可视化的效果如下，boundingbox左上方的文字分别表示类别和置信度。

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。