Pythonを使用してビデオ内の画像を検索

Question

私はこれについて正しい方向に進むのか、もっと効率的な方法があるのか​​疑問に思っていました。

ビデオの1フレームごとにこのイメージがその内側のどこかに格納されているように（フルサイズのフレームではなく小さなもの）、ビデオ内のイメージを検索しようとしています。

は、現在のような絵にビデオを引っ張っ：

import cv2 
vidcap = cv2.VideoCapture('My_Video.mp4') 
success,image = vidcap.read() 
count = 0 
success = True 
while success: 
    success,image = vidcap.read() 
    print ('Read a new frame: ', success) 
    cv2.imwrite("frame%d.jpg" % count, image)  # save frame as JPEG file 
    count += 1 

はその後のようなそれらすべてをループ：

import cv2 
import numpy as np 
from matplotlib import pyplot as plt 

img_rgb = cv2.imread('frame1.png') 
img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY) 
template = cv2.imread('small_icon_I_am_looking_for.png',0) 
w, h = template.shape[::-1] 

res = cv2.matchTemplate(img_gray,template,cv2.TM_CCOEFF_NORMED) 
threshold = 0.8 
loc = np.where(res >= threshold) 
for pt in zip(*loc[::-1]): 
    cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0,0,255), 2) 

cv2.imwrite('res.png',img_rgb) 

は、おそらく画像の保存をスキップする方法はありますか？私は何千時間ものビデオを通してこれをやっており、必要ではないかもしれない時間を使って感じるフレームをすべて保存して削除しています。毎回画像を保存する必要なしに、これをどのように検索することができますか？これは、私が何を意味するかの一例である再生されているスーパーマリオの映像があったと言う、それがこのコインを探します：

、そのようにそれを検出します。

これは現在機能していますが、より良い方法を探しています。

Answer 1

私があなたに誤解していない場合は、以下が有効です。全体として、あなたのコードはあなたが求めていることをするのに必要な最小限の変更だけで書かれています。あなたのwhileループの構造のため、最初のフレームを破棄する問題もありました。これを避けるための良い方法は、ループと半分/真のコンストラクトです：

import cv2 
import numpy as np 
from matplotlib import pyplot as plt 

def process_img(img_rgb, template, count): 
    img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY) 

    w, h = template.shape[::-1] 

    res = cv2.matchTemplate(img_gray,template,cv2.TM_CCOEFF_NORMED) 
    threshold = 0.8 
    loc = np.where(res >= threshold) 
    for pt in zip(*loc[::-1]): 
     cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0,0,255), 2) 

    # This will write different res.png for each frame. Change this as you require 
    cv2.imwrite('res{0}.png'.format(count),img_rgb) 


def main(): 
    vidcap = cv2.VideoCapture('My_Video.mp4') 
    template = cv2.imread('small_icon_I_am_looking_for.png',0) # open template only once 
    count = 0 
    while True: 
     success,image = vidcap.read() 
     if not success: break   # loop and a half construct is useful 
     print ('Read a new frame: ', success) 
     process_image(image, template, count) 
     count += 1