自动编码器自定义数据集tensorflow 2.3值错误：使用数据集作为输入时不支持'y'参数

w46czmvw 于 2023-01-17 发布在其他

关注(0)|答案(3)|浏览(126)

我正在尝试在Tensorflow 2.3中实现自动编码器。我将自己存储在磁盘上的图像数据集作为输入。有人能给我解释一下如何以正确的方式实现吗？
我尝试使用tf.keras.preprocessing.image_dataset_from_directory（）加载目录中的数据，但当我使用从上述方法获取的数据开始训练时，我得到以下错误。
“值错误：使用数据集作为输入时不支持y参数。”
PFB我正在运行的代码
'''

import tensorflow as tf
from convautoencoder import ConvAutoencoder
from tensorflow.keras.optimizers import Adam
import matplotlib.pyplot as plt
import numpy as np

EPOCHS = 25
batch_size = 1
img_height = 180
img_width = 180
data_dir = "/media/aniruddha/FE47-91B8/Laptop_Backup/Auto-Encoders/Basic/data"

train_ds = tf.keras.preprocessing.image_dataset_from_directory(
  data_dir,
  validation_split=0.2,
  subset="training",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

val_ds = tf.keras.preprocessing.image_dataset_from_directory(
  data_dir,
  validation_split=0.2,
  subset="validation",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

(encoder, decoder, autoencoder) = ConvAutoencoder.build(224, 224, 3)
opt = Adam(lr=1e-3)
autoencoder.compile(loss="mse", optimizer=opt)

H = autoencoder.fit(    train_ds, train_ds, validation_data=(val_ds, val_ds),   epochs=EPOCHS, batch_size=batch_size)

'''

tensorflow

来源：https://stackoverflow.com/questions/64214847/autoencoder-custom-dataset-tensorflow-2-3-valueerror-y-argument-is-not-suppor

3条答案

按热度按时间

wko9yo5t1#

我解决了这个问题。我没有将输入数据集作为元组馈送给模型进行训练。一旦我纠正了这个问题，训练就开始了。

赞(0）回复(0）举报 2023-01-17

sd2nnvve2#

我使用生成器将输入数据作为元组馈送到自动编码器。请在下面找到我的代码。

# initialize the training training data augmentation object
trainAug = ImageDataGenerator(rescale=1. / 255)

valAug = ImageDataGenerator(rescale=1. / 255)

# initialize the training generator
trainGen = trainAug.flow_from_directory(
    config.TRAIN_PATH,
    class_mode="input",
    classes=None,
    target_size=(64, 64),
    color_mode="grayscale",
    shuffle=True,
    batch_size=BS)
# initialize the validation generator
valGen = valAug.flow_from_directory(
    config.TRAIN_PATH,
    class_mode="input",
    classes=None,
    target_size=(64, 64),
    color_mode="grayscale",
    shuffle=False,
    batch_size=BS)
# initialize the testing generator
testGen = valAug.flow_from_directory(
    config.TRAIN_PATH,
    class_mode="input",
    classes=None,
    target_size=(64, 64),
    color_mode="grayscale",
    shuffle=False,
    batch_size=BS)

early_stop = EarlyStopping(monitor='val_loss', patience=20)
mc = ModelCheckpoint('best_model_1.h5', monitor='val_loss', mode='min', save_best_only=True)

# construct our convolutional autoencoder
print("[INFO] building autoencoder...")
(encoder, decoder, autoencoder) = ConvAutoencoder.build(64, 64, 1)
opt = Adam(learning_rate= 0.0001, beta_1=0.9, beta_2=0.999, epsilon=1e-04, amsgrad=False)
autoencoder.compile(loss="mse", optimizer=opt)

# train the convolutional autoencoder
H = autoencoder.fit(    trainGen,   validation_data=valGen, epochs=EPOCHS, batch_size=BS ,callbacks=[ mc , early_stop])

赞(0）回复(0）举报 2023-01-17

cfh9epnr3#

fit需要data * 和 * 标签，但它只接受一个tf.data.Dataset。要将data用作自动编码器的标签，应向数据集构造函数提供两次，例如：

dataset = tf.data.Dataset.from_tensor_slices((images, images))

赞(0）回复(0）举报 2023-01-17

我来回答

自动编码器自定义数据集tensorflow 2.3值错误：使用数据集作为输入时不支持'y'参数

3条答案

相关问题

热门标签

最新问答