This project uses TensorFlow 2.0 and transfer learning to classify different breeds of dogs. The data used comes from the Kaggle dog breed identification competition, which includes over 10,000 labeled images of 120 different dog breeds.
These instructions will guide you in setting up the project and running it on your local machine.
Before starting, you will need to have the following installed:
- Python 3.x
- TensorFlow 2.x
- TensorFlow Hub
- Kaggle (for downloading the dataset)
- Clone the repository
Clone this repository to your local machine. You can use the following command:
git clone https://github.com/pawaspy/dog-breed-classification.git
- Install the required Python packages
Navigate to the cloned repository and install the required packages using pip:
- Download the data
Use the Kaggle API to download the dog breed identification competition data:
kaggle competitions download -c dog-breed-identification
Unzip the downloaded file and place it in a directory accessible by the script.
- Run the script
Run the main script (e.g., main.py
) in the root directory of the project:
python main.py
Contributions are what make the open-source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature
) - Commit your Changes (
git commit -m 'Add some AmazingFeature'
) - Push to the Branch (
git push origin feature/AmazingFeature
) - Open a Pull Request