You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
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.
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.
Classification and Localization functionalities are available in their respective user interfaces (GUIs).
Classification
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.
Localization
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.
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.