5m4ck3r commited on
Commit
dfc3574
·
1 Parent(s): c1c2031

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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
- cordinates = data[1]["box"]
48
- crop_box = (data[1]["box"]['xmin'], data[1]["box"]['ymin'], data[1]["box"]['xmax'], data[1]["box"]['ymax'])
 
 
 
 
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: