88b164b98c209ea8e2afbfae93a01b9aacd8b0ab,local_utils/data_utils.py,TextFeatureReader,read_features,#Any#Any#Any#Any#,206

Before Change

        features = tf.parse_single_example(serialized_example,
                                           features={
                                               "images": tf.FixedLenFeature((), tf.string),
                                               "imagenames": tf.FixedLenFeature([1], tf.string),
                                               "labels": tf.VarLenFeature(tf.int64),
                                           })
        image = tf.decode_raw(features["images"], tf.uint8)
        w, h = input_size
        images = tf.reshape(image, [h, w, input_channels])
        labels = features["labels"]
        labels = tf.cast(labels, tf.int32)
        imagenames = features["imagenames"]
        return images, labels, imagenames


class TextFeatureIO(object):
    

After Change

        filename_queue = tf.train.string_input_producer([tfrecords_path], num_epochs=num_epochs)
        reader = tf.TFRecordReader()
        _, serialized_example = reader.read(filename_queue)
        return TextFeatureReader.extract_features(serialized_example, input_size, input_channels)

    @staticmethod
    def extract_features(serialized_sample, input_size: Tuple[int, int], input_channels: int) \
            -> Tuple[tf.Tensor, tf.Tensor, tf.Tensor]:

Instances