当前位置:   article > 正文

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

results.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='')
  • 1
  • 2
  • 3
  • 4

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/129786
推荐阅读
相关标签
  

闽ICP备14008679号