File size: 3,348 Bytes
7f3dfd7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 |
import argparse
def str2bool(v):
return v.lower() in ('true', 't')
def parse_args():
parser = argparse.ArgumentParser()
# Exp Controller
parser.add_argument(
"--rcd_dir",
type=str,
help="save the evaluation results (in a directory)",
)
parser.add_argument(
"--rcd_file",
type=str,
help="save the evaluation results (in a csv/xlsx file)",
)
parser.add_argument(
"--visualization",
type=str2bool,
default=False,
help="save the visualization for each case (img, gt, pred)",
)
parser.add_argument(
"--checkpoint",
type=str,
help="Checkpoint path",
)
parser.add_argument(
"--partial_load",
type=str2bool,
default=True,
help="Allow to load partial paramters from checkpoint",
)
parser.add_argument(
"--gpu",
type=str,
default=None,
)
parser.add_argument(
"--resume",
type=str2bool,
default=True,
help="Inherit medial results from an interrupted evaluation (no harm even if you evaluate from scratch)",
)
parser.add_argument(
"--save_interval",
type=int,
default=100
)
# Metrics
parser.add_argument(
"--dice",
type=str2bool,
default=True,
)
parser.add_argument(
"--nsd",
type=str2bool,
default=True,
)
# Med SAM Dataset
parser.add_argument(
"--datasets_jsonl",
type=str,
)
parser.add_argument(
"--text_prompts_json",
type=str,
help='This is needed for CVPR25 challenge, where multiple prompts (synonyms) are required.'
)
# Sampler and Loader
parser.add_argument(
"--online_crop",
type=str2bool,
default='False',
help='load pre-cropped image patches directly, or crop online',
)
parser.add_argument(
"--crop_size",
type=int,
nargs='+',
default=[288, 288, 96],
)
parser.add_argument(
"--max_queries",
type=int,
default=256,
)
parser.add_argument(
"--batchsize_3d",
type=int,
default=2,
)
parser.add_argument(
"--pin_memory",
type=str2bool,
default=False,
help='load data to gpu to accelerate'
)
parser.add_argument(
"--num_workers",
type=int,
default=4
)
# Knowledge Encoder
parser.add_argument(
"--text_encoder_partial_load",
type=str2bool,
default=True,
help="Allow to load partial paramters from checkpoint",
)
parser.add_argument(
"--text_encoder_checkpoint",
type=str,
)
parser.add_argument(
"--text_encoder",
type=str,
)
# MaskFormer
parser.add_argument(
"--vision_backbone",
type=str,
help='UNET or UNET-H'
)
parser.add_argument(
"--patch_size",
type=int,
nargs='+',
default=[32, 32, 32],
help='patch size on h w and d'
)
parser.add_argument(
"--deep_supervision",
type=str2bool,
default=False,
)
args = parser.parse_args()
return args
|