ljt019 commited on
Commit
203253a
1 Parent(s): 5eba4c0

dev: upload readme

Browse files
Files changed (1) hide show
  1. README.md +25 -3
README.md CHANGED
@@ -1,3 +1,25 @@
1
- ---
2
- license: mit
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