您现在的位置是:首页 >技术教程 >无脑005----在mmyolo框架下复现RTMDet-R网站首页技术教程

无脑005----在mmyolo框架下复现RTMDet-R

吃肉不能购 2023-06-26 12:00:02
简介无脑005----在mmyolo框架下复现RTMDet-R

背景
win11系统wsl虚拟机Ubuntu下GTX1060显卡
cuda 10.1
pytorch 1.7.1
python 3.8
耗时 两个小时
参考github链接:https://github.com/open-mmlab/mmyolo

1.安装环境

首先安装pytorch

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

测试pytorch是否支持调用GPU:

python
>>> import torch
>>> torch.cuda.is_available()
True
>>> torch.cuda.get_device_name()
'NVIDIA GeForce GTX 1060'

然后利用mmlab研发的引擎安装:

pip install -U openmim

如果安装成功会显示:

Successfully installed 、、、、、、 openmim-0.3.7 、、、、、、

然后继续安装

mim install "mmengine==0.6.0"
mim install "mmcv==2.0.0rc4"
mim install "mmdet==3.0.0rc6"

注意:

1.mmcv

如果使用语句:
mim install “mmcv>=2.0.0rc4,<2.1.0”
会导致安装mmcv报错,显示没有cuda_home,
因为我的wsl输入nvcc --version会显示command not found
提示我:sudo apt install nvidia-cuda-toolkit
但是当我使用
mim install “mmcv==2.0.0rc4”
的时候,就没有问题,GPU可以正常使用。

2.mmdet

安装mmdet的时候,会报错没有pycocotools,这个时候你必须先安装GCC库,再装pycocotools,否则还会报错
语句是:

sudo apt-get install build-essential
pip install pycocotools

然后

git clone https://github.com/open-mmlab/mmyolo.git
cd mmyolo
pip install -r requirements/albu.txt
mim install -v -e .

最后显示

Successfully installed mmdet-3.0.0 mmengine-0.7.3 mmyolo-0.5.0 prettytable-3.7.0 shapely-2.0.1 terminaltables-3.1.10 wcwidth-0.2.6

但是还需要安装mmrotate

mim install "mmrotate==1.0.0rc1"

然后就可以使用了

2.inference测试

python demo/image_demo.py ./data2 configs/rtmdet/rtmdet_x_syncbn_fast_8xb32-300e_coco.py work_dirs/rtmdet_x_syncbn_fast_8xb32-300e_coco_20221231_100345-b85cd476.pth --out-dir ./output

效果还可以
在这里插入图片描述
测试的图片是没有旋转的,之后再测试一下旋转的,就是多了一个mmrotate的库,不能白装
2023年4月29日14点10分
睡觉

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。