mudler's picture
Add/update model card for seg-large
4d93573 verified
---
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-Large β€” GGUF for rfdetr.cpp
GGUF-format weights of [Roboflow RF-DETR Seg-Large](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** β€” same accuracy as F32, 1.85Γ— smaller, and typically the fastest on modern CPUs thanks to ggml's F32Γ—F16 matmul fast path.
## Available files
| File | Quant | Size (MB) |
|---|---|---:|
| `rfdetr-seg-large-f32.gguf` | F32 | 140.9 |
| `rfdetr-seg-large-f16.gguf` ← **recommended** | F16 | 75.7 |
| `rfdetr-seg-large-q8_0.gguf` | Q8_0 | 45.1 |
| `rfdetr-seg-large-q4_K.gguf` | Q4_K | 35.9 |
> Accuracy + latency for this variant haven't been added to the [BENCHMARK.md](https://github.com/mudler/rf-detr.cpp/blob/main/BENCHMARK.md) sweep yet; the C++ implementation is verified to load and run `rfdetr-cli detect` end-to-end on every quant. Run `scripts/sweep_accuracy.py --variant seg-large` locally for parity numbers.
All accuracy numbers are computed against the upstream PyTorch reference (`rfdetr 1.7.0`) on 7 COCO val2017 images at threshold 0.5. Latency is measured with `rfdetr-cli bench` (8 iters + 3 warmup) at T=8 threads on a single AMD Ryzen 9 9950X3D image (`coco_kitchen.jpg`, 640x427).
## Architecture
- Backbone: DINOv2-small
- Input resolution: 504Γ—504
- Patch size: 12
- Decoder layers: 5
- Object queries: 200
- Task: instance segmentation (boxes + per-query masks)
- Mask resolution: 126Γ—126 per query (image_size / 4)
## Quantization notes
- **F32** β€” full-precision reference, ~120 MB. Bit-exact PyTorch parity.
- **F16** β€” matmul-multiplicand weights only; LayerNorms, conv kernels, embeddings, biases, and layer-scale gammas stay F32. Lossless on this model and consistently the fastest variant on CPU.
- **Q8_0** β€” best size/accuracy tradeoff under F16; ~3Γ— 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, 60 tensors) silently fall back to Q8_0 per ggml's quantizer logic β€” net compression is still ~3.8Γ— 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 rt-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-large rfdetr-seg-large-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-large-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.7.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.