当前位置:   article > 正文

彩笔的第一个模型:DOTA数据集YOLOv5目标检测_dota数据集用什么yalm

dota数据集用什么yalm

目录

前言

一、下载数据集DOTA和源码,下载YOLOv5模型,切割图片并转化label为YOLO格式

2、DOTA数据集的label格式不能在YOLO模型 里直接使用,需要转化

3、需要做成coco数据集的文件结构,包括文件夹名称。

二、修改YOLO参数,跑

1、要运行生成图片路径的脚本:utf-8.py

2.修改coco.yaml和yolov5s.yaml

3、修改train.py里的参数

总结


前言

   这是我跑通的第一个真正意义上的网络模型,写个博客总结/纪念一下,模型用的yolov5s,数据集是遥感卫星图像DOTA数据集。总体步骤分为以下几步:

1、数据集下载,下载YOLOv5模型,切割图片,转化labels格式,按照要求的文件结构放在服务器里

2、修改YOLO参数,跑

ps:中间出了无数问题,有的是我个人脑残能力不够,有的是莫名其妙的问题。见下文

 需要库的安装我不赘述了,在requirement.txt文件里

一、下载数据集DOTA和源码,下载YOLOv5模型,切割图片并转化label为YOLO格式

数据集链接在这里  DOTA

YOLO的源码是这个https://www.wpsshop.cn/w/AllinToyou/article/detail/679151

推荐阅读
相关标签
  

闽ICP备14008679号