示例#1
0
文件： train.py 项目： sqbqamar/3D_UNet 
def offline_evaluate(pred_func, output_file):
        df = get_eval_dataflow()
        if config.DYNAMIC_SHAPE_PRED:    
            eval_brats(
                df, lambda img: segment_one_image_dynamic(img, pred_func))
        else:
            eval_brats(
                df, lambda img: segment_one_image(img, pred_func))
示例#2
0
 def _setup_graph(self):
     self.pred = self.trainer.get_predictor(['image'],
                                            get_model_output_names())
     self.df = get_eval_dataflow()