Обнуление с помощью R keras layer_zero_padding_2d - PullRequest
Новая
       26

Обнуление с помощью R keras layer_zero_padding_2d

0 голосов
/ 13 октября 2018

Я пытаюсь добавить 2 строки / столбца к каждому изображению в наборе данных MNIST, используя layer_zero_padding_2d.Я хочу, чтобы изображения имели размер 32х32 вместо 28х28.Я знаю, что могу сделать это с другими методами, но я пытаюсь понять, почему моя модель выдает мне ошибку: «индекс списка вне диапазона»

Вот мой код: 

library(keras)

mnist <- dataset_mnist()

c(c(train_images, train_labels), c(test_images, test_labels)) %<-% mnist

train_images <- array_reshape(train_images, c(60000, 28, 28, 1))
train_images <- train_images / 255

test_images <- array_reshape(test_images, c(10000, 28, 28, 1))
test_images <- test_images / 255

train_labels <- to_categorical(train_labels)
test_labels <- to_categorical(test_labels)


mnist.LeNet5 <- keras_model_sequential() %>% 
  layer_zero_padding_2d(padding = 2)

mnist.LeNet5 %>% compile(
  optimizer = optimizer_rmsprop(),
  loss = "categorical_crossentropy",
  metrics = c("accuracy")
)


history <- mnist.LeNet5 %>% fit(x = train_images)
...