The data set used is --> https://www.kaggle.com/competitions/dogs-vs-cats/data
For this classification we used transfer learning
The pre-trained model used in this is https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/classification/5
MobileNet V2 is a family of neural network architectures for efficient on-device image classification and related tasks, originally published by Mark Sandler, Andrew Howard, Menglong Zhu, Andrey Zhmoginov, Liang-Chieh Chen: "Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation", 2018.