update device in retina , update req file

Author: Ali-Fayzi

face_detection/detections_models/retinaface/face_detection.py

@@ -13,8 +13,8 @@
 
 class Retina_Face_Detection:
     def __init__(self):
-        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        self.model = get_model("resnet50_2020-07-20", max_size=2048 , device=device)
+        device = torch.device("cuda"  if torch.cuda.is_available() else "cpu")
+        self.model = get_model("resnet50_2020-07-20", max_size=640 , device=device)
         self.model.eval()
         self.warmup()
 

requirements.txt

@@ -4,6 +4,6 @@ opencv_contrib_python==4.8.1.78
 opencv_python==4.7.0.72
 opencv_python_headless==4.8.0.74
 Pillow==9.5.0
-torch==1.13.1
+torch
 ultralytics==8.0.222
-retinaface-pytorch==0.0.8
+retinaface-pytorch