适用于yolov5单类检测的coco数据集制作及训练

最新推荐文章于 2024-12-04 11:58:36 发布
最新推荐文章于 2024-12-04 11:58:36 发布
阅读量7.1k 收藏 34
点赞数 8
分类专栏： cv 目标检测 机器学习 文章标签： python 人工智能 深度学习 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_52415611/article/details/113030688
版权
这篇博客介绍了如何制作适用于yolov5的单类（人）COCO128数据集，并详细阐述了训练过程。首先从COCO128数据集中筛选出含人的图片，然后使用yolov5框架进行训练，调整配置文件并指定训练参数。训练完成后，提供了不同epochs的权重供下载和使用。
摘要由CSDN通过智能技术生成

一、制作单类检测的coco128数据集（以人为例）

1、首先下载coco128数据集，解压后可看到该数据集文件存放位置适合使用yolov5进行训练。coco数据集可在下面链接下载。
https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip
在这里插入图片描述
2、假定我们要制作只检测人的数据集，打开一张图片及其标签。（标签在pycharm中打开，比较明显，直接用文本打开时数据没有分行，看起来很杂乱。）
在这里插入图片描述
在这里插入图片描述
标签中包含五个参量。第一个参量代表类别，其余参量分别为x，y，w，h；对应在含义可以通过上图理解。

通常制作数据集的方法是使用LabelImg进行标注，这里选择直接对labels中的数据进行更改，只保留需要的类以及图片。是一种讨巧的方法！

