前提・実現したいこと
OpenCVで映像を読み込み、動体を検出したシーンを画像として保存するようにしたいです。
発生している問題・エラーメッセージ
error: OpenCV(4.4.0) C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-71670poj\opencv\modules\imgproc\src\resize.cpp:3929: error: (-215:Assertion failed) !ssize.empty() in function 'cv::resize'
該当のソースコード
python
1import cv2 2save_path = "./sample" 3def main(): 4 # 映像の取得 5 cap = cv2.VideoCapture("./movie.mp4") 6 img1 = img2 = img3 = get_image(cap) 7 # 閾値の設定 8 th = 50 9 num = 1 10 while True: 11 ret, frame = cap.read() 12 if not ret: 13 break 14 # Enterキーが押されたら終了 15 if cv2.waitKey(1) == 13: break 16 diff = check_image(img1, img2, img3) 17 # 差分がthの値以上なら動きがあったと判定 18 cnt = cv2.countNonZero(diff) 19 if cnt > th: 20 print("動体を検出") 21 cv2.imshow('PUSH ENTER KEY', img3) 22 # 閾値以上の写真を画像として保存 23 cv2.imwrite(save_path + str(num) + ".jpg", img3) 24 num += 1 25 else: 26 cv2.imshow('PUSH ENTER KEY', diff) 27 # 比較用の画像を保存 28 img1, img2, img3 = (img2, img3, get_image(cap)) 29 # 後始末 30 cap.release() 31 cv2.destroyAllWindows() 32 33# 画像に動きがあったか調べる関数 34def check_image(img1, img2, img3): 35 # グレイスケール画像に変換 36 gray1 = cv2.cvtColor(img1, cv2.COLOR_RGB2GRAY) 37 gray2 = cv2.cvtColor(img2, cv2.COLOR_RGB2GRAY) 38 gray3 = cv2.cvtColor(img3, cv2.COLOR_RGB2GRAY) 39 # 絶対差分を調べる 40 diff1 = cv2.absdiff(gray1, gray2) 41 diff2 = cv2.absdiff(gray2, gray3) 42 # 論理積を調べる 43 diff_and = cv2.bitwise_and(diff1, diff2) 44 # 白黒二値化 45 _, diff_wb = cv2.threshold(diff_and, 30, 255, cv2.THRESH_BINARY) 46 # ノイズの除去 47 diff = cv2.medianBlur(diff_wb, 5) 48 return diff 49 50# 映像から画像を取得する 51def get_image(cap): 52 img = cap.read()[1] 53 img = cv2.resize(img, (600, 400)) 54 print(img.shape) 55 return img 56main()
補足情報(FW/ツールのバージョンなど)
プログラミング初学者ですが、ネットで調べたWebカメラを使ってリアルタイムに動体を検出するコードを参考に作ったのでコードに大きな間違いはないはずなのですが、、。
もしほかにいい構文があれば教えて頂きたいです。
回答2件
あなたの回答
tips
プレビュー