YOLOv10训练损失、mAP画图功能 | 支持多结果对比,多结果绘在一个图片(科研必备)

一、本文介绍

本文给大家带来的是YOLOv10系列的绘图功能,我将向大家介绍YOLO系列的绘图功能。我们在进行实验时,经常需要比较多个结果,针对这一问题,我写了点代码来解决这个问题,它可以根据训练结果绘制损失(loss)和mAP(平均精度均值)的对比图。这个工具不仅支持多个文件的对比分析,还允许大家在现有代码的基础上进行修,从而达到数据可视化的功能,大家也可以将对比图来放在论文中进行对比也是非常不错的选择。

先展示一下效果图-> 

 专栏回顾:YOLOv10改进系列专栏——本专栏持续复习各种顶会内容——科研必备 

损失对比图象->


目录

一、本文介绍

二、绘图工具核心代码 

### YOLOv11 训练过程中的图方法 在YOLOv11训练过程中,制图表对于监控模型性能至关重要。通过可视化损失函数、精度和其他指标的变化趋势,能够更直观地了解训练进展并及时发现潜在问题。 #### 使用TensorBoard进行实时监测 为了方便开发者查看训练状态,许现代深度学习框架集成了TensorBoard工具。该工具允许用户轻松记录和展示各种统计信息。具体操作如下: - 启动训练脚本时指定日志目录作为参数传递给`SummaryWriter`对象; - 在每个epoch结束之后调用writer.add_scalar()接口向事件文件写入数据点; ```python from torch.utils.tensorboard import SummaryWriter # 创建一个summary writer实例指向存储位置 writer = SummaryWriter('runs/yolov11_experiment_1') for epoch in range(num_epochs): ... # 将当前迭代次数下的loss值加入tensorboard writer.add_scalar('Loss/train', running_loss / len(train_loader), epoch) # 关闭writer连接 writer.close() ``` #### 自定义图逻辑实现 除了依赖第三方库外,也可以编写自定义代码来完成这一任务。通常情况下会涉及到以下几个方面的工作: - 收集每一轮次结束后产生的评估结果; - 利用matplotlib等图形化包制作折线图或其他形式的表现方式; - 定期保存图像至磁盘以便后续查阅; 下面给出一段简单的Python程序片段用于说明上述思路: ```python import matplotlib.pyplot as plt def plot_metrics(metrics, save_path=None): epochs = list(range(1, len(metrics['mAP']) + 1)) fig, ax1 = plt.subplots() color = 'tab:red' ax1.set_xlabel('Epoch') ax1.set_ylabel('Loss', color=color) lns1 = ax1.plot(epochs, metrics['val_loss'], label='Validation Loss', color=color) ax1.tick_params(axis='y', labelcolor=color) ax2 = ax1.twinx() color = 'tab:blue' ax2.set_ylabel('mAP@0.5', color=color) lns2 = ax2.plot(epochs, metrics['mAP'], label='Mean Average Precision (mAP)', color=color) ax2.tick_params(axis='y', labelcolor=color) lns = lns1+lns2 labs = [l.get_label() for l in lns] ax1.legend(lns, labs, loc=0) if save_path is not None: plt.savefig(save_path) plot_metrics(training_history, './output/metrics_plot.png')[^4] ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Snu77

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值