YOLOv8或者YOLOv10训练生成的exp文件进行遍历,找出results.csv文件中metrics/mAP50(B)、metrics/mAP50-95(B)值和args.yaml中的mode文件,将他们保存到一个excel文件中。
YOLO训练train.py代码如下:
模型训练好后生成exp文件如下:
生成excel文件完整代码如下:
import os
import pandas as pd
import yaml
def find_max_map50_and_model(root_dir):
results = []
for subdir, _, _ in os.walk(root_dir):
# 获取每个子文件夹中的所有文件夹
subfolders = [f.path for f in os.scandir(subdir) if f.is_dir()]
for folder in subfolders:
# 检查每个文件夹中是否包含results.csv和args.yaml文件
results_csv_path = os.path.join(folder, 'results.csv')
args_yaml_path