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:
|