card: RF-DETR-Seg family
Browse files
README.md
CHANGED
|
@@ -64,6 +64,16 @@ boxes = out["dets"].numpy()[0] # cxcywh, multiply by imag
|
|
| 64 |
keep = scores > 0.5 # done — no NMS
|
| 65 |
```
|
| 66 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 67 |
## Split deployment (`split/`)
|
| 68 |
|
| 69 |
`split/rfdetr-{nano,medium}_{backbone,head}.aimodel` separate the pure-ViT
|
|
|
|
| 64 |
keep = scores > 0.5 # done — no NMS
|
| 65 |
```
|
| 66 |
|
| 67 |
+
## RF-DETR-Seg (instance segmentation)
|
| 68 |
+
|
| 69 |
+
`rfdetr-seg-{nano,small,medium,large,xlarge,2xlarge}_float32.aimodel` — same
|
| 70 |
+
contract plus `masks [1, Q, R/4, R/4]`: per-query FULL-FRAME logit planes at
|
| 71 |
+
stride 4 (host: sigmoid > 0.5; no ROI plumbing, no NMS). All six gate on CPU
|
| 72 |
+
and GPU with binary-mask IoU 1.000 on stable scenes. M4 Max GPU:
|
| 73 |
+
seg-nano 312² **10.7 ms** → seg-2xlarge 768² **59.1 ms**.
|
| 74 |
+
|
| 75 |
+
<p align="center"><img src="demo_seg_coco_cats.jpg" width="440" alt="RF-DETR-Seg nano on Core AI"></p>
|
| 76 |
+
|
| 77 |
## Split deployment (`split/`)
|
| 78 |
|
| 79 |
`split/rfdetr-{nano,medium}_{backbone,head}.aimodel` separate the pure-ViT
|