TypeError：“ TFRecordDataset”操作的输入“文件名”的类型为float32，与预期的字符串类型不匹配

Question

我正在尝试使用此github代码cnn-watermark-removal来去除图像中的水印。 我已按照说明在data/VOCdevkit/VOC2012/JPEGImages中添加了自己的自定义图片，并使用给定的命令对其进行了培训。

在最后一行return (next_element, [iterator.make_initializer(x) for x in [train, val]])中  这部分代码给出了以上错误。

def dataset_split(dataset_fn, split):
    # import pdb; pdb.set_trace();
    records = get_records()
    split = int(len(records) * split)
    train, val = dataset_fn(records[:split]), dataset_fn(records[split:])
    iterator = tf.contrib.data.Iterator.from_structure(
        train.output_types, train.output_shapes)
    # import ipdb; ipdb.set_trace();
    next_element = iterator.get_next()
    return (next_element,
            [iterator.make_initializer(x) for x in [train, val]])

我在stackoverflow TypeError: Input 'filenames' of 'TFRecordDataset' 上发现了类似的问题，在我的情况下，预期错误类型为int32 unilike，即字符串。

我对tensorflow非常陌生。任何潜在客户高度赞赏。谢谢。