29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
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) |