在Tensorflow的输入流水线中捕获错误

时间:2018-08-05 21:59:53

标签: tensorflow

我编写了一个自定义数据增强功能,可在训练期间应用于我的图像-在一些情况下进行了测试,效果很好。现在,它是使用tf.data.Dataset API的输入管道的一部分。确认只需要迭代几批就可以了,所以我决定开始训练

我所有的数据都是TFRecord格式的,当我开始训练时,我在途中遇到了一个错误,所以那里有些图像不喜欢aug。

我尝试动态查找以捕获这些错误并在培训期间获取某种信息,这些信息可以帮助我了解失败的图像,因此我可以更深入地调查该情况,但实际上我什么也找不到。

只需在这里寻求更一般的建议,我在这里有什么选择?除了将TFRecords中的所有图像都转储到文件夹中并以这种方式进行遍历之外?

谢谢

0 个答案:

没有答案