refactor: move weights to models/ + per-mode README; update paths, deploy, docs
ci / smoke (push) Canceled after 0s

This commit is contained in:
andrew committed 2026-09-11 15:52:38 +07:00
1 parent 6e4af50646
commit 262f7d064e
10 files changed
+157 -55

No files matched your search

+9 -2
View File
@@ -1,11 +1,18 @@
from ultralytics import YOLO
import torch
import os
import sys
# All weights live in models/ (see models/README.md).
MODELS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "models")
def main():
model_path = "karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.pt"
# Usage: python export_model.py [model.pt] (default: karung-dimuat seg model)
default_model = os.path.join(MODELS_DIR, "karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.pt")
model_path = sys.argv[1] if len(sys.argv) > 1 else default_model
if not os.path.exists(model_path):
print(f"Error: {model_path} tidak ditemukan di folder ini!")
print(f"Error: {model_path} tidak ditemukan!")
print(f"Usage: python export_model.py [path-ke-model.pt]")
return
print("=" * 60)