Merge pull request #19 from deshwalmahesh/cuda_issue

changed code for Single GPU only

Author: deshwalmahesh

detection_helpers.py

@@ -20,7 +20,7 @@ def __init__(self, conf_thres:float = 0.25, iou_thresh:float = 0.45, agnostic_nm
         save_conf: whether to save confidences in 'save_txt' labels afters inference
         classes: Filter by class from COCO. can be in the format [0] or [0,1,2] etc
         '''
-        self.device = select_device("cuda" if torch.cuda.is_available() else 'cpu')
+        self.device = select_device("0" if torch.cuda.is_available() else 'cpu')
         self.conf_thres = conf_thres
         self.iou_thres = iou_thresh
         self.classes = classes