@@ -24,7 +24,7 @@ def highlightFace(net, frame, conf_threshold=0.7):
24
24
25
25
26
26
parser = argparse .ArgumentParser ()
27
- parser . add_argument ( '--image' )
27
+
28
28
29
29
args = parser .parse_args ()
30
30
@@ -36,14 +36,14 @@ def highlightFace(net, frame, conf_threshold=0.7):
36
36
genderModel = "gender_net.caffemodel"
37
37
38
38
MODEL_MEAN_VALUES = (78.4263377603 , 87.7689143744 , 114.895847746 )
39
- ageList = ['(0-2)' , '(4-6)' , '(8 -12)' , '(15 -20)' , '(25 -32)' , '(38 -43)' , '(48 -53)' , '(60 -100)' ]
39
+ ageList = ['(0-2)' , '(4-6)' , '(7 -12)' , '(12 -20)' , '(21 -32)' , '(33 -43)' , '(44 -53)' , '(54 -100)' ]
40
40
genderList = ['Male' ,'Female' ]
41
41
42
42
faceNet = cv2 .dnn .readNet (faceModel ,faceProto )
43
43
ageNet = cv2 .dnn .readNet (ageModel ,ageProto )
44
44
genderNet = cv2 .dnn .readNet (genderModel ,genderProto )
45
45
46
- video = cv2 .VideoCapture (args . image if args . image else 0 )
46
+ video = cv2 .VideoCapture (0 )
47
47
padding = 20
48
48
while cv2 .waitKey (1 )< 0 :
49
49
hasFrame ,frame = video .read ()
@@ -72,4 +72,4 @@ def highlightFace(net, frame, conf_threshold=0.7):
72
72
print (f'Age: { age [1 :- 1 ]} years' )
73
73
74
74
cv2 .putText (resultImg , f'{ gender } , { age } ' , (faceBox [0 ], faceBox [1 ]- 10 ), cv2 .FONT_HERSHEY_SIMPLEX , 0.8 , (0 ,255 ,255 ), 2 , cv2 .LINE_AA )
75
- cv2 .imshow ("Detecting age and gender: " , resultImg )
75
+ cv2 .imshow ("Detecting age and gender" , resultImg )
0 commit comments