Numpy配列の二つの画像データセットを一つにまとめたい

Question

kerasを使って画像認識を学習中です。

128*128ピクセルの犬と猫のカラー画像が8000枚ずつ入ったデータセットがあります。

dogs_train_datasets = dogs_dataset[:8000]
cats_train_datasets = cats_dataset[:8000]

print(cats_train_datasets.shape)
print(dogs_train_datasets.shape)

出力

(8000, 128, 128, 3)
(8000, 128, 128, 3)

cats_train_datasetsとdogs_train_datasetsを同じところに入れて(16000, 128, 128, 3)にするにはどうすれば良いでしょうか？

np.appendかなと思いましたが、うまくいきませんでした。

Answer 1

import numpyu as np
cd_train_datasets = np.concatenate((cats_train_datasets, dogs_train_datasets))

です。

Comments

[seyu0930]

ありがとうございました！