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.2k stars 236 forks source link

allow to load rgb,gray image to inference #47

Closed qqqhhh-any closed 1 year ago

qqqhhh-any commented 1 year ago

I modified the function anylabeling/views/labeling/utils/opencv.py/qt img to cv img so that always returns images in rgb format, and when we load gray images, we can use the model for inference normally.