roboflow / supervision

We write your reusable computer vision tools. 💜
https://supervision.roboflow.com
MIT License
24.03k stars 1.79k forks source link

Ultralytics all model support for detections #221

Closed hardikdava closed 1 year ago

hardikdava commented 1 year ago

Search before asking

Description

sv.Detections.from_yolov8() only supports yolov8 object detection and segmentation result conversion. But there are several other models are implemented in ultralytics such as SAM, FastSAM, MobileSAM, RT-DETR. Current method of conversion do not support all these models.

Use case

This feature will completely cover all object detection and segmentation results conversion to supervision detections. This will make easy integration of future ultralytics models also.

Additional

No response

Are you willing to submit a PR?

SkalskiP commented 1 year ago

@hardikdava, cool idea! 👍🏻

hardikdava commented 1 year ago

Thanks @SkalskiP , can you mark it for next release? Soon I will share colab for testing. It may requires to make one of the API as deprecated stage.

SkalskiP commented 1 year ago

Hi, @hardikdava 👋🏻! Yup, I marked it for the supervision-0.13.0 release.