Skip to content

Latest commit

 

History

History
18 lines (17 loc) · 900 Bytes

README.md

File metadata and controls

18 lines (17 loc) · 900 Bytes

Complex-YOLO

NOTE: IN PROGESS!

PyTorch Implementation of Complex-YOLO 3D
Much of the code has been adapted from this repo. However, I've changed a considerable amount of code, improved readibility and speed (from initial runs).
Update:

  • 3/3 - Fixed bug in loss. Commented region_loss.py and utils.py in relation to notation in paper.
  • 3/2 - Not able understand what loss function was doing. Bug in loss.

TODO:

  • Go through Kitti Dataset class definition
  • Construct model from original implementation
  • Download data
  • Modify loss function
  • Comment loss function. Understand everything that's going on
  • Test function
  • Visualization of results
  • Running loss for saving model
  • Streamline code in region_loss.py. Two for loops are unnecessary