Pythonを使って動画から画像を抽出して保存してみた

Pocket

1 はじめに
2 動画→画像変換
3 まとめ
4 参考

はじめに

こんにちは．

はんぺんです．

今回は，動画から画像を抽出して保存する方法を説明しようと思います．

今回はテニスの動画の分析に用いたmp4のものを用いました．

拡張子のところ変えれば多分他の拡張しでもできるはず．

動作環境は以下のとおりです．

Ubuntu 16.04
Anaconda3
Python 3.5

動画→画像変換

ライブラリのインポート

import os
import cv2

1 2	import os import cv2

まずライブラリを2種類インポートします．

1つ目のimport osではosモジュールをインポートし，オペレーティングシステムと対話するための多くの関数が使えるようになります．

今回は，フォルダを参照してファイル名を取得するのにこのosモジュールを使用しました．

10. 標準ライブラリミニツアー — Python 3.6.3 ドキュメント

2つ目のimport cv2によって，openCVが使えるようになります．

これによって，画像や動画を読み込むこんだり，フレームレートやサイズを取得したりすることができるようになります．

OpenCVとは？最新3.0の新機能概要とモジュール構成

＞OpenCV（正式名称： Open Source Computer Vision Library）は、オープンソースのコンピューター・ビジョン・ライブラリです。コンピューターで画像や動画を処理するのに必要な、さまざま機能が実装されており、BSDライセンスで配布されていることから学術用途だけでなく商用目的でも利用できます。加えて、マルチプラットフォーム対応されているため、幅広い場面で利用されていることが特徴です。

動画の情報

movie_name = 'tennis_006'
movie = './movie/'+movie_name+'.mp4'

1 2	movie_name = 'tennis_006' movie = './movie/'+movie_name+'.mp4'

動画のファイル名とパスを指定します．

プログラムがある階層にmovieというフォルダを作り，その中に動画(mp4)を置いていれば上記のように記載すれば取得できます．

画像生成

count = 0
cap = cv2.VideoCapture(movie)

while True:
    ret, frame = cap.read()
    if ret == True:
        count += 1
        cv2.imwrite('./image/sample_image/' + movie_name + '_' + str("{0:05d}".format(count)) +'.jpg', frame)
    else:
        break