【YOLO-V3-SPP 源码解读】四、训练模块一、整体代码1、LoadImagesAndLabels2、DataLoader3、DarkNet模型搭建4、学习率设置5、开始训练6、Compute_loss其他文件
Compute_loss代码入口：train_eval_body.py # pred: tensor格式 list列表 存放三个tensor 对应的是三个yolo层的输出 # 例如[batch_size, 3, 23, 23, 25] [batch_size, 3, 46, 46, 25] [batch_size, 3, 96, 96, 25] # [batch_size, anchor_num, grid_h, grid_w
