DataHandeling.py 文件源码

def _get_image(self):
        filename = self.raw_queue.dequeue()
        im_raw = tf.read_file(self.base_folder+filename)
        image_size = self.image_size + (1, )
        image = tf.reshape(tf.cast(tf.image.decode_png(im_raw, channels=1, dtype=tf.uint16), tf.float32),
                           image_size)

        return image, filename