Detectron2环境配置加测试Linux

首先 建一个干净的虚拟环境 (很重要)

conda create -n detectron2 python=3.7

然后进到这个环境  (detectroon2是这个环境的名字)

conda activate detectron2

然后 装pytorch   

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

(一定要注意自己的CUDA版本 我的是11.6所以用下面的代码)

然后 安装我们需要用的包  注意opencv的版本 最好是干净的环境 一套流程走下来 以免发生版本不兼容的问题

conda install cython opencv-python==4.2.0.34 matplotlib termcolor cloudpickle tabulate tensorboard tqdm yacs mock fvcore pydot wheel future

失败 conda list看一下 确实没下下来 

我们换pip

pip install cython opencv-python==4.2.0.34 matplotlib termcolor cloudpickle tabulate tensorboard tqdm yacs mock fvcore pydot wheel future

 成功!

然后我们要用coco数据集得下一个 pycocotools

直接git clone失败 用git lfs clone

一般大文件 我都会用git lfs clone 不过lfs是需要自己安装的 具体可以csdn找到

git lfs clone https://github.com/waleedka/coco

然后进到刚下载的包中

cd python setup.py build_ext install

这样pycocotools就安装好了

然后下载detectron2

git lfs clone https://github.com/facebookresearch/detectron2.git

 进到detectron2目录

python -m install -e detectron2  (失败!)官方的代码

python -m install -e .    (成功!) 注意这个点. 在e后面空格 .

然后进detectron2的github

去model zoo里下载对应的model的checkpoints

checkpoints上传到服务器之后 随便上传一张图片进行测试 我这里是用的 model_final_a54504.pkl

进到detectron2文件夹中后

python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml --input 3.jpg --opts MODEL.WEIGHTS model_final_a54504.pkl MODEL.DEVICE cpu

我这里实验室服务器满了 所以用的cpu 有gpu的话 把MODEL.DEVICE cpu去掉就好了

 结果如下

 可能碰到的问题总结一下

1.pytorch版本问题 这个一般很少出现 

2.git clone lfs的问题 我加了lfs一般都能行 

3.安装detectron2时 -e . 需要注意一下 not detectron2

相关文章

学习编程是顺着互联网的发展潮流,是一件好事。新手如何学习...
IT行业是什么工作做什么?IT行业的工作有:产品策划类、页面...
女生学Java好就业吗?女生适合学Java编程吗?目前有不少女生...
Can’t connect to local MySQL server through socket \'/v...
oracle基本命令 一、登录操作 1.管理员登录 # 管理员登录 ...
一、背景 因为项目中需要通北京网络,所以需要连vpn,但是服...