zzangjinsun / NLSPN_ECCV20

Park et al., Non-Local Spatial Propagation Network for Depth Completion, ECCV, 2020
MIT License
321 stars 55 forks source link

The largest resolution of depthmap? #15

Closed CC9310 closed 3 years ago

CC9310 commented 4 years ago

When i testing my private data of depth (768, 1152), while, the cuda run out of memory, can you specify the largest resolution your method can handle?

zzangjinsun commented 4 years ago

https://github.com/zzangjinsun/NLSPN_ECCV20/blob/46dfa4c42fc380a1261b4f7175753b897ec9bfb0/src/data/kittidc.py#L239-L247

You can try _--testcrop and _--topcrop to crop the top area during the testing. If it's not enough, modification on the code is needed.