Skip to content

Commit

Permalink
help menu update
Browse files Browse the repository at this point in the history
  • Loading branch information
KMarshallX committed Jul 14, 2024
1 parent 9ebcddf commit a8062b8
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 6 deletions.
4 changes: 4 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 2 additions & 2 deletions config/adapt_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand Down
6 changes: 4 additions & 2 deletions config/boost_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@

import argparse

import boost

boost_parser = argparse.ArgumentParser(description="VesselBoost training arguments")

# Train.py
Expand Down Expand Up @@ -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()
4 changes: 2 additions & 2 deletions config/pred_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand Down

0 comments on commit a8062b8

Please sign in to comment.