6096c0e6ec4df6a902747059a47e0f6f1bec7d77,aocr/util/data_gen.py,DataGen,__init__,#DataGen#,18

Before Change

                            for i in range(self.bucket_max_width + 1)}

        filename_queue = tf.train.string_input_producer([self.annotation_path], num_epochs=self.epochs)
        self.images, self.labels = parse_tfrecords(filename_queue)

    def clear(self):
        self.bucket_data = {i: BucketData()

After Change

        dataset = tf.contrib.data.TFRecordDataset([self.annotation_path])
        dataset = dataset.map(self._parse_record)
        dataset = dataset.shuffle(buffer_size=10000)
        self.dataset = dataset.repeat(self.epochs)

    def clear(self):
        self.bucket_data = {i: BucketData()
                            for i in range(self.bucket_max_width + 1)}

Instances