【发布时间】:2016-08-19 15:55:45
【问题描述】:
使用 Tensorflow,我得到一个 2048 维向量作为 pool3 层的输出。但是,使用 Keras 的 include_top=False 会给出 8,8,2048 维向量。如何获得与使用 TensorFlow 的 pool3 输出层相同的向量?
【问题讨论】:
标签: tensorflow keras
使用 Tensorflow,我得到一个 2048 维向量作为 pool3 层的输出。但是,使用 Keras 的 include_top=False 会给出 8,8,2048 维向量。如何获得与使用 TensorFlow 的 pool3 输出层相同的向量?
【问题讨论】:
标签: tensorflow keras
【讨论】:
model = InceptionV3(include_top=False, pooling='avg', input_shape=(299,299,3))
encode_model = InceptionV3(weights='imagenet')
encode_model = Model(encode_model.input, encode_model.layers[-2].output)
【讨论】: