admin commited on
Commit
3d3fab6
·
1 Parent(s): 9f1ed37
Files changed (2) hide show
  1. app.py +3 -3
  2. utils.py +2 -0
app.py CHANGED
@@ -56,9 +56,9 @@ class ValidImgDetector:
56
  detected_objects, out_im = predictor.recognize(image)
57
  has_child, has_female, has_male = False, False, False
58
  if len(detected_objects.ages) > 0:
59
- has_child = "是" if min(detected_objects.ages) < 18 else "否"
60
- has_female = "是" if "female" in detected_objects.genders else "否"
61
- has_male = "是" if "male" in detected_objects.genders else "否"
62
 
63
  return out_im[:, :, ::-1], has_child, has_female, has_male
64
 
 
56
  detected_objects, out_im = predictor.recognize(image)
57
  has_child, has_female, has_male = False, False, False
58
  if len(detected_objects.ages) > 0:
59
+ has_child = _L("是") if min(detected_objects.ages) < 18 else _L("否")
60
+ has_female = _L("是") if "female" in detected_objects.genders else _L("否")
61
+ has_male = _L("是") if "male" in detected_objects.genders else _L("否")
62
 
63
  return out_im[:, :, ::-1], has_child, has_female, has_male
64
 
utils.py CHANGED
@@ -32,6 +32,8 @@ ZH2EN = {
32
  "存在男性": "Has Male",
33
  "在线模式": "Online Mode",
34
  "网络图片链接": "Online Picture URL",
 
 
35
  }
36
 
37
 
 
32
  "存在男性": "Has Male",
33
  "在线模式": "Online Mode",
34
  "网络图片链接": "Online Picture URL",
35
+ "是": "Yes",
36
+ "否": "No",
37
  }
38
 
39