HTensor/IGEV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update evaluate_mvs.py

Browse Source
This commit is contained in:
Gangwei Xu 2023-03-20 20:05:10 +08:00 committed by GitHub
parent bac76cd757
commit 25bd7f4fb0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
IGEV-MVS/evaluate_mvs.py
View File

@ -26,7 +26,7 @@ cudnn.benchmark = True
parser = argparse.ArgumentParser(description='Predict depth, filter, and fuse') parser = argparse.ArgumentParser(description='Predict depth, filter, and fuse')
parser.add_argument('--model', default='IterMVS', help='select model') parser.add_argument('--model', default='IterMVS', help='select model')
parser.add_argument('--dataset', default='dtu_yao_eval', help='select dataset') parser.add_argument('--dataset', default='dtu_yao_eval', help='select dataset')
parser.add_argument('--testpath', default='/data/dtu_data/dtu_test/', help='testing data path') parser.add_argument('--testpath', default='/data/DTU_data/dtu_test/', help='testing data path')
parser.add_argument('--testlist', default='./lists/dtu/test.txt', help='testing scan list') parser.add_argument('--testlist', default='./lists/dtu/test.txt', help='testing scan list')
parser.add_argument('--maxdisp', default=256) parser.add_argument('--maxdisp', default=256)
parser.add_argument('--split', default='intermediate', help='select data') parser.add_argument('--split', default='intermediate', help='select data')
@ -447,4 +447,4 @@ if __name__ == '__main__':
print('scan: '+scan+' time = {:3f}'.format(time.time() - start_time)) print('scan: '+scan+' time = {:3f}'.format(time.time() - start_time))
else: else:
filter_depth(args.testpath, args.outdir, os.path.join(args.outdir, 'custom.ply'), filter_depth(args.testpath, args.outdir, os.path.join(args.outdir, 'custom.ply'),
args.geo_pixel_thres, args.geo_depth_thres, args.photo_thres, img_wh, geo_mask_thres=3) args.geo_pixel_thres, args.geo_depth_thres, args.photo_thres, img_wh, geo_mask_thres=3)