예제 #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 _eval(self):
     scores = eval_brats(
         self.df,
         lambda img: segment_one_image(img, [self.pred], is_online=True))
     for k, v in scores.items():
         self.trainer.monitors.put_scalar(k, v)