diff --git a/classification/AntBee/config/train_test_ce2.cfg b/classification/AntBee/config/train_test_ce2.cfg index 4444916..6d47c36 100755 --- a/classification/AntBee/config/train_test_ce2.cfg +++ b/classification/AntBee/config/train_test_ce2.cfg @@ -78,5 +78,5 @@ gpus = [0] # checkpoint mode can be [0-latest, 1-best, 2-specified] ckpt_mode = 1 output_dir = result -output_csv = restnet18_ce2.csv +output_csv = resnet18_ce2.csv save_probability = True diff --git a/segmentation/JSRT/config/evaluation.cfg b/segmentation/JSRT/config/evaluation.cfg index dae8519..d4f0030 100755 --- a/segmentation/JSRT/config/evaluation.cfg +++ b/segmentation/JSRT/config/evaluation.cfg @@ -4,7 +4,7 @@ label_list = [255] organ_name = lung ground_truth_folder_root = ../../PyMIC_data/JSRT -segmentation_folder_root = result +segmentation_folder_root = result/unet evaluation_image_pair = ./config/jsrt_test_gt_seg.csv diff --git a/segmentation/JSRT2/net_run_jsrt.py b/segmentation/JSRT2/net_run_jsrt.py index 93e690b..6ebecad 100644 --- a/segmentation/JSRT2/net_run_jsrt.py +++ b/segmentation/JSRT2/net_run_jsrt.py @@ -23,7 +23,7 @@ def main(): config = synchronize_config(config) log_dir = config['training']['ckpt_save_dir'] if(not os.path.exists(log_dir)): - os.mkdir(log_dir) + os.makedirs(log_dir) if sys.version.startswith("3.9"): logging.basicConfig(filename=log_dir+"/log_{0:}.txt".format(stage), level=logging.INFO, format='%(message)s', force=True) # for python 3.9