YOLOv8Detection / scripts /yolo_export.py
Jaime García Villena
Export with nms
9bf782e
raw
history blame
190 Bytes
from ultralytics import YOLO
if __name__ == '__main__':
model = YOLO('yolov8n.pt')
model.export(format = 'saved_model', keras = True, nms = True)
model.export(format = 'tflite')