質問: Augmentor.Pipeline(images_dir) p.rotate(probability=0.7, max_left_rotation=5, max_right_rotation=5) p.sample(1400) return x_train,x_test
回答: 今更感がありますが、、、 同じ問題で困った人向けに回答を記載します。 ### エラーの原因について ``` # エラーの再現 np.arange(451584).reshape(128, 1176) ''' ValueError Traceback (most recent call last) in ----> 1 np.arange(451584).reshape(128, 1176) ValueError: cannot reshape array of size 451584 into shape (128,1176) ''' ``` np.reshape メソッドを呼び出すときは、reshape前後で配列の総サイズが一致していなければないようです。 今回の配列サイズは`451584`ということですが、reshape後のサイズ`128 * 1176 = 150528` になっているため、エラーが発生しています。 ### 対策 この問題を解消するためには、 * 配列の元のサイズを 150528 にする もしくは * `reshape((3, 128, 1176))`や`reshape((128, 3, 1176))`などする ことで、変換前後で要素の総数が同じになるよう調節しなければいけないようです。 (色の話をしているようなので、`reshape((3, 128, 1176))`か`reshape((3, 1176, 128))`が正しそう...)