ultralytics/test_yaml.py

18 lines
404 B
Python
Raw Permalink Normal View History

2025-02-25 03:58:34 +00:00
import warnings
warnings.filterwarnings('ignore')
import os, tqdm
from ultralytics import YOLO
if __name__ == '__main__':
error_result = []
2025-02-25 07:10:10 +00:00
try:
model = YOLO(f'ultralytics/cfg/models/mtl/yolov8-cls.yaml')
model.info(detailed=True)
model.profile([224, 224])
model.fuse()
except Exception as e:
print(e)
2025-02-25 03:58:34 +00:00
for i in error_result:
print(i)