YOLOv10 / flops.py
Ean Yang
εˆε§‹εŒ–ιƒ¨η½²
69a5bd9
raw
history blame contribute delete
212 Bytes
from ultralytics import YOLOv10
model = YOLOv10('yolov10n.yaml')
model.model.model[-1].export = True
model.model.model[-1].format = 'onnx'
del model.model.model[-1].cv2
del model.model.model[-1].cv3
model.fuse()