paresh95 commited on
Commit
cd9af60
·
1 Parent(s): 008760f

PS |Updated requirements and fixed face threshold

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. src/face_symmetry.py +1 -1
requirements.txt CHANGED
@@ -6,4 +6,4 @@ imutils==0.5.4
6
  pillow==9.4.0
7
  pyyaml==6.0
8
  scikit-learn==1.2.2
9
- transfomers==4.28.1
 
6
  pillow==9.4.0
7
  pyyaml==6.0
8
  scikit-learn==1.2.2
9
+ transfomers==4.33.3
src/face_symmetry.py CHANGED
@@ -112,7 +112,7 @@ class GetFaceSymmetry:
112
  best_face_data, best_left_half, best_right_half = None, None, None
113
  for i in range(0, face_detections.shape[2]):
114
  confidence = face_detections[0, 0, i, 2]
115
- if confidence > 0.99:
116
  box = face_detections[0, 0, i, 3:7] * np.array(
117
  [self.w, self.h, self.w, self.h]
118
  )
 
112
  best_face_data, best_left_half, best_right_half = None, None, None
113
  for i in range(0, face_detections.shape[2]):
114
  confidence = face_detections[0, 0, i, 2]
115
+ if confidence > 0.98:
116
  box = face_detections[0, 0, i, 3:7] * np.array(
117
  [self.w, self.h, self.w, self.h]
118
  )