Chainerにおいて画像を扱う場合、ImageDatasetというクラスが用意されており、以下のように簡単に扱うことができます。
from chainer.datasets import ImageDataset image_files = ['1.png', '2.png'] train = ImageDataset(paths=image_files)
Datasetについては以下の記事にまとめていますので、興味があればご覧ください。
動画については用意されていないので、自分で独自のDatasetを用意する必要があります。なので、作りました。
使い方は、画像の場合と同じです。
from VideoDataset import VideoDataset video_files = ['1.mp4', '2.mp4'] train = VideoDataset(paths=video_files)
動画の読み込みにOpenCVとFFmepgを利用しているため予めインストールが必要です。インストールについては以下の記事を参考にどうぞ。
作ってみてから気づきましたが、動画を画像として読む込むだけなので、動画を事前に画像に変換しておいて、画像として読み込む方が良いかもしれません。
