Skip to content
Extraits de code Groupes Projets

CNN segmentation

Fusionnées Sting a demandé de fusionner cnn-segmentation vers master
1 file
+ 11
1
Comparer les modifications
  • Côte à côte
  • En ligne
+ 11
1
@@ -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 :
Chargement en cours