YOLOv5代码阅读笔记
YOLOv5代码阅读笔记1.train.py1.1 设置日志文件输出位置、名称等1.2 载入图片、标签1.3 确定图片训练的尺寸1.4 判断参数的形式为weight或bias,定义模型不同部分,并设置相应的学习率和参数。1.5 设置学习率的变化方式,这里采用cosine 学习率趋势1.6 定义训练集和验证集的数据读取方式1.7 统计样本分布1.8 获取anchor2. dataset.py 中的数据增强2.1 mosaic数据增强 def load_mosaic(self, index)2.2 色彩空间HS
转载
2020-07-23 09:11:44 ·
2048 阅读 ·
4 评论