4eb5897b172b7157632e7bee83624646142740ae,data_provider/shadownet_data_feed_pipline.py,,,#,314

Before Change

        )


if __name__ == "__main__":
    
    test code
    
    // test crnn data producer
    producer = CrnnDataProducer(
        dataset_dir="/media/baidu/Data/Sequence_Recognition/Synth_90K/90kDICT32px",
    )

    producer.generate_tfrecords(
        save_dir="/media/baidu/Data/Sequence_Recognition/Synth_90K/tfrecords",
        step_size=100000
    )

After Change

        flags="train"
    )

    train_images, train_labels, train_images_paths = train_dataset.inputs(
        batch_size=32
    )

    count = 0
    image_path_set = set()

    with tf.Session() as sess:

        while True:
            try:
                images, image_labels, image_paths = sess.run([train_images, train_labels, train_images_paths])

                for path in image_paths:
                    image_path_set.add(path)

                count += len(image_paths)
            except tf.errors.OutOfRangeError:
                print(count)
                print(len(image_path_set))

Instances