data source:-https://www.kaggle.com/arnaudeq/cats-vs-dogs-1000
-
I used tranfer learning model Vgg16 for training our dataset for further predictions.
-
What is Vgg16?
- VGG16 (also called OxfordNet) is a convolutional neural network architecture named after the Visual Geometry Group from Oxford, who developed it. It was used to win the ILSVR (ImageNet) competition in 2014.
- I got an accuracy score of almost 91% after training the dataset and testing on the validation dataset.