Scripts
Data preparation and model training scripts.
Files
prepare_data.py- Download and preprocess the Quick Draw! datasettrain_model.py- Train the CNN modelcheck_classes_mapping.py- Verify the number of output classes in a trained model
Usage
# 1. Download and preprocess data (~2GB download, configurable sample count)
python scripts/prepare_data.py
# 2. Train the model
python scripts/train_model.py
Configuration
Paths and training parameters are defined in config.py at the project root.
Output
prepare_data.pywrites processed splits todataset/processed/and class mappings tomodel/classes.jsontrain_model.pysaves the best model tomodel/best_model.kerasand training plots tooutputs/