YOLOV5训练完成后只有result.csv文件并没有可视化结果

在网上下载YOLOV5项目训练自己的数据集后,遇到了无可视化结果这一问题。在网上搜索后发现并无相关解决办法,只是一贯的说可能是中间训练终止了。这,我直接就笑了,我自己训练多少轮,完没完成还是知道的。
于是我在可视化脚本中去找result.png这行代码。
这是plot脚本下的result可视化模块
随后我发现不论是在train.py还是val.py模块下都没有引用这一函数。
train.py模块导入包
val.py导入包
于是选择新建一个脚本单独用来将保存下的result.csv文件可视化

# 将results.csv可视化
from utils.plots import plot_results

plot_results(file='C:/pythonx/yolo/yolov5-fire/runs/train/exp/results.csv', dir='')

短短两行代码问题解决,记得将file路径修改为自己result.csv所在文件目录,result.png文件也会生成在该目录下。


版权声明:本文为qq_43445205原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。