appliedinnovation / fast-depth

ICRA 2019 "FastDepth: Fast Monocular Depth Estimation on Embedded Systems"
MIT License
0 stars 0 forks source link

Refactor FastDepthDataset class #21

Open alexbarnett12 opened 3 years ago

alexbarnett12 commented 3 years ago

Currently, the FastDepthDataset that is used to load data during training is a derived class from DepthDataset in the bayesian-visual-odometry repo. This was done early on because at the time there was no quick way to create my own data loader without modifying Guo's repo or writing repetitive code. It's now gotten to the point where it is holding back progress though, and will make it especially hard to integrate real-world data into the training pipeline. The refactor will perform the following:

With this structure, any separate repo that is training a model off of simulation/real-world data only has to create its own derived class from RoshiDataset and perform preprocessing. This will be particularly useful when we expand to many other neural nets, such as segmentation, loop closure, optical flow, etc.