Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,8 +44,12 @@ def cbiwm(image, mask, coordinates):
|
|
| 44 |
def processFaceDetails(image):
|
| 45 |
ret = getPersonDetail(image)
|
| 46 |
data = faceDetectionModal(image)
|
| 47 |
-
|
| 48 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
cropped_image = image.crop(crop_box)
|
| 50 |
facedata = faceModal(cropped_image)
|
| 51 |
for imask in facedata:
|
|
|
|
| 44 |
def processFaceDetails(image):
|
| 45 |
ret = getPersonDetail(image)
|
| 46 |
data = faceDetectionModal(image)
|
| 47 |
+
if len(data) > 1:
|
| 48 |
+
cordinates = data[1]["box"]
|
| 49 |
+
crop_box = (data[1]["box"]['xmin'], data[1]["box"]['ymin'], data[1]["box"]['xmax'], data[1]["box"]['ymax'])
|
| 50 |
+
else:
|
| 51 |
+
cordinates = data[0]["box"]
|
| 52 |
+
crop_box = (data[0]["box"]['xmin'], data[0]["box"]['ymin'], data[0]["box"]['xmax'], data[0]["box"]['ymax'])
|
| 53 |
cropped_image = image.crop(crop_box)
|
| 54 |
facedata = faceModal(cropped_image)
|
| 55 |
for imask in facedata:
|