From bed06eb99cb1eabd815ce567d2e7ba0423dc0432 Mon Sep 17 00:00:00 2001 From: Sting <loic.allegre@ensiie.fr> Date: Fri, 23 Jun 2023 11:50:55 +0200 Subject: [PATCH] Update README --- README.md | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index f936290..d22d58f 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 : -- GitLab