Course project for vehicle image classification
We implemented this task using VGG16 transfer learning. The detailed information is on https://www.kaggle.com/carloalbertobarbano/vgg16-transfer-learning-pytorch
- Clone this repository.
- Download the pre-trained weight vgg16_bn.pth from Kaggle. Put vgg16_bn.pth and the cloned repo in a same directory, and copy the given deploy/ folder into the cloned repo.
- Run vgg16_transfer.py