Implementation of paper using pytorch on this dataset for final project of deep learning course.
The following steps were done in the project:
-
Data augmentation methods were used to increase the size of training data.
-
The pre-trained ResNet-18 was utilized and with approach of transfer learning, the model was fine-tuned on the new data.