![Logo](/gerardocipriano/Cerberus-Dog-Breed-Classification-and-Body-Localization-PyTorch/raw/main/res/secondLogo.png)
The goal of the project is to develop a PyTorch-based model capable of recognizing and classifying 3 dog breeds and localizing 3 specific body parts within images.
View Demo
·
Report Bug
·
Request Feature
Table of Contents
Cerberus is a PyTorch-based project for classifying dog breeds and localizing specific body parts within images. Our model is capable of recognizing and classifying 3 dog breeds (Toy Poodle, Beagle, Siberian Husky) and localizing 3 specific body parts (eyes, nose, tail) within images. This repository contains the code and resources for training and evaluating the model, as well as a user interface for interacting with the algorithm.
- Just Clone/Download the repo
Classification and Localization functionalities are available in their respective user interfaces (GUIs).
To utilize the classification feature, please follow these steps:
- Install the required dependencies mentioned in the .txt file.
- Run the .\main.py file.
- Within the classification GUI, you have the option to create a new alexnet model from scratch. This model can then be used for training, validation, or prediction purposes. If you already possess a pre-existing model, you can select it within the interface. Following the training phase, you can evaluate the model's performance using a cat dataset and examine the results.
If you wish to utilize the localization feature, please run the .\Object-detection\main.py file. (Note: A unified GUI will be introduced in a future version of the project.)
- In the localization GUI, you can upload an image of a dog and the model. This allows you to precisely identify and localize various body parts.
- Additionally, you can monitor the output in real-time during this process by selecting the "Show model validation data" option.
Mandatory features:
- Recognizing and classifying 3 dog breeds (Toy Poodle, Beagle, Siberian Husky)
- Localizing 3 specific body parts (eyes, nose, tail)
Optional features:
- GUI
Distributed under the MIT License. See LICENSE
for more information.
Gerardo Cipriano - [email protected]
Massimiliano Battelli - [email protected]
Project Link: https://github.com/gerardocipriano/Cerberus-Dog-Breed-Classification-and-Body-Localization-PyTorch