文章/答案/技术大牛

发布

社区首页 >问答首页 >什么使用tensorflow估计器创建多输入

问什么使用tensorflow估计器创建多输入
EN

Stack Overflow用户

提问于 2018-12-11 02:46:24

回答 1查看 865关注 0票数 0

对不起，我的英语很差。=

我创建了一个keras模型，并使用tf.keras.estimator.model_to_estimator转换为估计器，但是模型是多输入的，我可以创建数据集输入数据吗？

这是我的模型代码：

model = VGG19(include_top=False, input_shape=(182, 182 , 3))
y = model.output
y = keras.layers.Flatten()(y)
y = keras.layers.Dense(512, activation='relu')(y)
y = keras.layers.Dense(256, activation='relu')(y)
y = keras.layers.Dense(128, activation='relu')(y)
model = keras.Model(inputs=model.input, outputs=y)

input_image = keras.layers.Input(shape=(182, 182, 3))
input_anchor = keras.layers.Input(shape=(182, 182, 3))
out_image = model(input_image)
out_anchor = model(input_anchor)

out = keras.layers.concatenate([out_image, out_anchor])
out = keras.layers.Dense(1, activation='sigmoid')(out)
img_model = keras.Model([input_image, input_anchor], out)

face_model.compile(optimizer=tf.train.AdamOptimizer(1e-4, loss='binary_crossentropy', metrics=['accuracy'])

distribution = tf.contrib.distribute.CollectiveAllReduceStrategy(num_gpus_per_worker=0)
config = tf.estimator.RunConfig(model_dir='/home/zjq/test/image_model.h5', train_distribute=distribution)

est_model = tf.keras.estimator.model_to_estimator(keras_model=image_model, config=config)

现在，我有一个输入列表，形状是(100000,182,182，3)，(100000,182,182，3)，(100000，1)，如何定义输入函数返回tf.data.DataSet？

python

tensorflow

keras

回答 1

Stack Overflow用户

回答已采纳

发布于 2018-12-11 06:29:54

首先，命名输入占位符：

input_image = keras.layers.Input(shape=(182, 182, 3),name='image')
input_anchor = keras.layers.Input(shape=(182, 182, 3),name='anchor')

如果输入数据为train_data，形状为[(100000, 182, 182, 3), (100000, 182, 182, 3), (100000, 1)]，则执行以下操作：

BATCH_SIZE = 512
EPOCHS = 4

def input_fn(data, epochs, batch_size):
    # Convert the inputs to a Dataset.
    dataset = tf.data.Dataset.from_tensor_slices(({'image':data[0],'anchor':data[1]}, data[2]))
    # Shuffle, repeat, and batch the examples.
    SHUFFLE_SIZE = 1000
    dataset = dataset.shuffle(SHUFFLE_SIZE).repeat(epochs).batch(batch_size)
    dataset = dataset.prefetch(2)
    # Return the dataset.
    return dataset
est_model.train(lambda :input_fn(train_data,EPOCHS,BATCH_SIZE))

参数BATCH_SIZE、EPOCHS和SHUFFLE_SIZE可根据您的需要进行调整。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/53716629

复制

相似问题

问什么使用tensorflow估计器创建多输入
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问什么使用tensorflow估计器创建多输入EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问什么使用tensorflow估计器创建多输入
EN