supervision
supervision copied to clipboard
yolov5 instance segmentation inference
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!