Spaces:
Running
Running
Update
Browse files
app.py
CHANGED
|
@@ -4,12 +4,14 @@ from __future__ import annotations
|
|
| 4 |
|
| 5 |
import argparse
|
| 6 |
import functools
|
|
|
|
| 7 |
import pathlib
|
| 8 |
import subprocess
|
| 9 |
|
| 10 |
-
|
| 11 |
-
subprocess.call('pip uninstall -y opencv-python
|
| 12 |
-
subprocess.call('pip
|
|
|
|
| 13 |
|
| 14 |
import cv2
|
| 15 |
import face_alignment
|
|
@@ -49,8 +51,12 @@ def detect(
|
|
| 49 |
|
| 50 |
res = image.copy()
|
| 51 |
for pts in preds:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
for pt in np.round(pts).astype(int):
|
| 53 |
-
cv2.circle(res, tuple(pt),
|
| 54 |
return res
|
| 55 |
|
| 56 |
|
|
|
|
| 4 |
|
| 5 |
import argparse
|
| 6 |
import functools
|
| 7 |
+
import os
|
| 8 |
import pathlib
|
| 9 |
import subprocess
|
| 10 |
|
| 11 |
+
if os.environ.get('SYSTEM') == 'spaces':
|
| 12 |
+
subprocess.call('pip uninstall -y opencv-python'.split())
|
| 13 |
+
subprocess.call('pip uninstall -y opencv-python-headless'.split())
|
| 14 |
+
subprocess.call('pip install opencv-python-headless==4.5.5.62'.split())
|
| 15 |
|
| 16 |
import cv2
|
| 17 |
import face_alignment
|
|
|
|
| 51 |
|
| 52 |
res = image.copy()
|
| 53 |
for pts in preds:
|
| 54 |
+
tl = pts.min(axis=0)
|
| 55 |
+
br = pts.max(axis=0)
|
| 56 |
+
size = (br - tl).max()
|
| 57 |
+
radius = max(2, int(3 * size / 256))
|
| 58 |
for pt in np.round(pts).astype(int):
|
| 59 |
+
cv2.circle(res, tuple(pt), radius, (0, 255, 0), cv2.FILLED)
|
| 60 |
return res
|
| 61 |
|
| 62 |
|