前提・実現したいこと
作成中のプログラムの機能で動画のシーンチェンジ検出を実装したいと思い、OpenCVのソースコードを探したところ、わかりやすそうなコードが見つかったので参考にしようと思いました。自分のプログラムはC++で書いているので、元となるPythonのコードから自力で書き換えました。
実行したところエラーや警告は出ないのですが、1フレーム目から全てのフレームがシーンチェンジしていると判定されてしまいました。PythonもOpenCVも初心者なので理解が及んでいない部分もあり、解決が難しく困っています。
参考にしたのは以下のサイトです。
Python+OpenCVでアニメのカット検出 - https://nixeneko.hatenablog.com/entry/2017/09/07/000000
該当のソースコード
参考にしたコード
Python
1import cv2 2import numpy as np 3 4INFILE = "gochiusa#1.mp4" # movie file 5THRESH = 55.55679398148148 # threshold 6 7ESC_KEY = 27 # Escキー 8INTERVAL= 1 # 待ち時間 9 10WINNAME = "test" 11cv2.namedWindow(WINNAME) 12 13class MovieIter(object): #動画のフレームを返すイテレータ 14 def __init__(self, moviefile, size=None, inter_method=cv2.INTER_AREA): 15 #TODO: check if moviefile exists 16 self.org = cv2.VideoCapture(moviefile) 17 self.framecnt = 0 18 self.size = size #frame size 19 self.inter_method = inter_method 20 def __iter__(self): 21 return self 22 def __next__(self): 23 self.end_flg, self.frame = self.org.read() 24 if not self.end_flg: # end of the movie 25 raise StopIteration() 26 self.framecnt+=1 27 if self.size: # resize when size is specified 28 self.frame = cv2.resize(self.frame, self.size, interpolation=self.inter_method) 29 return self.frame 30 def __del__(self): # anyway it works without destructor 31 self.org.release() 32 33def MSE(pic): # mean square error 34 return np.mean(np.square(pic)) 35 36def MAE(pic): # mean absolute error 37 return np.mean(np.abs(pic)) 38 39def main(): 40 picsize = (64, 36) 41 42 frame_cnt = 0 43 frame_ultima = np.zeros((*picsize[::-1], 3)) # create empty image 44 45 for frame in MovieIter(INFILE, None): 46 frame_penult = frame_ultima 47 frame_ultima = cv2.resize(frame, picsize, interpolation=cv2.INTER_AREA) #指定サイズに縮小 48 49 cv2.imshow(WINNAME, frame) 50 key = cv2.waitKey(1) # quit when esc-key pressed 51 if key == ESC_KEY: 52 break 53 54 #差分画像作成 55 diff = frame_ultima.astype(np.int) - frame_penult.astype(np.int) 56 57 if MAE(diff)>=THRESH: #閾値よりMAEが大きい場合、カットと判定 58 print("Cut detected!: frame {}".format(frame_cnt)) 59 60 frame_cnt+=1 61 62 63if __name__ == "__main__": 64 main()
書き換えたコード
C++
1#include "stdafx.h" 2#include <opencv2/opencv.hpp> 3 4using namespace cv; 5 6int MAE(cv::Mat pic) { 7 return sizeof(abs(pic)); 8} 9 10int main(void) { 11 12 cv::VideoCapture INFILE; 13 INFILE.open("C:/Users/Admin/Downloads/gochiusa#1.mp4"); 14 15 double THRESH = 55.55679398148148; 16 17 cv::Mat frame; 18 19 String WINNAME = "test"; 20 21 int frame_cnt = 0; 22 23 double picsize[] = { 64,36 }; 24 25 cv::Mat frame_ultima; 26 27 for (;;) { 28 INFILE >> frame; 29 30 if (frame.empty() == true) { 31 break; 32 } 33 34 cv::imshow(WINNAME, frame); 35 cv::waitKey(1); 36 37 cv::Mat frame_penult = frame_ultima; 38 cv::resize(frame, frame_ultima, cv::Size(), picsize[0], picsize[1]); 39 40 cv::Mat diff = frame_ultima - frame_penult; 41 42 if (MAE(diff) >= THRESH) { 43 printf("Cut detected!: frame {%d}\n", frame_cnt); 44 45 frame_cnt += 1; 46 } 47 } 48}
補足情報(FW/ツールのバージョンなど)
Visual Studio 2017
OpenCV 4.1.2
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
退会済みユーザー
2019/11/12 10:06