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 


















