An RF-DETR Small model that detects the 80 COCO categories, exported to ONNX. Small and fast enough for realtime inference on a phone or in the browser on WebGPU. Built for dashradar.app. The weights are Roboflow's COCO checkpoint and nothing is trained here, so what this repo adds is the export and the contract around it. The architecture is described in arXiv:2511.09554.

onnx/model_fp16.onnx is the only build published here. fp16 weights and compute, fp32 inputs and outputs, so you feed it a normal float32 array. It is mixed precision: its GridSample nodes stay at fp32, because onnxruntime-web's fp16 GridSample shader is broken on WebGPU and fails by returning wrong boxes rather than by erroring.

Its size is in the CHANGELOG entry for the revision you pin, along with everything measured about that build.

There is no fp32 build. Only WebGPU is targeted here and only the fp16 file is streamed, so an fp32 file would be twice the download for a runtime nothing in this project uses, and a second artifact every release has to re-verify. The fp16 build is checked against an fp32 base at export time and against the upstream PyTorch reference before that, and those figures are in the CHANGELOG.

Want an fp32 build? Nothing here stops you. The source checkpoint is public and stable at https://storage.googleapis.com/rfdetr/small_coco/checkpoint_best_regular.pth, RFDETRSmall() downloads it for you, and model.export() writes ONNX. The full recipe this repo uses, including the onnxslim step, is in .agents/skills/cut-release/references/export.md.

The source checkpoint is not republished here either. It is Roboflow's and it is already public at the URL above. If you want it in transformers form, use stevenbucaille/rf-detr-small, which is the same checkpoint converted, and which this export was verified against.

The 300 is the query count. Read it off the tensor length rather than hardcoding it, so a future build that emits a different number still decodes.

No NMS is baked in. RF-DETR's set loss usually makes it unnecessary. Add a light dedup if duplicate boxes show up.

Preprocessing: largest centered square of the RGB image, bilinear resize to 512x512, scale to [0, 1], ImageNet mean [0.485, 0.456, 0.406] and std [0.229, 0.224, 0.225], NCHW. The cost of the crop is field of view. A 16:9 frame loses about 44% of its width.

Decoding: the class head is 91 wide, which is COCO's raw category id space and not a dense 80. Index 0 is COCO's N/A background slot and is never read. The 80 real categories sit sparsely across indices 1 to 90, with ten N/A gaps where COCO retired a category id. Assuming a dense 0..79 table reads the wrong label for most of the classes. So sigmoid the logits, and per query take the highest scoring index that the model's own names map actually names. One box gets one label. Boxes are normalized to the cropped square, not the original frame, so map them back through the crop origin.

Metadata: the build carries its label map, input contract, source checkpoint and release tag in ONNX metadata_props. Read them with ort.InferenceSession(path).get_modelmeta().custom_metadata_map, or open the file in Netron. The keys follow the convention Ultralytics' YOLO exporter set, so the values are Python reprs rather than JSON. Parse names with ast.literal_eval and you get 80 entries keyed by integer logit index, sparse across 1 to 90 to match the head. That map is the only machine readable record of what a slot means, so read it from the file instead of shipping your own table.

Confidence tracks apparent size, not distance. The same object scores far higher in a tight native-resolution crop than it does when a full frame is downscaled into 512x512, because the second view leaves it a few pixels wide. For distant objects use a tighter crop, not a lower threshold.

This repo publishes an untuned model. No confidence sweep was run here, and there is no dataset behind these tags to sweep against. The snippet above uses 0.35, which is the value the upstream model card's usage example keeps detections above. It is an example's value, not the cut any metric was reported at, and it is not tuned for any particular deployment. It is a place to start.

If a release ever does measure a threshold, it goes in the CHANGELOG entry for that version, so read the entry for the revision you pin. Tune against your own images either way. The right cut depends on whether a false positive or a miss costs you more, and this repo cannot know that.

dashradar.app can add a model by pasting a Hugging Face URL. Paste the URL of the build itself, on a tag:

A bare repo URL also resolves, since the app looks for a single ONNX file in the repo and there is only one now. It pins to a commit sha, though, and a sha tells you nothing later. Naming the file on a tag gives you a readable release name you can match against a CHANGELOG entry.

The stored entry is immutable either way. The app resolves a main URL to the commit sha before saving it.

Tags cut before the fp32 build was dropped hold two ONNX files, so a bare repo URL on one of those is ambiguous and the app rejects it. Name the file.

General purpose object detection over the 80 COCO categories, on single images or realtime video.

It is trained on COCO and inherits COCO's biases. The categories are the ones COCO chose to label, the images are the scenes COCO collected, and quality drops on anything outside that distribution. Accuracy is uneven across classes, and small or heavily occluded objects are the weakest case. The center crop means anything outside the middle square of a wide frame is never seen.

Do not rely on it for safety-critical decisions. A missed detection is a normal outcome rather than a bug.

reference/test.jpg is COCO val2017 image id 577932, released under the Attribution License (CC BY 2.0). The original is http://farm5.staticflickr.com/4019/5159956078_f820c56d6f_z.jpg. It is center-cropped from 640x543 and resized to 512x512, so it is exactly the model input and needs no preprocessing.

The model is RF-DETR by Roboflow, trained by them on COCO. stevenbucaille/rf-detr-small is the transformers conversion this export was checked against.

Cite the architecture, not this repo. The export adds no research.

Apache 2.0, matching upstream RF-DETR.