vietanhdev / anylabeling

Effortless AI-assisted data labeling with AI support from YOLO, Segment Anything (SAM+SAM2), MobileSAM!!
https://anylabeling.nrl.ai
GNU General Public License v3.0
2.27k stars 239 forks source link

Use Yolo detection bbox'es for SAM segmentation #130

Open AlienRouge opened 1 year ago

AlienRouge commented 1 year ago

Hello, how about adding the functionality of using bounding boxes from the custom Yolo model for SAM segmentation. This will allow you to refine the contours of the found objects and generate more accurate bounding boxes.