merge_benchmarks.py

import pandas as pd

def merge_results():
    try:
        # 读取两个 CSV
        df_nsm = pd.read_csv('benchmark_results_nsm.csv')
        df_x = pd.read_csv('benchmark_results_x.csv')
        
        # 合并
        df_final = pd.concat([df_nsm, df_x], ignore_index=True)
        
        # 按照模型大小排序 (n -> s -> m -> x)
        # 自定义排序逻辑
        model_order = {'yolov10n': 0, 'yolov10s': 1, 'yolov10m': 2, 'yolov10x': 3}
        df_final['sort_key'] = df_final['Model'].map(model_order)
        df_final = df_final.sort_values('sort_key').drop('sort_key', axis=1)
        
        # 打印
        print("\n=== Merged Benchmark Results ===")
        print(df_final)
        
        # 保存
        df_final.to_csv('benchmark_results_final.csv', index=False)
        print("\nSaved to benchmark_results_final.csv")
        
    except Exception as e:
        print(f"Error merging files: {e}")

if __name__ == "__main__":
    merge_results()