- LMDB of the Dataset should be stored in dataset folder
- Model Checkpoints will be outputted in the models file
Inspired and taken help from various pytorch and caffe implementations available over the internet specifically This one .
Follow the main.ipynb for training
Incase you want a pretrained model you can refer to this.
Further Details will be added later ! Stay Tuned 😀.