Seaship数据集+yolov5

在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]