OpenCV3.3.1 顔認識とトラッキングのサンプルプログラム #2

取り敢えず、プログラムは、こちらです。

''' Created on 2017/12/25 @author: nishi face_auto_tracking2.py OpenCV 3.3.1 Face detect and Tracker ''' import numpy as np import cv2 import sys #(major_ver, minor_ver, subminor_ver) = (cv2.__version__).split('.') def my_tracker_init(frame): # Read first frame. global bbox rc=1 w_cur=0 gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) faces = face_cascade.detectMultiScale(gray, 1.3, 5) for (x,y,w,h) in faces: if(w >= w_cur): # x,y,width,height bbox = (x,y,w,h) print("x=%d" % (x)) print("y=%d" % (y)) print("w=%d" % (w)) print("h=%d" % (h)) w_cur=w if w_cur > 0: rc=0 return rc def my_tracker_create(tracker_type): global tracker if tracker_type == 'BOOSTING': tracker = cv2.TrackerBoosting_create() if tracker_type == 'MIL': tracker = cv2.TrackerMIL_create() if tracker_type == 'KCF': tracker = cv2.TrackerKCF_create() if tracker_type == 'TLD': tracker = cv2.TrackerTLD_create() if tracker_type == 'MEDIANFLOW': tracker = cv2.TrackerMedianFlow_create() if tracker_type == 'GOTURN': tracker = cv2.TrackerGOTURN_create() if __name__ == '__main__' : face_cascade = cv2.CascadeClassifier('/app/Anaconda3/envs/opencv_3X/Library/etc/haarcascades/haarcascade_frontalface_default.xml') # Set up tracker. # Instead of MIL, you can also use tracker_types = ['BOOSTING', 'MIL','KCF', 'TLD', 'MEDIANFLOW', 'GOTURN'] # KFC -> 8 fps #tracker_type = tracker_types[2] # BOOSTING -> 43 fps #tracker_type = tracker_types[0] # MIL -> 10 fps #tracker_type = tracker_types[1] # TLD -> 5 #tracker_type = tracker_types[3] # MEDIANFLOW -> 72～73 fps tracker_type = tracker_types[4] # GOTURN -> error #tracker_type = tracker_types[5] my_tracker_create(tracker_type) # Read video #video = cv2.VideoCapture("videos/chaplin.mp4") # changed by nishi video = cv2.VideoCapture(0) # Exit if video not opened. if not video.isOpened(): print("Could not open video") sys.exit() # Define an initial bounding box # x,y,width,height bbox = (287, 23, 86, 320) # Uncomment the line below to select a different bounding box #bbox = cv2.selectROI(frame, False) tracking=0; while True: # Read a new frame ok, frame = video.read() if not ok: break if(tracking ==0 or tracking==2): rc=my_tracker_init(frame) if(rc==0): if(tracking == 0): # Initialize tracker with first frame and bounding box ok = tracker.init(frame, bbox) else: my_tracker_create(tracker_type) # Initialize tracker with first frame and bounding box ok = tracker.init(frame, bbox) tracking=1 else: # Start timer timer = cv2.getTickCount() # Update tracker ok, bbox = tracker.update(frame) # Calculate Frames per second (FPS) fps = cv2.getTickFrequency() / (cv2.getTickCount() - timer); # Draw bounding box if ok: # Tracking success p1 = (int(bbox[0]), int(bbox[1])) p2 = (int(bbox[0] + bbox[2]), int(bbox[1] + bbox[3])) cv2.rectangle(frame, p1, p2, (255,0,0), 2, 1) else : # Tracking failure cv2.putText(frame, "Tracking failure detected", (100,80), cv2.FONT_HERSHEY_SIMPLEX, 0.75,(0,0,255),2) tracking=2 # Display tracker type on frame cv2.putText(frame, tracker_type + " Tracker", (100,20), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50,170,50),2); # Display FPS on frame cv2.putText(frame, "FPS : " + str(int(fps)), (100,50), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50,170,50), 2); # Display result cv2.imshow("Tracking", frame) # Exit if ESC pressed k = cv2.waitKey(1) & 0xff if k == 27 : break #if(k != 0 and k != 255): # print("k=%d" % (k)) # 't' key -> my_tracker_init() if k == 116 : tracking=2

'ESC'キー入力で、終了します。
プログラム開始で、自動的にカスケードファイルを用いて、顔の認識をします。
認識が旨くいけば、後は、 Tracking を行います。
注) 't' キーを入力すると、再度、顔の認識をします。うまく顔を認識できない場合は、
どうぞ。