Upload folder using huggingface_hub
Browse files- README.md +36 -0
- countgd_litter_dense.pth +3 -0
README.md
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: other
|
| 3 |
+
license_name: countgdplusplus-research
|
| 4 |
+
license_link: https://github.com/niki-amini-naieni/CountGDPlusPlus
|
| 5 |
+
tags:
|
| 6 |
+
- object-counting
|
| 7 |
+
- litter
|
| 8 |
+
- open-vocabulary
|
| 9 |
+
- sam3
|
| 10 |
+
base_model: niki-amini-naieni/CountGDPlusPlus
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
# CountGD++ Litter — dense fine-tune (`countgd_litter_dense.pth`)
|
| 14 |
+
|
| 15 |
+
CountGD++ fine-tuned to count litter in photos, including extremely dense illegal-dumping piles.
|
| 16 |
+
|
| 17 |
+
- **Prompt:** `litter` · recommended confidence 0.23
|
| 18 |
+
- **Backbone/BERT frozen;** detection head fine-tuned with token focal loss + L1/GIoU on SAM 3 pseudo-labels
|
| 19 |
+
- **Teacher:** Meta SAM 3 (`facebook/sam3`), four-prompt union (`litter, trash, garbage, discarded packaging`), with **4×4 tiled inference** on human-flagged extreme-density images (tiled teacher mean 84 items vs 25 untiled on those scenes)
|
| 20 |
+
- **Training set:** 954 human-triaged images (194 extreme-density, 246 invalid images removed by review)
|
| 21 |
+
- **Holdout (200, stratified):** overall count MAE 10.9; dense MAE 51 → 17; normal MAE 9.5
|
| 22 |
+
- **48-image comparison:** mean count 65.8 (previous dense-blind fine-tune: 33.6; untuned CountGD++: 1.9)
|
| 23 |
+
|
| 24 |
+
## Checkpoint format
|
| 25 |
+
|
| 26 |
+
`torch.load(path, map_location="cpu", weights_only=False)` →
|
| 27 |
+
`{"model": state_dict, "epoch": 2, "prompt": "litter", "holdout": [...]}`.
|
| 28 |
+
Load `ckpt["model"]` into CountGD++ (`countgd_infer.py` in https://github.com/HugoMarchant/llitter-index).
|
| 29 |
+
|
| 30 |
+
## Limits
|
| 31 |
+
|
| 32 |
+
Counts are pseudo-label quality: SAM 3 still undercounts the very densest piles, and
|
| 33 |
+
near-dense images see a small accuracy regression. Counts are capped by CountGD++'s
|
| 34 |
+
query budget (~900). Trained on litter scenes; not a general object counter.
|
| 35 |
+
|
| 36 |
+
See RESULTS.md in the code repository for the full report.
|
countgd_litter_dense.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:035df6484d657db09d9ac89d2770c203a3c96bb324e867fd8603cb30c93bd746
|
| 3 |
+
size 937886333
|