hmarchant commited on
Commit
a78475c
·
verified ·
1 Parent(s): 368c1f1

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +36 -0
  2. 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