Dataset_reader_ImageSeqGen.py 文件源码

def next_batch(self, batch_size=1, sess=None):
        with tf.name_scope('Batch_getter') as scope:
            if sess is None :
                self.sess = tf.get_default_session()
            else:
                self.sess = sess

            images, seqs, masks = self.sess.run([self.images , self.complete_seq,  self.complete_mask], feed_dict={self.batch_size : batch_size})
            '''
            print(seqs)
            print(str(seqs[0])[0])
            process_seqs=[]
            process_masks =[]
            for index, c in enumerate(seqs):
                process_seqs.extend([c.decode()])
                process_masks.extend([masks[index].decode()])
            '''
            print(seqs)
            c_seqs = self.finalize_seq(seqs)
            c_masks = self.finalize_seq(masks)
            print(c_seqs)
            input_seqs = self.make_input_seq(c_seqs)
            output_seqs = self.make_output_seq(c_seqs)
            final_masks = self.make_input_seq(c_masks)
            return images , input_seqs, output_seqs, final_masks