diff --git a/README.md b/README.md index f9362906e6f0e4cc1b3ed20965285b9093ff881b..d22d58f3a0514c7ab184a0aa956d5de0c9e97d5c 100644 --- a/README.md +++ b/README.md @@ -77,9 +77,19 @@ $ ./process_train_data video_folder train_folder To prepare the training data for the model : ```bash -$ python music_processor.py train train_folder +$ python cnn_prepare_data.py train train_folder ``` +Prepared data will be stored in `./data/pickles/train_data.pickle` + +To train the model on the prepared data : +```bash +$ python cnn_train.py +``` + +The model will be written to `./models/model.pth` + + ## Infer To execute AutoKara on a MKV video file :