在Windows 10上训练YOLOv4时，将训练输出重定向到log文件，以便后续生成loss图表。与YOLOv3不同，直接使用tee命令在YOLOv4中无效，需要将标准输出和错误输出重定向到同一个文件。通过Python脚本extract_loss.py从log文件中提取loss信息，并生成log_loss.txt和iou_loss.txt。之后，使用visualization_loss.py和未完成的visualization_iou.py绘制loss和IoU图表，需根据实际情况调整脚本中的行数参数。
参考：https://blog.csdn.net/dz4543/article/details/88075408，https://blog.csdn.net/qq_34806812/article/details/81459982
用darknet训练的时候会显示一个动态loss图，但这个图不适合作为论文中使用。
在win10中做loss图，yolov4和yolov3有区别，用别的博主的针对yolov3的方法在yolov4中行不通! 用的别的博主的在linux中的方法在win10中也行不通！步骤如下：
（1）训练的时候输出重定向到log文件：(先新建一个visualization文件夹，然后在里面创建一个train_yolov4.log文本文件)
yolov3：（用yolov4的忽略这个）
./darknet detector train D:/software_engineer/darknet/darknet/cfg/voc.data D:/software_engineer/darknet/darknet/cfg/yolov3.cfg 2>1 | tee visualization/train_yolov3.log
yolov4：
./darknet detector train D:/software_engineer/yolov4/darknet-master/cfg/voc.data D:/software_engineer/yolov4/darknet-master/cfg/yolov4.cfg >> visualization/tra
最低0.47元/天 解锁文章
扫一扫
5113
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
