File size: 960 Bytes
203253a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# Cat and Dog Sketch Classifier
This project contains a machine learning model that differentiates between sketches of cats and dogs. It was created as a learning exercise to understand how AI models work and how to train them.
## Project Structure
- `quickdraw_data/` - Directory containing `cat.npy` and `dog.npy` files.
- `cat_dog_classifier.bin` - Trained model file.
- `config.json` - Configuration file with parameters for the model and training.
- `model.py` - Contains the model definition for the Convolutional Neural Network (CNN).
- `sample_predictions.png` - Image file with sample predictions from the model.
- `train_cat_dog_classifier.py` - Script to train the classifier.
- `requirements.txt` - Python Dependencies needed to run the training script
## Dependencies
To install the required Python packages to train the model, use the following command:
```bash
pip install -r requirements.txt
```
## License
MIT
|