@@ -24,7 +24,7 @@ def get_opts():
2424                        default = '/home/ubuntu/data/mvs_training/dtu/' ,
2525                        help = 'root directory of dtu dataset' )
2626    parser .add_argument ('--split' , type = str ,
27-                         default = 'val ' , choices = ['train' , 'val' , 'test' ],
27+                         default = 'test ' , choices = ['train' , 'val' , 'test' ],
2828                        help = 'which split to evaluate' )
2929    parser .add_argument ('--scans' , nargs = "+" , type = int , default = [],
3030                        help = 'specify scans to evaluate (must be in the split)' )
@@ -41,7 +41,7 @@ def get_opts():
4141                        help = 'number of groups in groupwise correlation, must be a divisor of 8' )
4242    parser .add_argument ('--img_wh' , nargs = "+" , type = int , default = [1152 , 864 ],
4343                        help = 'resolution (img_w, img_h) of the image, must be multiples of 32' )
44-     parser .add_argument ('--ckpt_path' , type = str , default = '' ,
44+     parser .add_argument ('--ckpt_path' , type = str , default = 'ckpts/exp2/_ckpt_epoch_10.ckpt ' ,
4545                        help = 'pretrained checkpoint path to load' )
4646    parser .add_argument ('--save_visual' , default = False , action = 'store_true' ,
4747                        help = 'save depth and proba visualization or not' )
0 commit comments