bukuroo/

RF-DETR-Seg-ONNX

RF-DETR-Seg

• Model type:
  Instance segmentation
• Official GitHub repository:
  RF-DETR
• Usage
  python

  1from ezonnx import RFDETRSeg, visualize_images 2model = RFDETRSeg("preview") # automatically load from Hugging Face 3ret = model("image.jpg") 4# Visualize 5visualize_images("Detection result",ret.visualized_img) 6# Access results 7print(ret.boxes, ret.classes, ret.scores, ret.masks)
• ONNX export from torch
  python

  1from rfdetr import RFDETRSegPreview 2model = RFDETRSegPreview() 3# train here 4model.export()