14 lines
346 B
Python
14 lines
346 B
Python
|
import warnings
|
||
|
|
warnings.filterwarnings('ignore')
|
||
|
|
from ultralytics import YOLO
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
# choose your yaml file
|
||
|
|
model = YOLO('ultralytics/cfg/models/v8/yolov8n.yaml')
|
||
|
|
model.info(detailed=True)
|
||
|
|
try:
|
||
|
|
model.profile(imgsz=[640, 640])
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
pass
|
||
|
|
model.fuse()
|