Ubuntu22.04+conda 安装Detectron2

1.虚拟环境

conda是可以创建虚拟环境的,如果已经有虚拟环境可以不用创建,如果为了互不干扰可以创建新的环境相关命令如下:

# 创建环境 
conda create -n det python=3.10
# 删除环境
conda remove -n det --all
# 激活环境
conda activate det

正常情况下创建环境和激活环境就可以了,如果安装过程中遇到了意外情况是可以删除环境,然后重新来过的

2.安装pytorch

pytoch的安装可以参照pytorch官网  Previous PyTorch Versions | PyTorch

具体的按照官网操作就可以了,我使用的是cuda11.7 加 pytorch1.12 具体的要根据自己的情况来选择

# CUDA 11.6
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

3. 安装fvcore

暂时不清楚这个是做什么用途,是按照知乎说明操作的

pip install fvcore

4.安装g++

安装过程中一直遇到一个错误,最后才发现是没有安装g++

错误如下:

 subprocess.CalledProcessError: Command '['which', 'g++']' returned non-zero exit status 1.
    [end of output]

安装命令:

sudo apt-get install build-essential

5.安装pycocotools

安装过程中总会遇到下载中断的情况,最终发现是因为下载源的问题,所以都需要指定下载源,具体如下

pip3 install pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple

6.安装detectron2

git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
python -m pip install -e detectron2

如果发现总是下载中断的情况需要指定下载源,具体如下:

python -m pip install -e detectron2 -i https://pypi.tuna.tsinghua.edu.cn/simple

7.验证

# 打开终端,确人是否在 detectron2 的文件夹路径下
python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input 1.jpg --output ./result.jpg --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

几个需要注意的问题:

1.如果遇到git不下来的情况,可以使用手工下载放到相应位置然后使用再终端操作

2.经常更新一下源 sudo apt-get update

3.pip install -e的开始位置为当前位置,如果在detectron2路径下安装的话是不能使用-e的,也就是说需要在detectron2的上一级目录才可以。即不能执行上面的 cd detectron2

猜你喜欢

转载自blog.csdn.net/matlab001/article/details/127682442