refactor: move weights to models/ + per-mode README; update paths, deploy, docs
ci / smoke (push) Canceled after 0s
ci / smoke (push) Canceled after 0s
This commit is contained in:
1 parent
6e4af50646
commit
262f7d064e
10 files changed
+157
-55
No files matched your search
+9
-2
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user