dev: upload readme
Browse files
README.md
CHANGED
@@ -1,3 +1,25 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Cat and Dog Sketch Classifier
|
2 |
+
|
3 |
+
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.
|
4 |
+
|
5 |
+
## Project Structure
|
6 |
+
|
7 |
+
- `quickdraw_data/` - Directory containing `cat.npy` and `dog.npy` files.
|
8 |
+
- `cat_dog_classifier.bin` - Trained model file.
|
9 |
+
- `config.json` - Configuration file with parameters for the model and training.
|
10 |
+
- `model.py` - Contains the model definition for the Convolutional Neural Network (CNN).
|
11 |
+
- `sample_predictions.png` - Image file with sample predictions from the model.
|
12 |
+
- `train_cat_dog_classifier.py` - Script to train the classifier.
|
13 |
+
- `requirements.txt` - Python Dependencies needed to run the training script
|
14 |
+
|
15 |
+
## Dependencies
|
16 |
+
|
17 |
+
To install the required Python packages to train the model, use the following command:
|
18 |
+
|
19 |
+
```bash
|
20 |
+
pip install -r requirements.txt
|
21 |
+
```
|
22 |
+
|
23 |
+
## License
|
24 |
+
|
25 |
+
MIT
|