18 lines
404 B
Python
18 lines
404 B
Python
import warnings
|
|
warnings.filterwarnings('ignore')
|
|
import os, tqdm
|
|
from ultralytics import YOLO
|
|
|
|
if __name__ == '__main__':
|
|
error_result = []
|
|
|
|
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)
|
|
|
|
for i in error_result:
|
|
print(i) |