【目标检测系列】yolov5训练自己的数据集(pytorch版)

0.有些东西是前面yolov3训练时候就说过的会在文中注明:

1.下载代码;

https://github.com/ultralytics/yolov5

2.准备数据集:

        数据格式和yolov3一模一样。如果做过yolov3的数据集可以直接搬来用就行。详细见https://blog.csdn.net/gbz3300255/article/details/106276897 的 第三节 -----------训练自己数据集的步骤。

准备文本文件有: train.txt    test.txt   val.txt   lables的文本文件

train.txt,记录数据集下图片名字,类似这样,数据集图片存放在/data/images/目录下。

BloodImage_00091.jpg
BloodImage_00156.jpg
BloodImage_00389.jpg
BloodImage_00030.jpg
BloodImage_00124.jpg
BloodImage_00278.jpg
BloodImage_00261.jpg

test.txt,与面形式一样,内容是需要测试的图的文件名

BloodImage_00258.jpg
BloodImage_00320.jpg
BloodImage_00120.jpg

val.txt,与面形式一样

  • 0
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值