Exemplo n.º 1
0
    def validate(self):

        self.is_train = False
        init_op = tf.global_variables_initializer()
        self.saver = tf.train.Saver()
        self.sess.run(init_op)
        if self.load(self.args.checkpoint_dir):
            print(" [*] Load SUCCESS")
        else:
            print(" [!] Load failed...")

        genSample = AbdomenSampler(self.args, 'validate')
        # self.generator_sample_targetwise(genSample, )
        outputdir = self.args.sample_dir + "/atlas_target/"
        ds_all = []
        bf_ds_all = []
        jt_all = []
        for img_fix, lab_fix in zip(genSample.img_fix, genSample.lab_fix):
            output_dir = outputdir + get_name_wo_suffix(img_fix)
            mk_or_cleardir(output_dir)
            print(output_dir)
            for img_mv, lab_mv in zip(genSample.img_mv, genSample.lab_mv):
                bfds, ds = self.gen_one_batch(genSample, img_fix, lab_fix,
                                              img_mv, lab_mv, output_dir)
                # print("sim= %f"%ds)
                ds_all.append(ds)
                bf_ds_all.append(bfds)
            # print(ds_all)
        self.logger.debug("%s total_sim %s->%s " % (Get_Name_By_Index(
            self.args.component), self.args.Tatlas, self.args.Ttarget))
        print_mean_and_std(bf_ds_all)
        print_mean_and_std(ds_all)
Exemplo n.º 2
0
    def validate(self):
        self.build_cost()
        self.meger_all()
        init_op = tf.global_variables_initializer()
        self.saver = tf.train.Saver()
        self.sess.run(init_op)
        if self.load(self.args.checkpoint_dir):
            print(" [*] Load SUCCESS")
        else:
            print(" [!] Load failed...")

        dices = []
        for i in range(self.valid_sampler.len):
            dices.append(self.sample_network(i))
        print_mean_and_std(dices)
Exemplo n.º 3
0
 def validate(self):
     self.is_train = False
     init_op = tf.global_variables_initializer()
     self.saver = tf.train.Saver()
     self.sess.run(init_op)
     if self.load(self.args.checkpoint_dir):
         print(" [*] Load SUCCESS")
     else:
         print(" [!] Load failed...")
     sims = []
     for i in range(self.validate_sampler.len):
         sim = self.__sample(i)
         sims.append(sim)
     print("%s total_sim %s->%s " % (Get_Name_By_Index(
         self.args.component), self.args.Tatlas, self.args.Ttarget))
     print_mean_and_std(sims)
Exemplo n.º 4
0
 def fusion(self):
     self.is_train = False
     init_op = tf.global_variables_initializer()
     self.saver = tf.train.Saver()
     self.sess.run(init_op)
     if self.load(self.args.checkpoint_dir):
         print(" [*] Load SUCCESS")
     else:
         print(" [!] Load failed...")
     DSs = []
     HDs = []
     start = time.time()
     for i in range(self.validate_sampler.len):
         ds, hd = self.fusion_one_target(i)
         DSs.append(ds)
         HDs.append(hd)
     end = time.time()
     print("runtime: %.2f" % ((end - start) / self.validate_sampler.len))
     print("%s total_sim %s->%s " % (Get_Name_By_Index(
         self.args.component), self.args.Tatlas, self.args.Ttarget))
     outpu2excel(self.args.res_excel, self.args.MOLD_ID + "_DS", DSs)
     outpu2excel(self.args.res_excel, self.args.MOLD_ID + "_HD", HDs)
     print_mean_and_std(DSs)
     print_mean_and_std(HDs)