It's currently a work in progress!
To-Do:
- Config for kicking off training. a. Pretrained model b. Train Configuration c. Save directory d. Evaluation Script e. Inference Script
- Loss a. Implement and verify all kinds of losses. b. Implement a config for losses.
- Loading from pre-trained models from darknet
- Conversion to TensorRT.
Tasks:
- Write a train step, verify distribution strategy.
- Overfit on one image
- Conversion to TensorRT
- Eval/Predict step.