A model to automatically identify and classify animals in images. It is trained on the Animals Detection Images Dataset from Kaggle. It relies on the pre-trained weights of EfficientNet B7.
The notebook on Kaggle can be viewed on https://www.kaggle.com/code/richardodhiambo/animal-detection-model
- Tensorflow - 2.14.0
- Numpy - 1.23.5
- Pandas - 1.5.3
Transfer learning from pre-trained weights of EfficientNet B7
- https://towardsdatascience.com/the-w3h-of-alexnet-vggnet-resnet-and-inception-7baaaecccc96
- https://medium.com/@enrico.randellini/image-classification-resnet-vs-efficientnet-vs-efficientnet-v2-vs-compact-convolutional-c205838bbf49
- https://keras.io/examples/vision/image_classification_efficientnet_fine_tuning/
- https://theaisummer.com/cnn-architectures/
This project is licensed under the MIT License.