YOLOv10画图功能 | 支持多指标结果对比

1.直接绘制训练后保存的exp各指标结果

效果展示：

 

给出代码： 

import os
import pandas as pd
import matplotlib.pyplot as plt


def plot_metrics_and_loss(experiment_names, metrics_info, loss_info, metrics_subplot_layout, loss_subplot_layout,
                          metrics_figure_size=(15, 10), loss_figure_size=(15, 10), base_directory='runs/detect'):
    # 映射原始列名到实际存在的列名
    column_name_mapping = {
        'train/box_loss': 'train/box_om',
        'train/cls_loss': 'train/cls_om',
        &