bukuroo/RF-DETR-Seg-ONNX

RF-DETR-Seg

ONNX models for inference with EZONNX

• Model type:
  Instance segmentation

• Official GitHub repository:
  RF-DETR

• Usage

  from ezonnx import RFDETRSeg, visualize_images
  model = RFDETRSeg("preview")  # automatically load from Hugging Face
  ret = model("image.jpg")
  # Visualize
  visualize_images("Detection result",ret.visualized_img)
  # Access results
  print(ret.boxes, ret.classes, ret.scores, ret.masks)
  

• ONNX export from torch

  from rfdetr import RFDETRSegPreview
  model = RFDETRSegPreview()
  # train here
  model.export()