Merge pull request #4 from hdnh2006/master

gpu support for autolabeling is now possible for yolov5

Author: vietanhdev

anylabeling/services/auto_labeling/yolov5.py

@@ -39,6 +39,8 @@ def __init__(self, model_config, on_message) -> None:
             raise Exception(f"Model not found: {model_abs_path}")
 
         self.net = cv2.dnn.readNet(model_abs_path)
+        self.net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
+        self.net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)
         self.classes = self.config["classes"]
 
     def pre_process(self, input_image, net):

requirements.txt

@@ -18,4 +18,4 @@ PyYAML==6.0
 colorama==0.4.5
 onnx==1.13.1
 onnxruntime==1.14.1
-qimage2ndarray==1.10.0
+qimage2ndarray==1.10.0