我正在尝试检查小批量输出是否等于将小批量的所有元素一一给出,以评估 Mobilenet 的特征向量。
看下面的代码:
model = tf.keras.models.Sequential(
(
hub.KerasLayer("https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4",
output_shape=[1280],
trainable=False
),
)
)
images = tf.random.uniform(shape=(20, 224, 224, 3))
features = model.predict(images)
for i in range(20):
image = tf.reshape(images[i, ...], (1, 224, 224, 3))
image_feature = model.predict(image)
self.assertTrue((image_feature == features[i, ...]).all())
assertTrue 在我的测试中失败。它是否应该为所有图像提供相同的特征向量,无论它们是作为小批量还是一张一张地提供?