Commit
Β·
be110dd
1
Parent(s):
ad89c3c
Init commit
Browse files
inference/face_detector.py
CHANGED
|
@@ -41,7 +41,7 @@ class FaceDetector(ABC):
|
|
| 41 |
y2 = max(min(round(y2), H), 0)
|
| 42 |
new_crop = [x1, y1, x2, y2]
|
| 43 |
final_crops.append(new_crop)
|
| 44 |
-
final_crops = np.array(final_crops, dtype=np.
|
| 45 |
return final_crops
|
| 46 |
|
| 47 |
def crop_faces(self, img, crops) -> List[np.ndarray]:
|
|
|
|
| 41 |
y2 = max(min(round(y2), H), 0)
|
| 42 |
new_crop = [x1, y1, x2, y2]
|
| 43 |
final_crops.append(new_crop)
|
| 44 |
+
final_crops = np.array(final_crops, dtype=np.int32)
|
| 45 |
return final_crops
|
| 46 |
|
| 47 |
def crop_faces(self, img, crops) -> List[np.ndarray]:
|