動画から連番で画像を保存するプログラム

Question

###前提・実現したいこと
動画(wmvファイル)を読み込んで、静止画(jpgファイル)を連番で保存したいです。

###発生している問題・エラーメッセージ
画像が出力されない

###該当のソースコード

import os
import shutil
import cv2

def video_2_frames(video_file='./test.wmv', image_dir='./image_dir/', image_file='img_%s.jpg'):
    # Delete the entire directory tree if it exists.
    if os.path.exists(image_dir):
        shutil.rmtree(image_dir)

    # Make the directory if it doesn't exist.
    if not os.path.exists(image_dir):
        os.makedirs(image_dir)

    # Video to frames
    i = 0
    cap = cv2.VideoCapture(video_file)
    while(cap.isOpened()):
        flag, frame = cap.read()  # Capture frame-by-frame
        if flag == False:  # Is a frame left?
            break
        cv2.imwrite(image_dir+image_file % str(i).zfill(6), frame)  # Save a frame
        print('Save', image_dir+image_file % str(i).zfill(6))
        i += 1

    cap.release()  # When everything done, release the capture

###試したこと
実行すると
Process finished with exit code 0
となります

###補足情報(言語/FW/ツール等のバージョンなど)
より詳細な情報

Comments

[LouiS0616]

標準出力には適切な表示がありますか？

[testjojo]

本来であればSave ./image_dir/img_000000.jpg Save ./image_dir/img_000001.jpg...のように出力されてからProcess finished with exit code 0が出力されるはずなのですが、Process finished with exit code 0しか出力されません。

Answer 1

下記リンク先の動画(mp4)を保存して質問者様のコードを走らせると問題なく動作しました。
空撮動画

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import shutil
import cv2


def video_2_frames(video_file='./test.wmv', image_dir='./image_dir/',
				   image_file='img_%s.jpg'):
	# Delete the entire directory tree if it exists.
	if os.path.exists(image_dir):
		shutil.rmtree(image_dir)

	# Make the directory if it doesn't exist.
	if not os.path.exists(image_dir):
		os.makedirs(image_dir)

	# Video to frames
	i = 0
	cap = cv2.VideoCapture(video_file)
	while (cap.isOpened()):
		flag, frame = cap.read()  # Capture frame-by-frame
		if flag == False:  # Is a frame left?
			break
		cv2.imwrite(image_dir + image_file % str(i).zfill(6),
					frame)  # Save a frame
		print('Save', image_dir + image_file % str(i).zfill(6))
		i += 1

	cap.release()  # When everything done, release the capture


def main():
	video_2_frames(video_file="aerial.mp4")


if __name__ == '__main__':
	main()
"""
出力結果
Save ./image_dir/img_000000.jpg
Save ./image_dir/img_000001.jpg
Save ./image_dir/img_000002.jpg
Save ./image_dir/img_000003.jpg
Save ./image_dir/img_000004.jpg
Save ./image_dir/img_000005.jpg
Save ./image_dir/img_000006.jpg
...
Save ./image_dir/img_000266.jpg
Save ./image_dir/img_000267.jpg
Save ./image_dir/img_000268.jpg
Save ./image_dir/img_000269.jpg
Save ./image_dir/img_000270.jpg
Save ./image_dir/img_000271.jpg
Save ./image_dir/img_000272.jpg

Process finished with exit code 0
"""

Comments

[testjojo]

解答ありがとうございます。
同じ動画を保存して試しましたがやはりだめでした。ソースコードと同じディレクトリにおいているのですが、うまく行きません

[wakame]

ちょっと気になったのですが質問者様の環境はタグ情報からpython2.7ということでよろしいでしょうか？でしたらimport cv2のみ書いたコードを走らせるとどうなるか教えてください。

[testjojo]

python2.7です。
Process finished with exit code 0と出力されます

[wakame]

ちょっと疑問なのですがpython2.7であるとすると質問者様のソースコードはpython2系対応しておらずエラーになるはずです、というのもprint()はpython3以降でしか使えないはずです。 (from __future__ import print_functionをしていれば別ですが・・・)

[wakame]

上記踏まえて確認です、質問者様のソースコード時はどのようにコードを実行されてますか？>> python ***.pyのように実行されていると思うのでターミナル等に入力しているコマンドを教えてください。

[LouiS0616]

> というのもprint()はpython3以降でしか使えないはず
横から失礼します、タプルを対象とするprint文と解釈されるので実行可能かと。

[wakame]

御指摘ありがとうございます、なるほど確かにタプルと認識されて表示されるようですね。( 手元に環境がないのでpaiza.ioで確認してみました。)