555终于跑通了 重新装了一个虚拟环境 有的坑参考前面一篇
https://blog.csdn.net/sueong/article/details/124254418?spm=1001.2014.3001.5501
detectron2官方Requirements
1.Linux or macOS with Python ≥ 3.6（python版本需大于3.6）
2.PyTorch ≥ 1.8 and torchvision that matches the PyTorch installation. Install them together at pytorch.org to make sure of
this（1.8版本以上去确保torchvision和pytorch相匹配）
3.OpenCV is optional but needed by demo and visualization（OpenCV可根据自己需要装）
ps：有的博客说pytorch1.8以下也可 这里最好是1.8的版本 因为后面会报错 要求PyTorch ≥ 1.8
总体安装参考官网https://github.com/facebookresearch/detectron2/blob/main/INSTALL.md
安装
新建一个conda环境并激活
conda create -n detectron python=3.7
conda activate detectron
参考官网：https://pytorch.org/get-started/locally/
配置CUDA和cudnn 配置pytorch
我的pytorch是1.8 cuda11.1
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
检查一下pytorch是否安装成功：
import torch # 如正常则静默
a = torch.Tensor([1.]) # 如正常则静默
a.cuda() # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda()) # 如正常则返回 "True"
exit()
detectron2安装
（1）安装detectron依赖
安装opencv库
pip install opencv-python
安装fvcore
pip install fvcore
安装cython
pip install cython
安装pycocotools：
pip install pycocotools
也有的博客用这种安装 但是会报错 建议用上一种
pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
（2）进入正题安装detectron2
官网说有两种安装方式 不推荐git的那种 会有各种坑和报错
建议用第二种
第二种安装方式如下
我的是torch1.8 cuda11.1 根据官网命令来装
python -m pip install detectron2 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.8/index.html
此时安装成功 会提示successfully install detectron2xxxxx
Detectron2测试
拉取代码
git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
保存图片input1.jpg在demo下
输出的路径为demo/result.jpg
命令路径改成绝对路径否则会报各种路径的错 input path was not found 或者xx doesn’t exist
python demo/demo.py \
--config-file /home/xxx/dete/detectron2/configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml \
--input /home/xxx/dete/detectron2/demo/input1.jpg --output /home/xxx/dete/detectron2/demo/result.jpg \
--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
执行结果
如不报错 则表示安装成功,即可开始使用detectron2。