最低0.47元/天 解锁文章
Pytorch Erroe:invalid argument 0: Sizes of tensors must match except in dimension 0. Got 62 and 56 i
Anthony's Blog
03-20 445
Error invalid argument 0: Sizes of tensors must match except in dimension 0. Got 62 and 56 in dimension 2 at C:\Users\builder\AppData\Local\Temp\pip-req-build-9msmi1s9\aten\src\TH/generic/THTensor.cpp...
用MATLAB如何制作自己的数据集,Windows10+YOLOv3实现检测自己的数据集（1）——制作自己的数据集...
weixin_39774905的博客
03-24 1617
一、数据标注在深度学习的目标检测任务中，首先要使用训练集进行模型训练。训练的数据集好坏决定了任务的上限。下面介绍两种常用的图像目标检测标注工具：Labelme和LabelImg。(1)LabelmeLabelme适用于图像分割任务和目标检测任务的数据集制作，它来自该项目：https://github.com/wkentaro/labelme。按照项目中的教程安装完毕后，应用界面如下图所示：它能够...
【保姆级教程】从YOLOv8目标检测coco128数据集和自己的数据集理解模型训练、验证与预测（pytorch+pychrm）
最新发布
qq_62844901的博客
12-04 2269
本文详细介绍了使用yolov8对coco128数据集进行环境准备，训练、验证、预测的全过程，也以coco128为例，认识了数据集目录和yaml配置文件的过程，也介绍了如何通过输出结果判断训练模型的效果，以供大家学习。
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 224 and 266
agonysome的博客
10-28 1000
查到了两个解决方案，其一如下，经过查验不是这个原因造成了我的错误： 原链接：https://www.cnblogs.com/zxj9487/p/11531888.html 解决：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 544 and 1935 in dimension 2 at ../aten/src/TH/generic/THTensor.cpp:711 这种错误
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 4 in
sunflower_sara的机器学习园地
08-23 1万+
报错如下： RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 4 in dimension 1 at /pytorch/aten/src/TH/generic/THTensorMath.cpp:3616 因为读取图像的时候三通道图像中混入了灰度或...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.
weixin_43722561的博客
05-22 772
解决报错：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 756 and 1008 in dimension 2原因：输入的图像数据的维度不完全一样(数据集图像大小不一)。解决方法： transforms.Compose([ transforms.RandomResizedCrop(224),# 随机裁剪]) 原因：输入的图像数据的
基于YOLOv9实现水下垃圾识别检测系统python源码+详细运行教程+训练好的模型+评估指标曲线.zip
08-27
基于YOLOv9实现水下垃圾识别检测系统python源码+详细运行教程+训练好的模型+评估指标曲线.zi 【使用教程】 一、环境配置 1、建议下载anaconda和pycharm 在anaconda中配置好环境，然后直接导入到pycharm中，在pycharm...
YOLOv5数据集制作与训练教程：井盖检测
"YOLOv5制作数据集与训练数据集为检测模型教程，通过labelimg进行图像标注，包括正常井盖和异常井盖两类，共283张图片。利用labelimg软件进行标注，保存为Yolo格式的txt文件。接着使用YOLOv5的train.py进行模型训练...
基于YOLOv9道路红绿灯行人车辆(8类)识别检测系统python源码+详细运行教程+训练好的模型+数据集+评估指标曲线.zip
10-10
需要准备yolo格式的目标检测数据集，如果不清楚yolo数据集格式，或者有其他数据训练需求，请看博主yolo格式各种数据集集合链接：https://blog.csdn.net/DeepLearning_/article/details/127276492 里面涵盖了上百种...
目标检测时出现RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.
Make AI，Not War
08-21 2546
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.出错代码case1：每个图象的目标个数不尽相同，原始参数适用case2：图象大小不一样case3：样本数不能整除batch_size（据说） 造成这个bug的原因有很多，一般都是在用pytorch读取数据时产生的，下面是几种常见的情况。 出错代码 dataG = ObjectDataset(image_ids, DIR_TRAIN+'/ima
pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 a
热门推荐
公众号/知乎: AI算法札记，欢迎关注交流
11-28 3万+
1.pytorch报错 Namespace(batchSize=16, gpu='0,1', lr=0.001, nepoch=210, num_workers=8) Epoch: 0 Epoch:0 [0|173089] loss:3.747539 Epoch:0 [1|173089] loss:3.629151 Epoch:0 [2|173089] loss:3.731266 Traceba...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 51 and 50 i
longshaonihaoa的博客
03-19 1447
我是因为加载数据时，bs>1，但是数据图像尺寸不同，所以报错，将bs=1即可
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 306 and 300
鱼落池中
01-04 1204
错题提示：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 306 and 300 in dimension 3 at /pytorch/aten/src/TH/generic/THTensor.cpp:689 经过排查后发现是torchvision.transoform...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 179 and 182
xingtianyao的博客
04-08 433
用pytorch Dataloader加载数据的时候报错如上， 原因是数据集未resize到统一大小，将训练的数据处理成统一大小就可以了。 mark。
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 544 and 193
weixin_43911343的博客
07-26 791
解决方案汇总：（持续更新） 参考了多位博主的文章，仅作学习用，如有侵权请联系我删除。 一、原因 二、解决方法 1.修改dataloader源码： 链接：here 2.修改padding使图片size相同 在maskrcnn_benchmark里，image_list类，后面找到源码再来更新。 ...
pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0
weixin_42630613的博客
05-28 353
数据加载错误导致 可能像https://blog.csdn.net/weixin_41278720/article/details/84586734一样的问题，但是这里没问题 我又查看了我的transform，由于没有resize导致报错 train_transform = transforms.Compose([ #transforms.Resize((32, 32)), transforms.RandomCrop(32, padding=4), # 将图像转化为32 * 32
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0—问题分析
专注于AI领域前沿技术学习与分享：目标检测、图像修复、超分重建、AI工程化
01-11 5491
墨理三生 感谢本次通过私信交流，协力探讨解决问题的同学 UP ：https://blog.csdn.net/weixin_44559592 探讨的 pytorch 报错信息如下： RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0 分析： Sizes of tensors dimension 说明 错误的发生是因为 tensors 的维度 不一致； 我的 第一反应是检查数据是不.
YoloV5训练测试用coco数据集标签文件发布
资源摘要信息:"本资源提供了针对YoloV5模型训练和测试所必需的coco数据集标注文件，包括两个主要部分：训练数据集(train2017.txt)和验证数据集(val2017.txt)。为了与YoloV5模型的输入要求相匹配，数据集的原始图像和...
评论 10
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值