DataHandeling.py 文件源码

def _get_image(self):
        image_size = self.image_size + (1, )
        filename_fw = self.raw_queue_fw.dequeue()
        im_raw = tf.read_file(self.base_folder+filename_fw)
        image_fw = tf.reshape(tf.cast(tf.image.decode_png(im_raw, channels=1, dtype=tf.uint16), tf.float32),
                              image_size)
        filename_bw = self.raw_queue_bw.dequeue()
        im_raw = tf.read_file(self.base_folder + filename_bw)

        image_bw = tf.reshape(tf.cast(tf.image.decode_png(im_raw, channels=1, dtype=tf.uint16), tf.float32),
                              image_size)

        return image_fw, filename_fw, image_bw, filename_bw