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

Update demo_video.py

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

3
IGEV-Stereo/demo_video.py
View File

@ -33,6 +33,7 @@ parser.add_argument('--corr_radius', type=int, default=4, help="width of the cor
parser.add_argument('--n_downsample', type=int, default=2, help="resolution of the disparity field (1/2^K)") parser.add_argument('--n_downsample', type=int, default=2, help="resolution of the disparity field (1/2^K)")
parser.add_argument('--slow_fast_gru', action='store_true', help="iterate the low-res GRUs more frequently") parser.add_argument('--slow_fast_gru', action='store_true', help="iterate the low-res GRUs more frequently")
parser.add_argument('--n_gru_layers', type=int, default=3, help="number of hidden GRU levels") parser.add_argument('--n_gru_layers', type=int, default=3, help="number of hidden GRU levels")
parser.add_argument('--max_disp', type=int, default=192, help="max disp of geometry encoding volume")
args = parser.parse_args() args = parser.parse_args()
model = torch.nn.DataParallel(IGEVStereo(args), device_ids=[0]) model = torch.nn.DataParallel(IGEVStereo(args), device_ids=[0])
@ -91,4 +92,4 @@ if __name__ == '__main__':
cv2.imshow('img', out_img) cv2.imshow('img', out_img)
cv2.waitKey(1) cv2.waitKey(1)
videoWrite.write(out_img) videoWrite.write(out_img)
videoWrite.release() videoWrite.release()