Spaces:
Running
on
A10G
Running
on
A10G
File size: 1,232 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 |
# ------------------------------------------------------------------------------
# The code is from GLPDepth (https://github.com/vinvino02/GLPDepth).
# For non-commercial purpose only (research, evaluation etc).
# ------------------------------------------------------------------------------
from configs.base_options import BaseOptions
class TestOptions(BaseOptions):
def initialize(self):
parser = BaseOptions.initialize(self)
# experiment configs
parser.add_argument('--ckpt_dir', type=str,
default='./ckpt/best_model_nyu.ckpt',
help='load ckpt path')
parser.add_argument('--result_dir', type=str, default='./results',
help='save result images into result_dir/exp_name')
parser.add_argument('--crop_h', type=int, default=448)
parser.add_argument('--crop_w', type=int, default=576)
parser.add_argument('--save_eval_pngs', action='store_true',
help='save result image into evaluation form')
parser.add_argument('--save_visualize', action='store_true',
help='save result image into visulized form')
return parser
|