gpu support for autolabeling is now possible for yolov5

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.33k stars 245 forks source link

Closed hdnh2006 closed 1 year ago

hdnh2006 commented 1 year ago

If dnn module with GPU support for openCV is available, the labeling will be done into the GPU.

You must install your dnn module with GPU support following by running the file opencv_dnn.sh.

This file is based on the instructions provided by Adrian Rosebrock (@jrosebr1) and he has the whole credit and not me.