Python OpenCV ndarray型のimageのVideoWriterへの書き込み

Question

Python OpenCV ndarray型のimageのVideoWriterへの書き込み

カメラから取得した動画をAVIとして保存する機能を作成しております。
産業用カメラのuEyeカメラから流れてきた画像をOpenCVのVideoWriterに保存する事を考えており、以下のサンプルコードを元に作成しております。
https://github.com/dcabecinhas/pyueye

現在、カメラから取得したデータをGUIで表示する部分は動いておりますが、AVIへの書き込みが上手く行えておりません。

発生している問題・エラーメッセージ

.writeをする時に上手く保存できていない。
以下に似たようなエラーを報告している方がおり、入力された画像のサイズなどがVideoWriterと違うと保存されない事がわかりました。
https://teratail.com/questions/63352

その後も画像のサイズや型を色々変えてやってみたのですが上手くいきませんでした。

該当のソースコード

元となるコードのmain.pyのprocess_imageの所でAVI化を行っております。

#!/usr/bin/env python
# Copyright省略

from pyueye_example_camera import Camera
from pyueye_example_utils import FrameThread
from pyueye_example_gui import PyuEyeQtApp, PyuEyeQtView
from PyQt4 import QtGui

from pyueye import ueye

import cv2
import numpy as np
import datetime
import os

size_width = 1280
size_height = 1024
fps = 30.0
codec = "DIVX"
aviname = "output.avi"
avi = cv2.VideoWriter()

def init_avi():
	print("init")
	aviname = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S.avi")
	fourcc = cv2.cv.CV_FOURCC(*codec)
	avi = cv2.VideoWriter(aviname, fourcc, fps, (size_width, size_height))

def done_avi():
	print("done")
	avi.release()
	cv2.destroyAllWindows()

def process_image(self, image_data):
	print("process")
    # reshape the image data as 1dimensional array
	image = image_data.as_1d_image()
	print(type(image_data)) // <type 'instance'>
	print(type(image)) // <type 'numpy.ndarray'>

	qtgui = QtGui.QImage(image.data,
		image_data.mem_info.width,
		image_data.mem_info.height,
		QtGui.QImage.Format_RGB888)
	avi.write(cv2.resize(image, (size_width,size_height)))

    # show the image with Qt
	return qtgui

def main():
	# we :wneed a QApplication, that runs our QT Gui Framework
	app = PyuEyeQtApp()

	init_avi()

	# a basic qt window
	view = PyuEyeQtView()
	view.show()
	view.user_callback = process_image

	# camera class to simplify uEye API access
	cam = Camera()
	cam.init()
	cam.set_colormode(ueye.IS_CM_BGR8_PACKED)
	cam.set_aoi(0,0, size_width, size_height)
	cam.alloc()
	cam.capture_video()

	# a thread that waits for new images and processes all connected views
	thread = FrameThread(cam, view)
	thread.start()

	# cleanup
	print("clean up")
	app.exit_connect(thread.stop)
	app.exec_()

	print("thread")
	thread.stop()
	thread.join()

	print("cam")
	cam.stop_video()
	cam.exit()

	done_avi()

if __name__ == "__main__":
	main()

補足情報

OS: Ubuntu 16.04.5 LTS
Python: Python 2.7.12
OpenCV: 2.4.9.1
カメラ: USB 3 uEye CP Rev. 2

#####どうぞよろしくお願いいたします。

Answer 1

以下を別のコーデックで試してみてはどうでしょうか。

codec = "DIVX"

image の型は image.dtype で uint8 になっていることを確認してください。

---

追記

VideoWriter がそもそも使えるのか確認してみてはどうでしょうか

vtest.avi

import cv2

# Video Reader を作成
cap = cv2.VideoCapture('vtest.avi')
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS)

# Video Writer を作成
fourcc = cv2.VideoWriter_fourcc(*'DIVX')
writer = cv2.VideoWriter('output.avi', fourcc, fps, (width, height))

while True:
    ret, frame = cap.read()
    if not ret:
        break
    writer.write(frame)

writer.release()
cap.release()
cv2.destroyAllWindows()

追記

おそらく原因がわかりました。
グローバル変数で avi = cv2.VideoWriter() としたあと、init_avi() 内で avi = cv2.VideoWriter(aviname, fourcc, fps, (size_width, size_height)) としていますが、これはローカル変数のため、関数を抜けた段階で破棄されます。
グローバル変数の avi に代入されているわけではありません。

関数内でグローバル変数に代入する場合は global avi とする必要があります。
こちらの記事 で言語仕様を確認してください。

そのあと、process_image() 内で avi を参照したときは、グローバル変数の avi = cv2.VideoWriter() を参照しています。
そのため、動画は出力されません。
print(avi) でオブジェクトの ID を確認してみてください。

実証コード

import datetime

import cv2

avi = cv2.VideoWriter()
print(avi)  # <VideoWriter 0000016D5B99CAD0>

def init_avi():
    aviname = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S.avi")
    fourcc = cv2.VideoWriter_fourcc(*"DIVX")
    avi = cv2.VideoWriter("output.avi", fourcc, 30., (1280, 1024))
    print(avi)  # <VideoWriter 0000016D5E1BF1B0>


def func():
    print(avi)  # <VideoWriter 0000016D5B99CAD0>

init_avi()
func()

Comments

[k_egashi]

ご回答ありがとうございます。
codecをXVID, MPEG, DIV3, MJPG等色々変更して試してみましたが挙動変わらずでした。
```
print(image.dtype)
```
ではuint8となっており問題なさそうでした。
それ以外に確認できる項目等あればご教授頂けますか。

[tiitoi]

VideoWriter は使えたことはあるのでしょうか？
もし、まだないのであれば、追記したコードで確認してみてください。

[k_egashi]

VideoWriter自体は使えており、追記頂いたコードも問題なく動きました。
よろしくお願いします。

[tiitoi]

五月雨式で申し訳ないですが、image = image_data.as_1d_image() の直後の image.shape はどうなっていますか？

[k_egashi]

こちらこそお手を煩わせてしまい申し訳ございません。
image.shapeの実行結果は以下になりました。
(1024, 1280, 3)

[tiitoi]

おそらく原因がわかったので追記しました。
もし違ったら教えてください。

[k_egashi]

無事動かすことができました！確かにIDが違うものでした。
非常に困っていたので助かりました。本当にありがとうございました！