16 lines
304 B
Python
16 lines
304 B
Python
from ultralytics import YOLO
|
|
import os
|
|
dir = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
|
|
|
|
model = YOLO(dir + "/best.pt")
|
|
|
|
# Exportiere es ins TFLite-Format
|
|
model.export(
|
|
format="tflite",
|
|
dynamic=False,
|
|
inference_type="uint8",
|
|
quantize=True,
|
|
calibration_data="path/to/rep_dataset/"
|
|
) |