mmdetection的安装、测试(以Faster-rcnn为例进行训练测试)
一、如何创建虚拟环境
看看有几个cpu: watch --color -n1 gpustat -cpu
创建自己的环境:conda create -n jj python=3.7
失败:出现下面的
-30-generic ubuntu/14.04.2 glibc/2.19
UID:GID : 1062:1063
netrc file : None
offline mode : False
An unexpected error has occurred. Conda has prepared the above report.
解决办法:
conda clean -i
重新进行创建
查询是否创建成功:conda info --envs
运行自己环境:source activate jj
退出虚拟环境:conda deactivate
建立一个文件夹:make ljj
进入文件夹:cd ljj
下载darknet: git clone https://github.com/pjreddie/darknet.git
进入darknet:cd darknet/
改makefile
运行:make
安装opencv:pip install opencv-python==4.1.0.25
conda install cudatookit=10.1
下载:wget https://pjreddie.com/media/files/yolov3.weights
运行:./darknet detect cfg/yolov3.cfg cfg/yolov3.weights data/dog.jpg
出现错误:按照官网yolo的步骤./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg出现
“Loading weights from yolov3.weights…Couldn’t open file: yolov3.weights”
解决:更改代码
./darknet detect cfg/yolov3.cfg cfg/y olov3.weights data/dog.jpg
二、安装mmdection
Linux下查看cuda版本号:cat /usr/local/cuda/version.txt
安装pytorch:大佬教程
测试是否成功:输入python
import torch
torch.cuda.is_available()
出现true即成功。
import torchvision
torchvision.__version__
'0.6.1+cu101'
import torch
torch.__version__
'1.5.1+cu101'
大佬方法:安装mmection
注意:在按照大佬教程中出现了错误,第三步安装mmcv
改为官方教程中命令
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
eg: pip install mmcv-full==1.2.2 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html
下载预训练模型:

测试代码根据自己情况,
python demo/image_demo.py demo/demo.jpg configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth
可能会出现mmcv版本过低问题,卸载之后再安装高版本的。

三、训练mmdection
以faster-rcnn为例,VOC2007数据集
跟着大佬教程一步一步来
VOC数据集下载,数据集介绍
图片大小不一样,要进行resize
一般用到voc.py和vVOCOperationLibrary.py,更改voc.py里的内容,运行!
修改参数:跟着教程
训练代码:
python ./tools/train.py ./configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py
训练时出现问题:
1.utf-8

原因:代码中有中文,即使是在注释中
2.
改batch-size
在mmdetection/configs/base/datasets/voc0712.py里

都改成1了???
训练结果:













四、测试mmdection
测试代码:测试单张图片
python demo/image_demo.py data/VOCdevkit/VOC2007/JPEGImages/000198.png configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py work_dirs/faster_rcnn_r50_fpn_1x_coco/latest.pth
问题:用test.py出现问题

测试结果:

