From a8062b8cdf99922e6070abcc5f8369d83ed68596 Mon Sep 17 00:00:00 2001 From: KMarshallX Date: Sun, 14 Jul 2024 14:23:40 +1000 Subject: [PATCH] help menu update --- .gitignore | 4 ++++ config/adapt_config.py | 4 ++-- config/boost_config.py | 6 ++++-- config/pred_config.py | 4 ++-- 4 files changed, 12 insertions(+), 6 deletions(-) diff --git a/.gitignore b/.gitignore index 8f17ad7..1d4a153 100644 --- a/.gitignore +++ b/.gitignore @@ -4,6 +4,10 @@ /*.sh train_zero.py train_test.py +test_0315.py +test_plt.click.py +test_postpo_0319.py +zcrosval_test.py test_time_adaptation_test.py /__pycache__ /data diff --git a/config/adapt_config.py b/config/adapt_config.py index 857ad7b..6a13c42 100644 --- a/config/adapt_config.py +++ b/config/adapt_config.py @@ -30,9 +30,9 @@ # postprocessing / threshold # hard thresholding value -adapt_parser.add_argument('--thresh', type=float, default=0.1, help=argparse.SUPPRESS) +adapt_parser.add_argument('--thresh', type=float, default=0.1, help="binary threshold for the probability map after prediction, default=0.1") # connected components analysis threshold value (denoising) -adapt_parser.add_argument('--cc', type=int, default=10, help=argparse.SUPPRESS) +adapt_parser.add_argument('--cc', type=int, default=10, help="connected components analysis threshold value (denoising), default=10") adapt_parser.add_argument('--pretrained', type=str, default = "/pretrained_model_path/", help="path of the prertrained model") diff --git a/config/boost_config.py b/config/boost_config.py index ce9f6c4..fbd947d 100644 --- a/config/boost_config.py +++ b/config/boost_config.py @@ -8,6 +8,8 @@ import argparse +import boost + boost_parser = argparse.ArgumentParser(description="VesselBoost training arguments") # Train.py @@ -59,8 +61,8 @@ # postprocessing / threshold # hard thresholding value -boost_parser.add_argument('--thresh', type=float, default=0.1, help=argparse.SUPPRESS) +boost_parser.add_argument('--thresh', type=float, default=0.1, help="binary threshold for the probability map after prediction, default=0.1") # connected components analysis threshold value (denoising) -boost_parser.add_argument('--cc', type=int, default=10, help=argparse.SUPPRESS) +boost_parser.add_argument('--cc', type=int, default=10, help="connected components analysis threshold value (denoising), default=10") args = boost_parser.parse_args() \ No newline at end of file diff --git a/config/pred_config.py b/config/pred_config.py index 38b02b4..0996c87 100644 --- a/config/pred_config.py +++ b/config/pred_config.py @@ -29,9 +29,9 @@ # postprocessing / threshold # hard thresholding value -pred_parser.add_argument('--thresh', type=float, default=0.1, help=argparse.SUPPRESS) +pred_parser.add_argument('--thresh', type=float, default=0.1, help="binary threshold for the probability map after prediction, default=0.1") # connected components analysis threshold value (denoising) -pred_parser.add_argument('--cc', type=int, default=10, help=argparse.SUPPRESS) +pred_parser.add_argument('--cc', type=int, default=10, help="connected components analysis threshold value (denoising), default=10") pred_parser.add_argument('--pretrained', type=str, default = "/pretrained_model_path/", help="path of the prertrained model")