在SeaShip数据集上验证YoloV5
操作系统: ArchLinux
需要代理
1.conda安装配置
已经完成的小伙伴可以直接跳过
安装Anaconda
yay安装
yay -S anaconda
激活Anaconda环境
source /opt/anaconda/bin/activate root
关闭Anaconda环境
source /opt/anaconda/bin/deactivate root
Anaconda 常用命令
管理Conda
检测Conda版本
conda --veriosn
升级当前版本的conda
conda update conda
管理环境
查看环境目录
conda env list
创建环境
conda create -n (环境名) (包名)
打开环境
conda activate (环境名)
或
source activate (环境名)
退出环境
source deactivate
卸载环境
conda remove --name (环境名) --all
将环境导入YAML文件当中
分享代码时需要将自己的环境也同时分享,故应该将package的环境导入到environment.yaml中
#导出conda环境,默认路径在主文件夹下
conda env export > environment.yaml
#通过yaml信息创建环境
conda env create -f emvironment.yaml
管理包
查看包
conda list
更新包
conda upgrade --all
搜索包
conda search name
安装包
conda install (包名)
or
pip install (包名)
卸载包
conda remove (包名)
or
pip uninstall (包名)
2.下载数据集
下载数据集
#下载地址
http://www.lmars.whu.edu.cn/prof_web/shaozhenfeng/datasets/SeaShips%287000%29.zip
下载好的数据集有这样三个文件,Annotations是保留xml文件,JPEGImages是图像文件,ImageSets下有Main文件夹中含有四个文本文件。
3.环境配置以及源码下载
conda配置
conda creat -n yolov5 python=3.8
下载yolov5源码
git clone https://github.com/ultralytics/yolov5.git
cd yolov5
pip install -r requirements.txt
成功安装界面
4.数据集格式转化
cd data
将之前下载的数据集当中的Annotations与JPEGImage移动到该目录下
在同级目录下创建makedata.py,并运行: (PS:记得改成自己的绝对路径,相对路径容易没有经验的同学容易报错还不知道怎么改)
import xml.etree.ElementTree as ET
import os
import shutil
import random
xml_file_path = '/home/自己的绝对路径/data/Annotations/' # *********检查和自己的xml文件夹名称是否一致*********
images_file_path = '/home/自己的绝对路径/data/JPEGImages/' # *******检查和自己的图像文件夹名称是否一致********
# 最好不要用"images"作文件夹名字,因为后续会对图片进行划分,我命名的新文件夹叫"images"
# ******改成自己的类别名称***********
classes = ["ore carrier", "general cargo ship", "bulk cargo carrier", "container ship", "fishing boat", "passenger ship"]
# 数据集划分比例,训练集75%,验证集15%,测试集15%
train_percent = 0.7
val_percent = 0.15
test_percent = 0.15
# 此处不要改动,只是创一个临时文件夹
if not os.path.exists('data/temp_labels/'):
os.makedirs('data/temp_labels/')
txt_file_path = 'data/temp_labels/'
def convert(size, box):
dw = 1. / size[0]
dh = 1. / size[1]
x = (box[0]