TripletSemiHardLoss 的最后小批量

问题描述

我使用的是 Tensorflow 2.3

我正在为 tfrecord 使用以下代码

dataset = tf.data.TFRecordDataset(filenames=filenames)
dataset = dataset.shuffle(buffer_size=100000)
# dataset handling API
dataset = dataset.cache()
dataset = dataset.map(parse_examples,num_parallel_calls=tf.data.experimental.AUTOTUNE)
dataset = dataset.prefetch(buffer_size=tf.data.experimental.AUTOTUNE)
dataset = dataset.batch(50)

假设我有 101 张图片，那么我将有 3 个批次，其中最后一个批次仅包含 1 个元素。

如果我使用 TripletSemiHardLoss，这会出错，因为它需要正例和反例。有没有方便的方法来确保所有批次都有 50 个元素或确保每个批次都有正例和反例？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）