如何在ubuntu下安装detectron2_detectron2 + ubuntu + cpu

本文档详细介绍了在Ubuntu系统中,如何在CPU环境下安装Detectron2。内容包括创建conda环境,安装PyTorch、OpenCV等依赖,解决清华conda源问题,从GitHub克隆COCOAPI,编译并安装Detectron2,以及运行预训练模型的Demo。文章还提到了可能遇到的问题及解决方案,如模型权重指定、设备设置为CPU等。
摘要由CSDN通过智能技术生成

一开始觉得找到的资料都是关于GPU下配置的,可能CPU会有很多坑

结果并没有,就清华conda源坑了我一下

#创建conda环境

#20.3.16. conda清华源怎么也弄不出来,好像是main还是free哪个包出问题了

#一开始以为是配了ss,ip有问题,ping也ping不通,学到了,ping域名不能加http这些

#换了中科大源解决了,底下安装pytorch还是可以用清华的镜像

conda create -n detectron2 python=3.7

#安装pytorch, cpu

#用清华源记得删掉 -c pytorch

#速度就是一切

pip install opencv-python(清华源

pip install cython(清华源

#这里也要下很多包,你看到哪个包下的慢,就crtl+c,然后用pip或者conda结合国内源下载

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

# 编译,同上,下包慢就手动来

#网上看到有的博客里编译的语句是啥啥setup.py,然后还要手动下载fvcore,3.16这个版本已经改变了,就按照github最新的命令来

python -m pip install -e .

#接下来看一下人家给出的demo,我一句一句理一下

Inference Demo with Pre-trained Models

#选择model zoo里面的一个模型,实际上也都保存在configs里面了吧

#不知道啊,.yaml文件是干嘛的,pkl文件可以是模型也可以是模型参数

#去model_zoo里面找到mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl这个文件,下载到本地,可以新建个什么文件夹叫model_zoo之类的,文件夹结构可以像他提供的这样,但是如果你下载的不是特别多,也可以用一个文件夹

#不知道r_50_fpn_3x到底是啥意思,反正都和训练相关吧,我就先用着了

Pick a model and its config file from model zoo, for example, mask_rcnn_R_50_FPN_3x.yaml.

We provide demo.py that is able to run builtin standard models. Run it with:

cd demo/

#--input input1.jpg input2.jpg 制定了输入,可以自己找一张input1.jpg放在demo文件夹下。

# [--other-options] 别的选项的占位,没有就把这句删了

# --opts MODEL.WEIGHTS 你存储model_final_f10217.pkl的路径 MODEL.DEVICE cpu

# 这一句指明了model weights的地址,你用原来的路径是会去网上下载的,model.device cpu,一定不能少,我们就cpu。

python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml

--input input1.jpg input2.jpg

[--other-options]

--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

The configs are made for training, therefore we need to specify MODEL.WEIGHTS to a model from model zoo for evaluation. This command will run the inference and show visualizations in an OpenCV window.

For details of the command line arguments, see demo.py -h or look at its source code to understand its behavior. Some common arguments are:

To run on your webcam, replace --input files with --webcam.

To run on a video, replace --input files with --video-input video.mp4.

To run on cpu, add MODEL.DEVICE cpu after --opts.

To save outputs to a directory (for images) or a file (for webcam or video), use --output.

b678cfdc3a37bcf6fcf8cc216e6f1a58.png

#demo运行后会有这种输出,还有控制界面啥的,无所谓了,我奇怪的一点是,怎么都没办法退出运行,就一直卡在那里,ctrl+c也没有反应,奇怪。

下一篇熟悉一下detectron2的API,因为我只需要使用它训练好的模型,所以关于训练的部分不会涉及

我写的应该都是关于推理部分的代码

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值