satyenrajpal/complex-yolo

Implementation of Complex-YOLO 3D in PyTorch

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

complex-yolo's Issues

How to choose anchor boxes in Complex-YOLO/config.json ?

How you have designed anchors using

https://github.com/satyenrajpal/Complex-YOLO/blob/master/config.json#L11

Doubt Regarding the get_target Function

Why is the get_target function returning an array of shape(50,7)

RegionLoss Function error

Hi @satyenrajpal @krishnatoshniwal in the region loss .py the class prob and conf prob are not considered for the checking the IOU in the build target function , should we make some changes there

Loss function

@satyenrajpal @krishnatoshniwal Hi thanks for this repo , i have following few queries

1. Can you please share what is the changes made in your code and the repo mentioned
2. When i train a model in the repo mentioned i get negative tensors for 7:14

Thanks in advance