activate cuda

detector/utils.py

@@ -12,14 +12,10 @@ device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 print('Loading models...', device)
 
 model_plates = torch.hub.load('ultralytics/yolov5', 'custom',
-                              path=os.path.join(BASE_DIR, 'detector', 'static', 'plates.pt'))
-
-model_plates.to(device).eval()
+                              path=os.path.join(BASE_DIR, 'detector', 'static', 'plates.pt'), device=device)
 
 model_chars = torch.hub.load('ultralytics/yolov5', 'custom',
-                             path=os.path.join(BASE_DIR, 'detector', 'static', 'chars.pt'))
-
-model_chars.to(device).eval()
+                             path=os.path.join(BASE_DIR, 'detector', 'static', 'chars.pt'), device=device)
 
 
 def pad_img_to_fit_bbox(img, x1, x2, y1, y2):