Yolo-v5模型训练速度,与GeForce的AI算力描述

1.GeForce RTX3070 Ti官网参数:

GeForce RTXTM 3070 Ti 和 RTX 3070 显卡采用第 2 代 NVIDIA RTX 架构 - NVIDIA Ampere 架构。该系列产品搭载专用的第 2 代 RT Core ,第 3 代 Tensor Core、全新的 SM 多单元流处理器以及高速显存,助您在高性能要求的游戏中所向披靡。

2.一处未知来源的评测表:

3.另一处基于Ti3090跑yolov5的测试结果

3090量化到FP32,使用官方的Pytorch跑,完整的60 classes coco数据集一个是18.04小时。

Coco训练集大概是12万张图片,60 classes.

4.Yolov5实测数据

  • 运算环境:I5-12400
  • 数据集的规模:9000张图片;25.5万个标记框,5classes
  • 模型,训练环境Yolo-v5s, Pytorch, FP32, 5个类别
  • 1.5 hours/epochs
  • 结果:Optimizer stripped from runs\train\exp39\weights\last.pt, 14.3MB

附录 A 最简训练步骤

模型环境初始化的部分很简单,从略:

1.在yolov5 的data目录准备.yaml文件

相关数据集是在csdn下载的,大概6元钱。2276张图片。

train: ./smoker/images/train

val: ./smoker/images/val

nc: 1

names: "smoker"

2.将数据集放置在yolov5主目录内。smoker目录

3.修改train.py

def parse_opt(known=False):

parser = argparse.ArgumentParser()

parser.add_argument('--weights ', type=str, default='./yolov5-master/weights/yolov5s.pt', help='initial weights path')

parser.add_argument('--cfg ', type=str, default='./yolov5-master/models/yolov5s.yaml', help='model.yaml path')

parser.add_argument('--data ', type=str, default='./data/smoker.yaml' , help='dataset.yaml path')

parser.add_argument('--hyp', type=str, default='data/hyps/hyp.scratch.yaml', help='hyperparameters path')

parser.add_argument('--epochs ', type=int, default=2) # *300

普遍用的训练模型都是这个yolov5s,

Model Summary: 270 layers, 7022326 parameters, 7022326 gradients

7M个参数。梯度和参数是一回事,都是指一层一层的转换矩阵的总参数个数。

4.模型测试效果:

大概30分钟跑完一轮

相关推荐
2603_95470831几秒前
微能网的核心硬件协调控制装置有哪些功能?
大数据·运维·人工智能·架构·能源
workflower2 分钟前
AI 产业链带动效应加速,催生AI 原生企业
人工智能·机器学习·机器人·云计算·无人机
骑着蜗牛撵大象3273 分钟前
大模型工程化实战(十七):金融级 AI 落地——决策可追溯/可复现/可追责/不可抵赖这四件事怎么做
人工智能·金融·哈希·大模型工程化·决策可追溯·可复现
学编程就要猛3 分钟前
Spring AI Alibaba(上)
java·人工智能·spring·springaialibaba
Latchh7 分钟前
视频封面为什么会黑屏或截到旧帧
图像处理·人工智能·计算机视觉·音视频
小马哥crazymxm11 分钟前
Arxiv论文周选 (2026-W35)
论文阅读·人工智能·计算机视觉·目标跟踪
shaibdoio15 分钟前
瞬维AI落地经验:AI Agent工具调用准确率怎么提
数据库·人工智能·oracle
一缕82年的清风17 分钟前
当 AI 承包了 90% 的代码,架构师那致命的 10% 到底在控什么?
人工智能·vibecoding
头发够用的程序员17 分钟前
TensorRT 自定义算子插件实战(三):手搓 2×2 最大池化 customMaxpool
人工智能·pytorch·python·深度学习·神经网络·边缘计算·jetson
lisw0521 分钟前
人工智能网络:结构、动力学与经济学!
人工智能