YOLOv5自定义数据集训练
发布时间
阅读量：
阅读量
基于 YOLOV5 模型进行自定义数据集训练的实践流程
前言
【个人学习笔记记录，如有错误，欢迎指正】
YOLO-V5 代码仓库地址：https://github.com/ultralytics/yolov5
一、环境配置与系统搭建
以下代码适用于 Windows 系统运行！！！
二、需注意的配置参数介绍
在训练脚本中，定位到用于解析选项的 parse_opt 函数。
parser.add_argument('--weights', type=str, default=ROOT / 'yolov5s.pt', help='initial weights path')
parser.add_argument('--data', type=str, default=ROOT / 'data/coco128.yaml', help='dataset.yaml path')
parser.add_argument('--epochs', type=int, default=300)
parser.add_argument('--batch-size', type=int, default=16, help='total batch size for all GPUs, -1 fo全部评论 (0)
还没有任何评论哟~