Deeds67 commited on
Commit
eb76679
·
verified ·
1 Parent(s): 8ab477d

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +53 -0
  2. detection/model.onnx +3 -0
README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - onnx
5
+ - object-detection
6
+ - rf-detr
7
+ - pet-detection
8
+ library_name: onnx
9
+ pipeline_tag: object-detection
10
+ ---
11
+
12
+ # RF-DETR Small (ONNX) — pet detection for Gallery
13
+
14
+ ONNX export of [RF-DETR](https://github.com/roboflow/rf-detr) Small by Roboflow, used by
15
+ [Gallery](https://github.com/open-noodle/gallery) for pet detection.
16
+
17
+ Unmodified COCO-pretrained weights. No fine-tuning.
18
+
19
+ ## Files
20
+
21
+ | Path | Description |
22
+ | ---------------------- | -------------------------------- |
23
+ | `detection/model.onnx` | RF-DETR Small, opset 17, batch 1 |
24
+
25
+ ## Inference contract
26
+
27
+ Getting any of this wrong degrades output silently rather than erroring.
28
+
29
+ **Input** — `input`, shape `[1, 3, 512, 512]`, float32:
30
+
31
+ 1. Decode to **RGB** (not BGR)
32
+ 2. Resize to 512×512 — plain square resize, **not** letterboxed
33
+ 3. Scale to `[0, 1]`
34
+ 4. Normalise with ImageNet statistics: mean `[0.485, 0.456, 0.406]`, std `[0.229, 0.224, 0.225]`
35
+ 5. Transpose HWC → CHW, add batch dimension
36
+
37
+ **Output** — two tensors:
38
+
39
+ | Name | Shape | Meaning |
40
+ | -------- | -------------- | ---------------------------------------------- |
41
+ | `dets` | `[1, 300, 4]` | Boxes as cx, cy, w, h — normalised to `[0, 1]` |
42
+ | `labels` | `[1, 300, 91]` | Class logits, **pre-sigmoid** |
43
+
44
+ Apply sigmoid to `labels`, then threshold. Classes use the **91-class COCO id space**
45
+ (90 categories plus background), not the contiguous 80-class space YOLO uses — so
46
+ `bird=16, cat=17, dog=18, horse=19, sheep=20, cow=21`.
47
+
48
+ The 300 queries are already deduplicated. **No NMS step is required.**
49
+
50
+ ## License
51
+
52
+ Apache-2.0, inherited from RF-DETR. See the
53
+ [upstream repository](https://github.com/roboflow/rf-detr) for full terms.
detection/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31fe724f8d5dbff5d09eb4c2c44d2d4083c086c167aec54e11e694e652307f9b
3
+ size 114448791