Algorithm-Repo/SourceCode/MachineLearningUtils/VINNOFramework/YOLO/test_yaml.py

import warnings
warnings.filterwarnings('ignore')
import os, tqdm
from ultralytics import YOLO

if __name__ == '__main__':
    error_result = []
    for yaml_path in tqdm.tqdm(os.listdir('ultralytics/cfg/models/v8')):
        if 'rtdetr' not in yaml_path and 'cls' not in yaml_path and 'world' not in yaml_path:
            try:
                model = YOLO(f'ultralytics/cfg/models/v8/{yaml_path}')
                model.info(detailed=True)
                model.profile([640, 640])
                model.fuse()
            except Exception as e:
                error_result.append(f'{yaml_path} {e}')
    
    for yaml_path in tqdm.tqdm(os.listdir('ultralytics/cfg/models/v10')):
        if 'rtdetr' not in yaml_path and 'cls' not in yaml_path and 'world' not in yaml_path:
            try:
                model = YOLO(f'ultralytics/cfg/models/v10/{yaml_path}')
                model.info(detailed=True)
                model.profile([640, 640])
                model.fuse()
            except Exception as e:
                error_result.append(f'{yaml_path} {e}')
    
    for i in error_result:
        print(i)