The project is all about learning the pytorch and build a CNN from scratch. In this project I have created the Dog Breed Classifier, which can identify the breed of the dog if the dog is detected in the image and also can detect the human face if it is there in the image.
Python 3.5+ , pytorch 1.0+, torchvision 0.6+, facenet_pytorch
-
Clone the repository. create the
data/
folder. -
Download the dog dataset. Unzip the folder and place it in the repo, at location
data/dog_images
. Thedog_images/
folder should contain 133 folders, each corresponding to a different dog breed. and store it -
Download the human dataset. Unzip the folder and place it in the repo, at location
data/lfw
. If you are using a Windows machine, you are encouraged to use 7zip to extract the folder. -
Make sure you have already installed the necessary Python packages according to the README in the program repository.
-
Open a terminal window and navigate to the project folder. Open the notebook and follow the instructions.
jupyter notebook dog_app.ipynb