fit_generator()を用いて，深層学習モデルの学習途中に出たエラー

前提・実現したいこと

kerasを用いて3DCNNを実装して，動画の自動分類を行うとしています.
モデルは,kerasで構築.ジェネレータは自作のものです.
環境は,google colabを利用しています.

あとは，学習をするだけの状態になったのでfit_generator()で学習を開始したところ，以下のエラー文がでてきました.
途中までは学習が済んでいるのに,6/165でいきなりinputの形が違う，というエラーがでた理由がわかりません.
何かアドバイスをよろしくお願いします.

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

 6/165 [>.............................] - ETA: 7:57 - loss: 16.1181 - acc: 0.0000e+00Traceback (most recent call last):
 File "main.py", line 64, in <module>
    epochs=1)
...
...
...
ValueError: Error when checking input: expected conv3d_1_input to have 5 dimensions, but got array with shape (8, 1)

該当のソースコード

python
1validation_ratio = 0.3
2
3num_sample = len(lists)
4sample_idxs = range(0,num_sample)
5sample_idxs = np.random.permutation(sample_idxs)
6train_sample_idx = sample_idxs[0:int((1-validation_ratio)*num_sample)]
7valid_sample_idx = sample_idxs[int((1-validation_ratio)*num_sample):]
8train_generator = video_generator.generate_train_from_directory(8,'data',train_sample_idx)
9valid_generator = video_generator.generate_valid_from_directory(8,'data',valid_sample_idx)
10
11model = C3d_model.get_model(summary=True)
12
13print("学習開始")
14
15history = model.fit_generator(train_generator,steps_per_epoch=len(train_sample_idx)/8,
16                                epochs=1)
17                                
18acc = model.evaluate_generator(valid_generator,steps = 10,
19                                verbos=1)
20
21print("loss: ",acc[0])
22print("accuracy: " ,acc[1])

試したこと

バッチサイズを32→8にすることで一回の計算負荷を小さくしてみた.

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

行動規範の内容に同意します

回答2件

ベストアンサー

おそらく、6 iteration 目に読み込んだサンプルのどれかがモデルの入力の形状に合っていないということではないでしょうか？
フレーム数が他のと違う、フレームの大きさが他のと違うなど

投稿2019/11/19 08:39

tiitoi

総合スコア21956

ruto844

2019/12/03 12:06

インプットサンプルの形状を調べて見たところ, モデルへの入力形式(batch_size, 20, 480,640,3)に対して, 20(動画の秒数)でなく,20秒未満の動画がデータセット内に存在することがわかりました. データセットから秒数の異なる(20秒でない)動画データを覗くことによりエラーを改善することができました.ありがとうございました.

行動規範の内容に同意します