win10系统下detectron2的环境配置
win10系统下detectron2的配置过程提示:这里可以添加系列文章的所有文章的目录,目录需要自己手动添加第一章detectron2的环境配置第二章detectron2配置过程遇到的问题第三章detectron2在qgis的环境中配置及遇到的问题第四章detectron2训练自己的数据集第四章detectron2的模型转化为caffe2模型第五章caffe2模型的使用提示:写完文章后,目录可以
win10系统下detectron2的配置过程
目录
一、anaconda的虚拟环境配置
新建一个虚拟环境,通过anaconda的相关使用,学习和新建一个虚拟环境(detectron2)来配置detectron2的环境。
二、NVIDIA显卡驱动
这里采用CUDA11.1(其他旧版本驱动未测试)。
因此如果目前cuda的版本不是10.1或10.1以上,就卸载掉,然后安装cuda11.1。
cuda11.1完成以后,新增系统环境变量:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
三、安装VS2019
detectron2需要vs2019的环境,安装VS2019社区版的链接:https://visualstudio.microsoft.com/zh-hans/vs/
如果电脑上有多个vs版本,我们需要设置一下命令:
SET MSSdk=1
SET DISTUTILS_USE_SDK=1
call “C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Auxiliary\Build\vcvarsall.bat” amd64 -vcvars_ver=19.27
每个人的vcvarsall.bat路径和版本不一样,找到自己的路径和版本,然后替换掉上面的。
C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\
四、安装cocoapi
下载链接:https://download.csdn.net/download/q5c521/20658250?spm=1001.2014.3001.5501
参考以下命令行安装:
cd coco/PythonAPI
# install pycocotools locally
python setup.py build_ext --inplace
# install pycocotools to the Python site-packages
python setup.py build_ext install
检验是否安装成功的命令:如果没有报错,就是成功
import pycocotools;
from pycocotools.coco import COCO
五、安装fvcore
下载链接:https://download.csdn.net/download/q5c521/20658250?spm=1001.2014.3001.5501
参考以下命令行安装:setup.py在fvcore文件夹下
cd fvcore
python setup.py build --force develop
检验是否安装成功的命令:如果没有报错,就是成功
import fvcore
六、安装detectorn2
detectron2的代码下载链接:https://github.com/facebookresearch/detectron2
下载的时候我们可以选择detectron2的版本,同时还可以查看各个版本的优点。
安装命令:
python setup.py build develop
七、detectorn2测试
gpu模型下:
python demo/demo.py \
--config-file /home/**/project/detectron2/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \
--input ceshi/test.jpg --output outputs/ \
--opts MODEL.WEIGHTS '/home/**/project/detectron2/pre_train_model/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl'
cpu模式下:
python demo/demo.py \
--config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \
--input img/test1.jpg --output outputs/ \
--opts MODEL.WEIGHTS '/home/project/detectron2/pre_train_model/model_final_f10217.pkl' \
MODEL.DEVICE cpu

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。
更多推荐
所有评论(0)