Views
No views yet
Roboflow/rf-detr-seg-small for zeromodels. One implementation runs unmodified on TensorFlow / Torch / JAX.RFDETRInstanceSegment): each query predicts a class, box, and mask.1import os
2os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
3
4from PIL import Image
5from zeromodels.models.rf_detr import (
6 RFDETRInstanceSegment,
7 RFDETRImageProcessor,
8)
9
10model = RFDETRInstanceSegment.from_weights("zeromodels/rfdetr-seg-small")
11processor = RFDETRImageProcessor.from_weights("zeromodels/rfdetr-seg-small")
12
13image = Image.open("your_image.jpg").convert("RGB")
14output = model(processor(image)["pixel_values"], training=False)
15result = processor.post_process_instance_segmentation(
16 output, threshold=0.5, target_sizes=[(image.height, image.width)]
17)[0]
18print(result["label_names"], result["scores"])from_weights("zeromodels/<variant>") (use RFDETRInstanceSegment for this repo):| Variant | Hub | Task |
|---|---|---|
rfdetr-nano | zeromodels/rfdetr-nano | object detection |
rfdetr-small | zeromodels/rfdetr-small | object detection |
rfdetr-medium | zeromodels/rfdetr-medium | object detection |
rfdetr-base | zeromodels/rfdetr-base | object detection |
rfdetr-large | zeromodels/rfdetr-large | object detection |
rfdetr-seg-preview | zeromodels/rfdetr-seg-preview | instance segmentation |
rfdetr-seg-nano | zeromodels/rfdetr-seg-nano | instance segmentation |
rfdetr-seg-small | zeromodels/rfdetr-seg-small | instance segmentation |
rfdetr-seg-medium | zeromodels/rfdetr-seg-medium | instance segmentation |
rfdetr-seg-large | zeromodels/rfdetr-seg-large | instance segmentation |
rfdetr-seg-xlarge | zeromodels/rfdetr-seg-xlarge | instance segmentation |
rfdetr-seg-xxlarge | zeromodels/rfdetr-seg-xxlarge | instance segmentation |
KERAS_BACKEND before importing Keras / zeromodels.RFDETRImageProcessor.from_weights(...) so the processor resolution matches the variant (bare constructor defaults to base's 560).RFDETRDetect + post_process_object_detection.RFDETRInstanceSegment + post_process_instance_segmentation.hf: prefix, e.g. RFDETRInstanceSegment.from_weights("hf:Roboflow/rf-detr-seg-small").