supervision icon indicating copy to clipboard operation
supervision copied to clipboard

yolov5 instance segmentation inference

Open DrawingProcess opened this issue 8 months ago • 15 comments

Search before asking

  • [X] I have searched the Supervision issues and found no similar feature requests.

Description

yolov5 is a detection and 'instance segmentation' model. so, I think If there is a mask area, I think it should be return.

like below

return cls(
    xyxy=yolov5_detections_predictions[:, :4],
    confidence=yolov5_detections_predictions[:, 4],
    class_id=yolov5_detections_predictions[:, 5].astype(int),
)

Use case

No response

Additional

No response

Are you willing to submit a PR?

  • [X] Yes I'd like to help by submitting a PR!

DrawingProcess avatar Jun 16 '24 13:06 DrawingProcess