如何用yolov5训练自己的图片

首先配置yolov5环境，详细看这篇https://blog.csdn.net/qq_16792139/article/details/

首先采用脚本工具转换好xml为txt，脚本工具上传至群文件了 Q群477592543， yolo5totxt.py

配置官方文档查看github源地址https://github.com/ultralytics/yolov5/wiki/Train-Custom-Dat

数据集格式

记得修改yaml里面的类名和类数目，还有train.py里面的80替换成你的类数

然后运行

python train.py --img 640 --batch 16 --epochs 5 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ''

就可以了。

然而，报错

cuda内存不足，好吧有小伙伴知道怎么解决吗

如何用yolov5训练自己的图片相关推荐

Yolov5训练日记~如何用Yolov5训练识别自己想要的模型~
目录一.数据集准备二.标签设置三.模型训练四.模型测试最近尝试了Yolov5训练识别人体,用的是自己尝试做的训练集.见识到Yolo的强大后,决定分享给大家. 一.数据集准备数据集是从百度图 ...
【YoloV5】YoloV5训练图片增强 python代码
YoloV5训练图片增强 python代码 Generate More Label – 生成配套的Label #!/usr/bin/python # -*- coding=utf-8 -*- from ...
【目标检测】yolo系列：从yolov1到yolov5之YOLOv5训练自己数据集（v6.0）
一.源码下载及requirments 源码下载地址:https://github.com/ultralytics/yolov5 (持续更新中) 本人所用环境如下: pytorch:1.8(因为cuda ...
手把手教你使用YOLOV5训练自己的数据集并用TensorRT加速
点击上方"3D视觉工坊",选择"星标" 干货第一时间送达前言本文主要介绍目标检测YOLOV5算法来训练自己的数据集,并且使用TensorRT来对训练好的模型 ...
yolov5训练自己的数据
组件知识笔记: yolov5组件笔记_jacke121的专栏-CSDN博客_yolov5深度可分离卷积参考:使用YOLOv5训练自己的数据集_laovife的博客-CSDN博客_使用yolov5训练 ...
YOLOv5训练自己的数据集实现视频的识别
写在前面我本来是使用这个模型进行手写签名的定位,但是因为比赛的主办方原因,数据不允许公开,所以我使用动物世界的一段开头视屏来制作我的数据集.这整个模型跑通的过程中,我参考了很多不错的博客,写这篇博客 ...
YOLOV5训练代码train.py注释与解析
YOLOv5代码注释版更新啦,注释的是最近的2021.07.14的版本,且注释更全 github: https://github.com/Laughing-q/yolov5_annotations Y ...
yolov5训练自己的数据集，OpenCV DNN推理
学更好的别人, 做更好的自己. --<微卡智享> 本文长度为4238字,预计阅读9分钟前言上一篇<OpenCV--自学笔记>搭建好了yolov5的环境,作为目标检测在应用中 ...
Windows10+YOLOv5训练自己的数据集
Windows10+YOLOv5训练自己的数据集一.环境和配置 1.1 安装anaconda 1.2 在anaconda中安装pytorch虚拟环境 1.3 安装CUDA和cudnn 1.4 安装p ...
Yolov5训练自己的数据集（windows10）
Yolov5训练自己的数据集(windows10) 环境配置+训练数据集一.Anaconda 的安装教程(图文) Anaconda下载下载地址:https://www.anaconda.com/d ...

如何用yolov5训练自己的图片

如何用yolov5训练自己的图片相关推荐

最新文章

热门文章