- This code was written as part of mini project
- The Notebook titled Training.ipynb has all the code for training the model.Additionally it can also be found on kaggle here
- The script titled SingleInference.py has all the code for running inference on single image.Please make sure to load your pretrained weights before running the script.
- Refer pdf for further explaination
- Author : Shreyas Bhat 181ME175 (Batch of 2022 NITK)
darthroco / cs351-ml Goto Github PK
View Code? Open in Web Editor NEWThis code was written as part of mini project