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出现问题
在这里插入图片描述

测试结果:
在这里插入图片描述
在这里插入图片描述