TRI-ML / packnet-sfm

TRI-ML Monocular Depth Estimation Repository
https://tri-ml.github.io/packnet-sfm/
MIT License
1.24k stars 243 forks source link

Allow batch size > 1 when training NRS #123

Closed jdriscoll319 closed 2 years ago

jdriscoll319 commented 3 years ago

Create Tcw transforms corresponding to batch size. During reconstruct() apply each ray surface in the batch to each corresponding depthmap in the batch. During project() fix tensor manipulation and multiplication to handle batch size greater than one. Essentially, performing all of same operations while looping over each element in the batch.