Update ageidentification.py

Age Level Identification/ageidentification.py

-hjk
\ No newline at end of file
+import cv2
+def faceBox(faceNet, frame):
+    frameHight = frame.shape[0]
+    frameWidth = frame.shape[1]
+    blob = cv2.dnn.blobFromImage(frame, 1.0, (227,227), [104,117,123], swapRB=False)
+    faceNet.setInput(blob)
+    detection = faceNet.forward()
+    bboxs = []
+    for i in range(detection.shape[2]):
+        confidence=detection[0, 0, i, 2]
+        if confidence > 0.7:
+           x1 = int(detection[0, 0, i, 3] * frameWidth)
+           y1 = int(detection[0, 0, i, 4] * frameHight)
+           x2 = int(detection[0, 0, i, 5] * frameWidth)
+           y2 = int(detection[0, 0, i, 6] * frameHight)
+           bboxs.append([x1,y1,x2,y2])
+           cv2.rectangle(frame, (x1,y1), (x2,y2),(0,255,0),0)
+    return frame, bboxs
+
+faceProto = "opencv_face_detector.pbtxt"
+faceModel = "opencv_face_detector_uint8.pb"
+ageProto = "age_deploy.prototxt"
+ageModel = "age_net.caffemodel"
+genderProto = "gender_deploy.prototxt"
+genderModel = "gender_net.caffemodel"
+faceNet = cv2.dnn.readNet(faceModel, faceProto)
+ageNet = cv2.dnn.readNet(ageModel, ageProto)
+genderNet = cv2.dnn.readNet(genderModel, genderProto)
+MODEL_MEAN_VALUES = (78.4263377603, 87.7689143744, 114.895847746)
+ageList = ['(0-2)', '(4-6)', '(8-12)', '(15-20)', '(25-32)', '(38-43)', '(48-53)', '(60-100)']
+genderList = ['Male', 'Female']
+video = cv2.VideoCapture(0)
+while True:
+    ret,frame = video.read()
+    frame, bboxs = faceBox(faceNet, frame)
+    for bbox in bboxs:
+        face = frame[bbox[1]:bbox[3], bbox[0]:bbox[2]]
+        blob = cv2.dnn.blobFromImage(face, 1.0, (227,227),MODEL_MEAN_VALUES,swapRB=False)
+        genderNet.setInput(blob)
+        genderPred = genderNet.forward()
+        gender = genderList[genderPred[0].argmax()]
+        ageNet.setInput(blob)
+        agePred = ageNet.forward()
+        age = ageList[agePred[0].argmax()]
+        lable = "{}, {}".format(gender,age)
+        cv2.putText(frame,lable,(bbox[0],bbox[1]-10), cv2.FONT_HERSHEY_SIMPLEX, 0.8, (255,255,255), 2)
+    cv2.imshow("Age Detector", frame)
+    k = cv2.waitKey(1)
+    if k == ord('q'):
+        break
+video.release()        
+cv2.distroyAllWindows()