Changed bounding box threshold

somasundaram1702 · Aug 17, 2020 · c26d7eb · c26d7eb
1 parent 1c5b612
commit c26d7eb
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/detect_face_video.py b/detect_face_video.py
@@ -35,13 +35,14 @@ def vid_detect_face(self,frame):
     def vid_crop_resize(self):
       self.face = []
       for i in range(len(self.res_v)):
-        x1=self.res_v[i]['box'][0]
-        y1=self.res_v[i]['box'][1]
-        x2=self.res_v[i]['box'][0]+self.res_v[i]['box'][2]
-        y2=self.res_v[i]['box'][1]+self.res_v[i]['box'][3]
-        self._face = self.img_v[y1:y2,x1:x2]
-        self._face = cv2.resize(self._face,(160,160),cv2.INTER_AREA)
-        self.face.append(self._face)
+          if self.res_v[i]['confidence']>=0.90:
+            x1=self.res_v[i]['box'][0]
+            y1=self.res_v[i]['box'][1]
+            x2=self.res_v[i]['box'][0]+self.res_v[i]['box'][2]
+            y2=self.res_v[i]['box'][1]+self.res_v[i]['box'][3]
+            self._face = self.img_v[y1:y2,x1:x2]
+            self._face = cv2.resize(self._face,(160,160),cv2.INTER_AREA)
+            self.face.append(self._face)
       return(self.face)   
 
     def load_facenet(self):