ノイズの多い手書き数字からノイズを除去するために、Denoising Autoencoderを作成した。
次に、作成したモデルに自分で用意したノイズの多い手書き数字を入力し、ノイズを除去したいと思います。そのためのコードを書きましたが、エラーが発生し、ノイズ除去後の画像が表示されません。
ノイズの多い手書き数字からノイズを除去するために、以下のようなDenoising Autoencoderを作成した。
python
1from keras.datasets import mnist 2import numpy as np 3import matplotlib.pyplot as plt 4import keras 5from keras import layers 6from keras.callbacks import TensorBoard 7 8(x_train, _), (x_test, _) = mnist.load_data() 9 10x_train = x_train.astype('float32') / 255. 11x_test = x_test.astype('float32') / 255. 12x_train = np.reshape(x_train, (len(x_train), 28, 28, 1)) 13x_test = np.reshape(x_test, (len(x_test), 28, 28, 1)) 14 15noise_factor = 0.5 16x_train_noisy = x_train + noise_factor * np.random.normal(loc=0.0, scale=1.0, size=x_train.shape) 17x_test_noisy = x_test + noise_factor * np.random.normal(loc=0.0, scale=1.0, size=x_test.shape) 18 19x_train_noisy = np.clip(x_train_noisy, 0., 1.) 20x_test_noisy = np.clip(x_test_noisy, 0., 1.) 21 22input_img = keras.Input(shape=(28, 28, 1)) 23 24x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(input_img) 25x = layers.MaxPooling2D((2, 2), padding='same')(x) 26x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(x) 27encoded = layers.MaxPooling2D((2, 2), padding='same')(x) 28 29x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(encoded) 30x = layers.UpSampling2D((2, 2))(x) 31x = layers.Conv2D(32, (3, 3), activation='relu', padding='same')(x) 32x = layers.UpSampling2D((2, 2))(x) 33decoded = layers.Conv2D(1, (3, 3), activation='sigmoid', padding='same')(x) 34 35autoencoder = keras.Model(input_img, decoded) 36autoencoder.compile(optimizer='adam', loss='binary_crossentropy') 37 38autoencoder.fit(x_train_noisy, x_train, 39 epochs=25, 40 batch_size=128, 41 shuffle=True, 42 validation_data=(x_test_noisy, x_test), 43 callbacks=[TensorBoard(log_dir='/tmp/tb', histogram_freq=0, write_graph=False)]) 44autoencoder.save("model.h5")
次に、自分で用意したノイズの多い手書き数字「noise_number.png」を作成したモデルに入力し、ノイズを除去したいと思います。そのために以下のようなコードを書いたのですが、エラーが発生せず、ノイズ除去後の画像も表示されません。どうすればこの問題を解決できるでしょうか?具体的なコードを教えていただけると助かります。
python
1from PIL import Image 2from keras.models import load_model 3 4img = Image.open('/content/noise_number.png').convert('L') 5img=img.resize((28,28)) 6img = np.array(img) 7img=img.reshape(28,28,1) 8autoencoder=load_model("model.h5") 9pred = autoencoder.predict(img[np.newaxis])

バッドをするには、ログインかつ
こちらの条件を満たす必要があります。