Commit ·
07fd56c
1
Parent(s): f1ee07e
face recognition
Browse files
app/Hackathon_setup/face_recognition.py
CHANGED
|
@@ -122,8 +122,6 @@ def get_face_class(img1):
|
|
| 122 |
if det_img1 == 0:
|
| 123 |
# fallback: use original image
|
| 124 |
det_img1 = Image.fromarray(cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY))
|
| 125 |
-
else:
|
| 126 |
-
det_img1 = Image.fromarray(cv2.cvtColor(det_img1, cv2.COLOR_BGR2GRAY))
|
| 127 |
|
| 128 |
# 4 Transform the face
|
| 129 |
img_tensor = trnscm(det_img1).unsqueeze(0)
|
|
|
|
| 122 |
if det_img1 == 0:
|
| 123 |
# fallback: use original image
|
| 124 |
det_img1 = Image.fromarray(cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY))
|
|
|
|
|
|
|
| 125 |
|
| 126 |
# 4 Transform the face
|
| 127 |
img_tensor = trnscm(det_img1).unsqueeze(0)
|
requirements.txt
CHANGED
|
@@ -5,7 +5,7 @@ requests==2.23.0
|
|
| 5 |
jinja2==3.1.2
|
| 6 |
python-multipart==0.0.6
|
| 7 |
|
| 8 |
-
scikit-learn==1.6.
|
| 9 |
joblib==1.3.2
|
| 10 |
Pillow==9.4.0
|
| 11 |
torch==2.1.0
|
|
|
|
| 5 |
jinja2==3.1.2
|
| 6 |
python-multipart==0.0.6
|
| 7 |
|
| 8 |
+
scikit-learn==1.6.1
|
| 9 |
joblib==1.3.2
|
| 10 |
Pillow==9.4.0
|
| 11 |
torch==2.1.0
|