摘要: YOLOv5代码详解 (第一部分) 1. train.py 1.1 使用nvidia的apex接口计算混合精度训练 1.2 获取文件路径 1.3 获取数据路径 1.4 移除之前的结果 1.5 创建模型 1.6 检查训练和测试图片尺寸 1.7 设置优化器参数 1.8 加载预训练模型和权重,并写入训练结 阅读全文
posted @ 2024-01-29 22:06 ^酥糖^ 阅读(460) 评论(0) 推荐(0)