【Python】OpenCVを使って動画に映る人の上に別の画像を貼る【フリーランス案件】

import numpy as np
import cv2
import time

# 体全体のカスケードファイル(パスはカスケード分類器のxmlファイルを指定)
fullbody_detector =　cv2.CascadeClassifier("/Users/hts/Documents/python_work/work1/haarcascade_fullbody.xml")
# サンプル動画
cap = cv2.VideoCapture('sample.avi')
# 人の上に表示する画像
mark = cv2.imread('mark.png')
height, width = mark.shape[:2]

# Shi-Tomasiのコーナー検出パラメータ
feature_params = dict( maxCorners = 10,
                       qualityLevel = 0.1,
                       minDistance = 1,
                       blockSize =  7)

# Lucas-Kanade法のパラメータ
lk_params = dict( winSize  = (10,10),
                  maxLevel = 2,
                  criteria = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 0.03))

# 画像保存用


# 最初のフレームの処理
end_flag, frame = cap.read()
# グレースケール変換
gray_prev = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
#動画サイズ取得
width_org = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height_org = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
#フレームレート取得
fps = cap.get(cv2.CAP_PROP_FPS)

#フォーマット指定
fmt = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
#注）グレースケールの画像を出力する場合は第5引数に0を与える
writer = cv2.VideoWriter('output.mp4', fmt, fps, (width_org, height_org))
writer_2 = cv2.VideoWriter('origin.mp4', fmt, fps, (width_org, height_org))
# 元の配列と同じ形にして0を代入
mask = np.zeros_like(frame)

while(end_flag):
    # グレースケールに変換
    writer_2.write(frame)
    gray_next = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    start = time.time()  
    # 全身の人を検出 
    # minSize:物体が取り得る最小サイズ。これよりも小さい物体は無視される
    # minNeighbors:物体候補となる矩形は，最低でもこの数だけの近傍矩形を含む
    body = fullbody_detector.detectMultiScale(gray_next,scaleFactor=1.1, minNeighbors=3, minSize=(40, 40))
    end = time.time()  
    # 検出時間を表示  
    print("{} : {:4.1f}ms".format("detectTime", (end - start) * 1000))

   
    
    # 人検出した数表示のため変数初期化
    human_cnt = 0
    # 人検出した部分を長方形で囲う
   
    
    for (x, y, w, h) in body:
        cv2.rectangle(frame, (x, y),(x+w, y+h),(0,255,0),2)
        frame[y+30:y+30+height, x:x+height]=mark
        # 人検出した数を加算
        human_cnt += 1
        

    # 人検出した数を表示
    cv2.putText(frame, "Human Cnt:{}".format(int(human_cnt)),(10,550), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0,255,0), 1, cv2.LINE_AA)
    # ウィンドウに表示
    cv2.imshow('plant_programer', frame)
    writer.write(frame)
    

   # ESCキー
    k = cv2.waitKey(1)
    if k == 27:
        break

    # 次のフレーム、ポイントの準備
    gray_prev = gray_next.copy()
    end_flag, frame = cap.read()

# 終了処理
writer.release()
cv2.destroyAllWindows()
cap.release()