Animal-Detection-and-Recommendation
Animals Detection and Recommendation System using Yolov5 model and website using FastAPI and Jinja2 template.
- Classifies and detect animals in an image and recommends image based on detection.
- Collected Dataset from COCO Dataset, Kaggle, Pixabay, Unsplash
- 5 animals - Cat, Dog, Tiger, Deer and Elephant
- Trained the model using Multilabel Classification and Yolov5 model
- Web Implementation for uploading multiple images and detect animals
- Tools & Technologies: pandas, numpy, matplotlib, sklearn, keras, tensorflow, Yolov5, Jinja2 template, FastAPI, Roboflow