diff --git a/onnx2tflite.py b/onnx2tflite.py index 2993a2f4..5c66f05b 100644 --- a/onnx2tflite.py +++ b/onnx2tflite.py @@ -7,7 +7,7 @@ model = YOLO("runs/detect/train/weights/best.pt") model.export(format="tflite") # creates 'yolo11n_float32.tflite' # Load the exported TFLite model -tflite_model = YOLO("runs/detect/train/weights/best.tflite") +tflite_model = YOLO("runs/detect/train/weights/best_saved_model/best_float32.tflite") # Run inference results = tflite_model("datasets/valid/images/2007_000042_jpg.rf.dadefef10ec4b9422a1689f7ce40588e.jpg") \ No newline at end of file diff --git a/runs/detect/train/weights/best.onnx b/runs/detect/train/weights/best.onnx index 3e21b710..fe6cf6a0 100644 Binary files a/runs/detect/train/weights/best.onnx and b/runs/detect/train/weights/best.onnx differ diff --git a/runs/detect/train/weights/best_saved_model/best_float16.tflite b/runs/detect/train/weights/best_saved_model/best_float16.tflite new file mode 100644 index 00000000..1d3a3083 Binary files /dev/null and b/runs/detect/train/weights/best_saved_model/best_float16.tflite differ diff --git a/runs/detect/train/weights/best_saved_model/best_float32.tflite b/runs/detect/train/weights/best_saved_model/best_float32.tflite new file mode 100644 index 00000000..396f9665 Binary files /dev/null and b/runs/detect/train/weights/best_saved_model/best_float32.tflite differ diff --git a/runs/detect/train/weights/best_saved_model/fingerprint.pb b/runs/detect/train/weights/best_saved_model/fingerprint.pb new file mode 100644 index 00000000..e5b7c1f4 --- /dev/null +++ b/runs/detect/train/weights/best_saved_model/fingerprint.pb @@ -0,0 +1 @@ +Î]Ɏ Ɋ㪛 ̋明(ӎ2:3350961098177553315 \ No newline at end of file diff --git a/runs/detect/train/weights/best_saved_model/metadata.yaml b/runs/detect/train/weights/best_saved_model/metadata.yaml new file mode 100644 index 00000000..f0edffb2 --- /dev/null +++ b/runs/detect/train/weights/best_saved_model/metadata.yaml @@ -0,0 +1,40 @@ +description: Ultralytics YOLO12s model trained on data.yaml +author: Ultralytics +date: '2025-06-04T03:14:12.116915' +version: 8.3.148 +license: AGPL-3.0 License (https://ultralytics.com/license) +docs: https://docs.ultralytics.com +stride: 32 +task: detect +batch: 1 +imgsz: +- 640 +- 640 +names: + 0: aeroplane + 1: bicycle + 2: bird + 3: boat + 4: bottle + 5: bus + 6: car + 7: cat + 8: chair + 9: cow + 10: diningtable + 11: dog + 12: horse + 13: motorbike + 14: person + 15: pottedplant + 16: sheep + 17: sofa + 18: train + 19: tvmonitor +args: + batch: 1 + fraction: 1.0 + half: false + int8: false + nms: false +channels: 3 diff --git a/runs/detect/train/weights/best_saved_model/saved_model.pb b/runs/detect/train/weights/best_saved_model/saved_model.pb new file mode 100644 index 00000000..ddc12f4c Binary files /dev/null and b/runs/detect/train/weights/best_saved_model/saved_model.pb differ diff --git a/runs/detect/train/weights/best_saved_model/variables/variables.data-00000-of-00001 b/runs/detect/train/weights/best_saved_model/variables/variables.data-00000-of-00001 new file mode 100644 index 00000000..e12cbc7c Binary files /dev/null and b/runs/detect/train/weights/best_saved_model/variables/variables.data-00000-of-00001 differ diff --git a/runs/detect/train/weights/best_saved_model/variables/variables.index b/runs/detect/train/weights/best_saved_model/variables/variables.index new file mode 100644 index 00000000..aadf262c Binary files /dev/null and b/runs/detect/train/weights/best_saved_model/variables/variables.index differ