To run the code succefully first create a folder and copy all the python or ipython notebook to the respective folder along with train and test dataset. Then navigate to that folder and run jupyter notebook inside that folder. make sure to change the path of your train and test set. The testset is provided. The link of training set is attached herewith https://drive.google.com/file/d/12_WTFi9ppvD-loaWUWpUar25Z3nT5k9P/view.
Download the trained weight file from the link given below.
https://drive.google.com/file/d/1lycxJWYk9Iccitk_2Ktl9F6ICNZa7_7X/view?usp=sharing
make sure to run the code with GPU support