Spaces:
Running
on
A10G
Running
on
A10G
File size: 2,611 Bytes
bcec54e |
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 |
# ------------------------------------------------------------------------------
# The code is from GLPDepth (https://github.com/vinvino02/GLPDepth).
# For non-commercial purpose only (research, evaluation etc).
# Modified by Zigang Geng (zigang@mail.ustc.edu.cn).
# ------------------------------------------------------------------------------
import argparse
def str2bool(v):
if isinstance(v, bool):
return v
if v.lower() in ('yes', 'true', 't', 'y', '1'):
return True
elif v.lower() in ('no', 'false', 'f', 'n', '0'):
return False
else:
raise argparse.ArgumentTypeError('Boolean value expected.')
class BaseOptions():
def __init__(self):
pass
def initialize(self):
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
# base configs
parser.add_argument('--resume_from', type=str, default='')
parser.add_argument('--exp_name', type=str, default='')
parser.add_argument('--gpu_or_cpu', type=str, default='gpu')
parser.add_argument('--data_path', type=str, default='/data/ssd1/')
parser.add_argument('--dataset', type=str, default='nyudepthv2',
choices=['nyudepthv2', 'kitti', 'imagepath'])
parser.add_argument('--batch_size', type=int, default=8)
parser.add_argument('--workers', type=int, default=8)
# depth configs
parser.add_argument('--max_depth', type=float, default=10.0)
parser.add_argument('--max_depth_eval', type=float, default=10.0)
parser.add_argument('--min_depth_eval', type=float, default=1e-3)
parser.add_argument('--do_kb_crop', type=int, default=1)
parser.add_argument('--kitti_crop', type=str, default=None,
choices=['garg_crop', 'eigen_crop'])
parser.add_argument('--pretrained', type=str, default='')
parser.add_argument('--drop_path_rate', type=float, default=0.3)
parser.add_argument('--use_checkpoint', type=str2bool, default='False')
parser.add_argument('--num_deconv', type=int, default=3)
parser.add_argument('--num_filters', nargs='+', type=int, default=[32,32,32])
parser.add_argument('--deconv_kernels', nargs='+', type=int, default=[2,2,2])
parser.add_argument('--shift_window_test', action='store_true')
parser.add_argument('--shift_size', type=int, default=2)
parser.add_argument('--flip_test', action='store_true')
return parser
|