|
1234567891011121314151617181920212223242526272829303132333435363738 |
- import argparse
- import os
-
-
- parse = argparse.ArgumentParser(description='PyTorch Polyp Segmentation')
-
- "-------------------data option--------------------------"
- # parse.add_argument('--root', type=str, default='/media/external_10TB/10TB/pourmand/CVC-EndoSceneStill')
- # parse.add_argument('--dataset', type=str, default='EndoScene')
-
- parse.add_argument('--root',default='/media/external_10TB/10TB/pourmand/Kvasir-SEG/Kvasir-SEG')
- parse.add_argument('--dataset',type=str,default='kvasir_SEG')
-
- parse.add_argument('--train_data_dir', type=str, default='')
- parse.add_argument('--valid_data_dir', type=str, default='')
- parse.add_argument('--test_data_dir', type=str, default='')
-
- "-------------------training option-----------------------"
- parse.add_argument('--mode', type=str, default='train')
- parse.add_argument('--nEpoch', type=int, default=10)
- 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=0)
- 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()
|