5. 実践編: MRI画像のセグメンテーションで自前のデータセットを用いたところ、エラーが出てしまいます。ご教授いただけると幸いです。

[mint]

お世話になっております。

この度セマンティックセグメンテーションを勉強・使用することになり、資料を拝見させていただいている者です。

5. 実践編: MRI画像のセグメンテーションに従い、全結合型ニューラルネットワークを構築する事が出来ました。

そこで、自分で用意したデータセットを用いて学習をさせようとしたのですが、学習を実行した段階で以下のエラーが出てしまっています。

エラーメッセージで調べましたが勉強不足もあり解決方法が分からず・・・

このフォーラムの設立意図とは少し違った質問になるかもしれません。大変心苦しいのですがご教授いただければと思っています。

エラーメッセージは以下の通りです。

Exception in main training loop: all the input array dimensions for the concatenation axis must match exactly, but along dimension 1, the array at index 0 has size 1 and the array at index 2 has size 4 Traceback (most recent call last): File "/usr/local/lib/python3.6/dist-packages/chainer/training/trainer.py", line 316, in run update() File "/usr/local/lib/python3.6/dist-packages/chainer/training/updaters/standard_updater.py", line 175, in update self.update_core() File "/usr/local/lib/python3.6/dist-packages/chainer/training/updaters/standard_updater.py", line 181, in update_core in_arrays = convert._call_converter(self.converter, batch, self.device) File "/usr/local/lib/python3.6/dist-packages/chainer/dataset/convert.py", line 73, in _call_converter return converter(batch, device) File "/usr/local/lib/python3.6/dist-packages/chainer/dataset/convert.py", line 58, in wrap_call return func(*args, **kwargs) File "/usr/local/lib/python3.6/dist-packages/chainer/dataset/convert.py", line 223, in concat_examples [example[i] for example in batch], padding[i]))) File "/usr/local/lib/python3.6/dist-packages/chainer/dataset/convert.py", line 254, in _concat_arrays [array[None] for array in arrays]) File "<__array_function__ internals>", line 6, in concatenate Will finalize trainer extensions and updater before reraising the exception.

使用するデータセットは元画像が64×64のグレースケール、ラベル画像が64×64の2値画像です。拡張子はPNG形式です。

よろしくお願いいたします。