File size: 4,248 Bytes
375b643
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
aa02851
375b643
 
 
 
 
aa02851
 
 
 
375b643
aa02851
 
375b643
 
 
 
 
 
 
 
 
 
 
 
 
aa02851
 
 
 
375b643
 
 
 
 
 
aa02851
375b643
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
aa02851
375b643
 
 
aa02851
375b643
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
---
license: apache-2.0
library_name: rfdetr.cpp
tags:
  - object-detection
  - rfdetr
  - gguf
  - ggml
  - cpp-inference
  - image-segmentation
  - instance-segmentation
pipeline_tag: image-segmentation
base_model: roboflow/rfdetr
---

# RF-DETR Seg-Medium β€” GGUF for rfdetr.cpp

GGUF-format weights of [Roboflow RF-DETR Seg-Medium](https://github.com/roboflow/rf-detr) (segmentation variant) for use with [rfdetr.cpp](https://github.com/mudler/rf-detr.cpp), a C++/ggml implementation that matches the upstream PyTorch model on CPU.

This repo contains all four standard quantizations of this variant. **F16 is the recommended default** β€” near-identical accuracy to F32 (see the table below), 1.87Γ— smaller than F32, and it takes ggml's F32Γ—F16 matmul fast path.

## Available files

| File | Quant | Size (MB) | Recall @ IoU 0.5 | Recall @ IoU 0.95 | Mean mask IoU | Pixel agreement | Latency (median ms, T=8) |
|---|---|---:|---:|---:|---:|---:|---:|
| `rfdetr-seg-medium-f32.gguf` | F32 | 133.9 | 0.9841 | 0.9841 | 0.9975 | 0.9999 | β€” |
| `rfdetr-seg-medium-f16.gguf` ← **recommended** | F16 | 71.6 | 0.9841 | 0.9841 | 0.9975 | 0.9999 | β€” |
| `rfdetr-seg-medium-q8_0.gguf` | Q8_0 | 42.5 | 0.9841 | 0.9637 | 0.9938 | 0.9999 | β€” |
| `rfdetr-seg-medium-q4_K.gguf` | Q4_K | 33.6 | 0.9296 | 0.6420 | 0.9699 | 0.9993 | β€” |

All accuracy numbers are computed against the upstream PyTorch reference (`rfdetr 1.9.0`) on 7 COCO val2017 images at threshold 0.5.
No latency benchmark has been recorded for this variant yet, so the latency column is left empty. Run `scripts/quick_bench.sh` locally for timings on your own hardware.

## Architecture

- Backbone: DINOv2-small
- Input resolution: 432Γ—432
- Patch size: 12
- Decoder layers: 5
- Object queries: 200
- Task: instance segmentation (boxes + per-query masks)
- Mask resolution: 108Γ—108 per query (image_size / 4)

## Quantization notes

- **F32** β€” the full-precision conversion, 134 MB, and the closest match to the PyTorch reference (see the table above for measured agreement).
- **F16** β€” matmul-multiplicand weights only; LayerNorms, conv kernels, embeddings, biases, and layer-scale gammas stay F32. Accuracy tracks F32 closely on this model, and it takes ggml's F32Γ—F16 matmul fast path.
- **Q8_0** β€” best size/accuracy tradeoff under F16; ~3.2Γ— smaller than F32 with effectively identical detections.
- **Q4_K** β€” smallest practical quant. Rows with `ne[0] % 256 != 0` (the decoder's 128-dim MLP halves) silently fall back to Q8_0 per ggml's quantizer logic β€” net compression is still ~4.0Γ— over F32. Use only when the size budget is tight; expect a measurable Recall@0.95 drop relative to F16/Q8_0 (see file table above).

## Usage

```bash
# 1. Clone + build rfdetr.cpp
git clone https://github.com/mudler/rf-detr.cpp
cd rf-detr.cpp
cmake -B build -DRFDETR_BUILD_CLI=ON && cmake --build build -j

# 2. Download a quant (F16 recommended)
hf download mudler/rfdetr-cpp-seg-medium rfdetr-seg-medium-f16.gguf --local-dir models/

# 3. Run segmentation (writes per-detection PNG masks to /tmp/seg_masks/)
build/bin/rfdetr-cli detect \
    --model models/rfdetr-seg-medium-f16.gguf \
    --input my_image.jpg \
    --threshold 0.5 --threads 8 \
    --masks /tmp/seg_masks \
    --output detections.json
```

## Accuracy methodology

All accuracy metrics are computed against the upstream PyTorch reference (`rfdetr 1.9.0`) on 7 COCO val2017 images at threshold 0.5. Each detection match uses greedy Hungarian-style assignment by IoU (β‰₯ 0.5 lenient, β‰₯ 0.95 strict) with class equality required.

Mask metrics are pixel-wise IoU between binary masks at the **original** image resolution (not the network's working resolution), after sigmoid + bicubic upsample of the per-query mask logits. **Pixel agreement** is the fraction of pixels where the C++ and PyTorch binary masks match.

See [BENCHMARK.md](https://github.com/mudler/rf-detr.cpp/blob/main/BENCHMARK.md) and [`benchmarks/results/accuracy_sweep.json`](https://github.com/mudler/rf-detr.cpp/blob/main/benchmarks/results/accuracy_sweep.json) for the full sweep across the (variant Γ— quant) cells.

## License

Apache-2.0 β€” matches the upstream [rfdetr](https://github.com/roboflow/rf-detr) license.