diff --git a/.gitignore b/.gitignore index 36aedab..e5ff2c5 100644 --- a/.gitignore +++ b/.gitignore @@ -26,19 +26,21 @@ output/ *.tmp.* *.log -# Media & large model binaries +# Media (videos/images stay local) *.mp4 *.avi *.mkv *.jpg *.jpeg *.png -*.engine -*.onnx -*.pt *.zip *.tar.gz +# TensorRT engines are Jetson build artifacts — rebuild via: +# python export_model.py models/.pt +# .pt (PyTorch) and .onnx ARE tracked so `git clone` is self-sufficient. +*.engine + # Test & coverage artifacts .pytest_cache/ .coverage diff --git a/AGENTS.md b/AGENTS.md index 47620a4..4cd97ac 100644 --- a/AGENTS.md +++ b/AGENTS.md @@ -34,7 +34,8 @@ Indonesian (`karung`=sack, `truk`=truck); YOLO class names are English tracker keeps `("sack", "truck", "box")` (`src/tracking.py`); counting uses `MultiClassLineCounter` = dual `LineCrossCounter`s on one shared line (`src/counting.py`). Same line geometry + 30px dedup for sacks and boxes. -- All weights live in `models/` (untracked; per-mode detector/filter matrix in +- All weights live in `models/` (`.pt`/`.onnx` tracked; `.engine` gitignored — + rebuild via `export_model.py`; per-mode detector/filter matrix in `models/README.md`). Verified classes: `truck-detector`={truck}, `model_karung_truk`/`v4-best`={sack,truck}, `karung-dimuat-*-seg-200e`= {person,sack} (seg; persons drawn, never counted), `best`={sack}, @@ -49,9 +50,9 @@ Indonesian (`karung`=sack, `truk`=truck); YOLO class names are English Deps: `requirements.txt` (unpinned; **never** pip-install torch from PyPI on the Jetson — use the NVIDIA wheels already on device). For visual checks run on a video file (`--source`), not the live RTSP. -- **Don't commit binaries/state**: `.gitignore` excludes `*.pt/*.onnx/*.engine`, - `*.mp4/*.jpg/*.png`, `*.db`, `.env`, `batch_history_folder/`. Large weights and - videos already sit untracked in the working tree — leave them alone. +- **Don't commit state**: `.gitignore` excludes `*.engine` (rebuildable), + `*.mp4/*.jpg/*.png`, `*.db`, `.env`, `batch_history_folder/`. Videos and local + DBs already sit untracked in the working tree — leave them alone. ## Deploy (Jetson `192.168.192.96`, user `jetson`) diff --git a/models/best.onnx b/models/best.onnx new file mode 100644 index 0000000..b0554de Binary files /dev/null and b/models/best.onnx differ diff --git a/models/best.pt b/models/best.pt new file mode 100644 index 0000000..1ee1e3e Binary files /dev/null and b/models/best.pt differ diff --git a/models/karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.onnx b/models/karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.onnx new file mode 100644 index 0000000..ac9721d Binary files /dev/null and b/models/karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.onnx differ diff --git a/models/karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.pt b/models/karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.pt new file mode 100644 index 0000000..3dacf48 Binary files /dev/null and b/models/karung-dimuat-detection-di-feedmill-yolo26n-seg-200e.pt differ diff --git a/models/model_karung_truk.onnx b/models/model_karung_truk.onnx new file mode 100644 index 0000000..3e2a8a9 Binary files /dev/null and b/models/model_karung_truk.onnx differ diff --git a/models/model_karung_truk.pt b/models/model_karung_truk.pt new file mode 100644 index 0000000..13fcb8e Binary files /dev/null and b/models/model_karung_truk.pt differ diff --git a/models/truck-detector.pt b/models/truck-detector.pt new file mode 100644 index 0000000..ef147f3 Binary files /dev/null and b/models/truck-detector.pt differ diff --git a/models/v4-best (1).pt b/models/v4-best (1).pt new file mode 100644 index 0000000..6809e10 Binary files /dev/null and b/models/v4-best (1).pt differ diff --git a/models/v4-best.onnx b/models/v4-best.onnx new file mode 100644 index 0000000..1322da8 Binary files /dev/null and b/models/v4-best.onnx differ diff --git a/models/v4-best.pt b/models/v4-best.pt new file mode 100644 index 0000000..6809e10 Binary files /dev/null and b/models/v4-best.pt differ diff --git a/models/yolo11n-bbox-100ep-sack+box-20260909-best.onnx b/models/yolo11n-bbox-100ep-sack+box-20260909-best.onnx new file mode 100644 index 0000000..6b1172f Binary files /dev/null and b/models/yolo11n-bbox-100ep-sack+box-20260909-best.onnx differ diff --git a/models/yolo11n-bbox-100ep-sack+box-20260909-best.pt b/models/yolo11n-bbox-100ep-sack+box-20260909-best.pt new file mode 100644 index 0000000..49878ae Binary files /dev/null and b/models/yolo11n-bbox-100ep-sack+box-20260909-best.pt differ