linux-conda安装detectron2+测试demo

新建conda环境

conda create -n det python=3.6
conda activate det

安装pytorch

pytorch官网
选择自己安装的版本
linux-conda安装detectron2+测试demo_第1张图片复制命令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装openCV(可选,运行demo要用到)

conda install -c menpo opencv3

安装fvcore(可选)

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

安装pycocotools(可选)

pip install cython;
pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

安装detectron2

github官方页面
官网给出三种方式,我使用的是第二种

python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
# (add --user if you don't have permission)

# Or, to install it from a local clone:
git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2

# On macOS, you may need to prepend the above commands with a few environment variables:
CC=clang CXX=clang++ ARCHFLAGS="-arch x86_64" python -m pip install ...

激动人心的提示语
在这里插入图片描述

测试

进入detectron2/demo目录下
新建images文件夹存放测试图片001.jpg(自己决定图片)
新建results文件夹,结果会存放在这里
model我没有下载文件,直接链接到了网上的模型,毕竟这只是测试一下环境能否运行,之后的模型都是自己训练的

运行命令
python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml 
--input images/001.jpg 
--output results 
--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

这时,我的机器报错了
在这里插入图片描述看了多篇博客,最后的解决方法是

sudo apt-get update
sudo apt-get install libgtk2.0-0

但是又报错sudo,还没搞清楚原因,不过上面的命令去掉sudo运行也ok
再次运行测试demo命令,终于成功,放一张成果图
linux-conda安装detectron2+测试demo_第2张图片整个安装过程一个bug接着一个,从头进行了3次,终于成功了,希望自己之后的学习过程就算不是一帆风顺,也能化险为夷,绝处逢生!

你可能感兴趣的:(机器学习,深度学习,机器学习,python)