/SupervisedDepthPrediction

Pytorch framework for supervised depth prediction

Primary LanguagePython

SupervisedDepthPrediction

This is a distributed training framework for supervised depth prediction based on Pytorch 1.0 (Pytorch version >= 1.2 is best). Now it provide the implementation of DORN(state of the art in KITTI depth prediction benchmark), and you can implement your model in your customed dataset with a little modification.

Highlights

  • Distributed & Single GPU Flexible selection between distributed training with multi gpus and a single gpu.
  • Flexible Visulization Implementation You can implement your visulizers for network comprehensive analysis.
  • Suport Various Optimizers and Learning-rate Policy Provide all the optimizers and learning-rate schedulers in pytorch. And support poly lr_scheduler and warmup, which are widely used in segmentation and detection.
  • Support Grad Clip Provide grad clip to avoid gradient exploding.
  • Mixed Precision Training Support mixed precision training with NVIDIA apex lib.
  • Sync BN Support Sync BN when training with multi gps.
  • Break-point Restoration Support continue to train from a break-point.

Installation

Dataset

Training

Testing

Results