【Faster RCNN & detectron2】detectron2实现Faster RCNN目标检测

逃离我推掉我的手 2022-09-03 05:24 339阅读 0赞

### 目录 ###

*  [1. 背景介绍][1.]
 *  [2.安装步骤][2.]
 *  [3.Faster RCNN目标检测][3.Faster RCNN]
 *  [4. 效果][4.]
 *  [5.错误解决][5.]
 *  [6.参考博客][6.]
 *  [7，下一节代码解析][7]

在上一篇博客记录了 [SlowFast的复现][SlowFast]过程，slowfast其中有一部分是detectron2实现Faster RCNN对人的目标检测。

这一篇博客就单独记录detectron2实现Faster RCNN目标检测的解析

[B站视频讲解][B]

# 1. 背景介绍 #

*  detectron2[项目地址][Link 1]
 *  detectron2[文档][Link 2]

# 2.安装步骤 #

安装：

pip install -U torch torchvision cython
    pip install -U 'git+https://github.com/facebookresearch/fvcore.git' 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
    git clone https://github.com/facebookresearch/detectron2 detectron2_repo
    pip install -e detectron2_repo

# 3.Faster RCNN目标检测 #

在终端输入：

python3 demo.py --config-file ../configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml \
      --input ../img/1.JPG \
      --output ../img/1_1.jpg \
      --opts MODEL.WEIGHTS detectron2://COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl

# 4. 效果 #

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1doaWZmZVlG_size_16_color_FFFFFF_t_70]

# 5.错误解决 #

运行过程中可能会出现

Traceback (most recent call last):  
File “demo.py”, line 7, in  
import cv2  
File “/opt/conda/lib/python3.7/site-packages/cv2/**init**.py”, line 5, in  
from .cv2 import \*  
ImportError: libGL.so.1: cannot open shared object file: No such file or directory

解决方案：

在终端运行：

apt update
    apt install libgl1-mesa-glx

# 6.参考博客 #

[Ubuntu18.04 配置detectron2实现Faster RCNN目标检测和Mask RCNN实例分割(一)][Ubuntu18.04 _detectron2_Faster RCNN_Mask RCNN]

# 7，下一节代码解析 #

[【代码解析】显示detectron2中faster R-CNN的检测结果，利用visualizer显示][detectron2_faster R-CNN_visualizer]

[1.]: https://blog.csdn.net/WhiffeYF/article/details/113813923?spm=1001.2014.3001.5501#1__7
[2.]: https://blog.csdn.net/WhiffeYF/article/details/113813923?spm=1001.2014.3001.5501#2_12
[3.Faster RCNN]: https://blog.csdn.net/WhiffeYF/article/details/113813923?spm=1001.2014.3001.5501#3Faster_RCNN_23
[4.]: https://blog.csdn.net/WhiffeYF/article/details/113813923?spm=1001.2014.3001.5501#4__33
[5.]: https://blog.csdn.net/WhiffeYF/article/details/113813923?spm=1001.2014.3001.5501#5_36
[6.]: https://blog.csdn.net/WhiffeYF/article/details/113813923?spm=1001.2014.3001.5501#6_55
[7]: https://blog.csdn.net/WhiffeYF/article/details/113813923?spm=1001.2014.3001.5501#7_58
[SlowFast]: https://blog.csdn.net/WhiffeYF/article/details/113527759
[B]: https://www.bilibili.com/video/BV1Qi4y1K7yU/
[Link 1]: https://github.com/facebookresearch/detectron2
[Link 2]: https://detectron2.readthedocs.io/en/latest/index.html
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1doaWZmZVlG_size_16_color_FFFFFF_t_70]: /images/20220829/762be3195cd24382a5e02bcc6b86b9f4.png
[Ubuntu18.04 _detectron2_Faster RCNN_Mask RCNN]: https://blog.csdn.net/linghu8812/article/details/105701408/
[detectron2_faster R-CNN_visualizer]: https://blog.csdn.net/WhiffeYF/article/details/114291154