|
|
@@ -0,0 +1,36 @@ |
|
|
|
import argparse |
|
|
|
import os |
|
|
|
|
|
|
|
|
|
|
|
parse = argparse.ArgumentParser(description='PyTorch Polyp Segmentation') |
|
|
|
|
|
|
|
"-------------------data option--------------------------" |
|
|
|
parse.add_argument('--root', type=str, default='/scratch/krushi1992/MICCAI2021/new-med-up/CVC-frame/') |
|
|
|
parse.add_argument('--dataset', type=str, default='EndoScene') |
|
|
|
parse.add_argument('--train_data_dir', type=str, default='train') |
|
|
|
parse.add_argument('--valid_data_dir', type=str, default='valid') |
|
|
|
parse.add_argument('--test_data_dir', type=str, default='test') |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"-------------------training option-----------------------" |
|
|
|
parse.add_argument('--mode', type=str, default='train') |
|
|
|
parse.add_argument('--nEpoch', type=int, default=200) |
|
|
|
parse.add_argument('--batch_size', type=float, default=4) |
|
|
|
parse.add_argument('--num_workers', type=int, default=0) |
|
|
|
parse.add_argument('--use_gpu', type=bool, default=True) |
|
|
|
parse.add_argument('--load_ckpt', type=str, default=None) |
|
|
|
parse.add_argument('--model', type=str, default='EUNet') |
|
|
|
parse.add_argument('--expID', type=int, default=1) |
|
|
|
parse.add_argument('--ckpt_period', type=int, default=50) |
|
|
|
parse.add_argument('--weight_const', type=float, default=0.3) |
|
|
|
|
|
|
|
"-------------------optimizer option-----------------------" |
|
|
|
parse.add_argument('--lr', type=float, default=1e-3) |
|
|
|
parse.add_argument('--weight_decay', type=float, default=1e-5) |
|
|
|
parse.add_argument('--mt', type=float, default=0.9) |
|
|
|
parse.add_argument('--power', type=float, default=0.9) |
|
|
|
|
|
|
|
parse.add_argument('--nclasses', type=int, default=1) |
|
|
|
|
|
|
|
opt = parse.parse_args() |