目标检测框架Detectron2 试玩

我的电脑环境：ubuntu18.04.5 GTX 1060 6GB cuda10.1 （nvcc -V 和  nvidia-smi 要匹配）

python3.6.6 

torch1.6

torchvison 0.2.1

1. 安装

一些依赖安装：

sudo pip install 'git+https://github.com/facebookresearch/fvcore'

sudo pip install opencv-python 

sudo pip install matplotlib

sudo pip install opencv-python

sudo pip install cloudpickle

sudo pip install matplotlib

等等。。。其它一些依赖看自己电脑环境配置

1.1 源码安装： 推荐

git clone git@github.com:facebookresearch/detectron2.git

cd detectron2

python setup.py build develop # 这build 不知道是不是编译此项目的意思

1.2 pip安装

python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' #权限不足时 python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' --user

2. demo

demo01

图像分割：model ：COCO-PanopticSegmentation/panoptic_fpn_R_101_3x/139514519/model_final_cafdb1.pkl

对应config-file ：/home/yao/detectron2-master/configs/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml

cd detectron2-master/demo：

python3 demo.py  
    --config-file /home/yao/detectron2-master/configs/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml
    --input /home/yao/detectron2-master/demo/test_imgs/1.jpg # 这里可以用视屏
    --output /home/yao/detectron2-master/demo/outputs
    --opts MODEL.WEIGHTS /home/yao/detectron2-master/configs/model_pkl/139514519/model_final_cafdb1.pkl

demo02

视屏分割：

cd detectron2-master/demo：

python3 demo.py  

    --config-file /home/yao/detectron2-master/configs/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml
    --video-input /home/yao/detectron2-master/demo/test_imgs/robots.mp4
    --confidence-threshold 0.6
    --output /home/yao/detectron2-master/demo/outputs/robots_out.mp4
    --opts MODEL.WEIGHTS /home/yao/detectron2-master/configs/model_pkl/139514519/model_final_cafdb1.pkl

可能遇到的bug：

1. 安装过程中出现：ImportError: cannot import name '_C'+ 

可能是编译的问题，第一次我使用 python -m pip install -e .  安装遇到的， 后来改为python setup.py build develop 好像就好了。

这个问题网上的回答不多好像，我没找到好的解决。

2. socket.timeout: The read operation timed out

使用 pip --default-timeout=100 install 。。。

3. error: command '/usr/bin/nvcc' failed with exit status 1...  这个问题挺头疼的。

我的解决是：

先查一下nvcc 是不是在报错的路径下，我的不是。我的是在/usr/local/cuda-10.1/bin 下

添加环境变量：

export PATH="/usr/local/cuda-10.1/bin:$PATH"

export LD_LIBRARY_PATH="/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH"

source ~/.bashrc

最后

以上就是幸福唇彩为你收集整理的目标检测框架Detectron2 试玩的全部内容，希望文章能够帮你解决目标检测框架Detectron2 试玩所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。