【目标检测】metrice_curve和loss_curve对比图可视化

代码如下:

py 复制代码
import warnings
warnings.filterwarnings('ignore')

import os
import pandas as pd
import numpy as np
import matplotlib.pylab as plt

pwd = os.getcwd()

names = ['model1', 'model2', 'model3','ours']

plt.figure(figsize=(10, 10))

plt.subplot(2, 2, 1)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['metrics/precision(B)'] = data['metrics/precision(B)'].astype(np.float32).replace(np.inf, np.nan)
    data['metrics/precision(B)'] = data['metrics/precision(B)'].fillna(data['metrics/precision(B)'].interpolate())
    plt.plot(data['metrics/precision(B)'], label=i)
plt.xlabel('epoch')
plt.title('precision')
plt.legend()

plt.subplot(2, 2, 2)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['metrics/recall(B)'] = data['metrics/recall(B)'].astype(np.float32).replace(np.inf, np.nan)
    data['metrics/recall(B)'] = data['metrics/recall(B)'].fillna(data['metrics/recall(B)'].interpolate())
    plt.plot(data['metrics/recall(B)'], label=i)
plt.xlabel('epoch')
plt.title('recall')
plt.legend()

plt.subplot(2, 2, 3)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['metrics/mAP50(B)'] = data['metrics/mAP50(B)'].astype(np.float32).replace(np.inf, np.nan)
    data['metrics/mAP50(B)'] = data['metrics/mAP50(B)'].fillna(data['metrics/mAP50(B)'].interpolate())
    plt.plot(data['metrics/mAP50(B)'], label=i)
plt.xlabel('epoch')
plt.title('mAP_0.5')
plt.legend()

plt.subplot(2, 2, 4)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['metrics/mAP50-95(B)'] = data['metrics/mAP50-95(B)'].astype(np.float32).replace(np.inf, np.nan)
    data['metrics/mAP50-95(B)'] = data['metrics/mAP50-95(B)'].fillna(data['metrics/mAP50-95(B)'].interpolate())
    plt.plot(data['metrics/mAP50-95(B)'], label=i)
plt.xlabel('epoch')
plt.title('mAP_0.5:0.95')
plt.legend()

plt.tight_layout()
plt.savefig('metrice_curve.png')
print(f'metrice_curve.png save in {pwd}/metrice_curve.png')

plt.figure(figsize=(15, 10))

plt.subplot(2, 3, 1)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['train/box_loss'] = data['train/box_loss'].astype(np.float32).replace(np.inf, np.nan)
    data['train/box_loss'] = data['train/box_loss'].fillna(data['train/box_loss'].interpolate())
    plt.plot(data['train/box_loss'], label=i)
plt.xlabel('epoch')
plt.title('train/box_loss')
plt.legend()

plt.subplot(2, 3, 2)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['train/dfl_loss'] = data['train/dfl_loss'].astype(np.float32).replace(np.inf, np.nan)
    data['train/dfl_loss'] = data['train/dfl_loss'].fillna(data['train/dfl_loss'].interpolate())
    plt.plot(data['train/dfl_loss'], label=i)
plt.xlabel('epoch')
plt.title('train/dfl_loss')
plt.legend()

plt.subplot(2, 3, 3)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['train/cls_loss'] = data['train/cls_loss'].astype(np.float32).replace(np.inf, np.nan)
    data['train/cls_loss'] = data['train/cls_loss'].fillna(data['train/cls_loss'].interpolate())
    plt.plot(data['train/cls_loss'], label=i)
plt.xlabel('epoch')
plt.title('train/cls_loss')
plt.legend()

plt.subplot(2, 3, 4)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['val/box_loss'] = data['val/box_loss'].astype(np.float32).replace(np.inf, np.nan)
    data['val/box_loss'] = data['val/box_loss'].fillna(data['val/box_loss'].interpolate())
    plt.plot(data['val/box_loss'], label=i)
plt.xlabel('epoch')
plt.title('val/box_loss')
plt.legend()

plt.subplot(2, 3, 5)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['val/dfl_loss'] = data['val/dfl_loss'].astype(np.float32).replace(np.inf, np.nan)
    data['val/dfl_loss'] = data['val/dfl_loss'].fillna(data['val/dfl_loss'].interpolate())
    plt.plot(data['val/dfl_loss'], label=i)
plt.xlabel('epoch')
plt.title('val/dfl_loss')
plt.legend()

plt.subplot(2, 3, 6)
for i in names:
    data = pd.read_csv(f'runs/train/{i}/results.csv')
    data['val/cls_loss'] = data['val/cls_loss'].astype(np.float32).replace(np.inf, np.nan)
    data['val/cls_loss'] = data['val/cls_loss'].fillna(data['val/cls_loss'].interpolate())
    plt.plot(data['val/cls_loss'], label=i)
plt.xlabel('epoch')
plt.title('val/cls_loss')
plt.legend()

plt.tight_layout()
plt.savefig('loss_curve.png')
print(f'loss_curve.png save in {pwd}/loss_curve.png')

可视化结果展示

相关推荐
LaughingZhu10 分钟前
Product Hunt 每日热榜 | 2025-10-03
人工智能·经验分享·搜索引擎·产品运营
Godspeed Zhao17 分钟前
自动驾驶中的传感器技术65——Navigation(2)
人工智能·机器学习·自动驾驶
智能交通技术18 分钟前
iTSTech:智慧物流中自动驾驶、无人机与机器人的协同应用场景分析 2025
人工智能·机器学习·机器人·自动驾驶·无人机
先把态度摆正19 分钟前
自动驾驶决策规划算法(开幕式)
人工智能·机器学习·自动驾驶
风遥~41 分钟前
快速了解并使用Matplotlib库
人工智能·python·数据分析·matplotlib
用户51914958484544 分钟前
每个JavaScript开发者都应掌握的33个核心概念
人工智能·aigc
看今朝·1 小时前
【软件工程3.0】以UTDD/ATDD的理念深度融入AI生产流程
人工智能·软件工程·软件工程3.0·ai生产过程
IT古董2 小时前
【第五章:计算机视觉-项目实战之图像分割实战】2.图像分割实战:人像抠图-(5)模型训练与测试
人工智能·计算机视觉
qq_314009832 小时前
大模型之用LLaMA-Factory微调Deepseek-r1-8b模型实践
人工智能·语言模型
丁学文武2 小时前
大模型原理与实践:第三章-预训练语言模型详解_第2部分-Encoder-Decoder-T5
人工智能·语言模型·自然语言处理·大模型·t5·encoder-decoder