This shows you the differences between two versions of the page.
Both sides previous revision Previous revision Next revision | Previous revision Next revision Both sides next revision | ||
auto2d_tutorial [2019/12/10 08:48] twagner |
auto2d_tutorial [2019/12/12 16:37] twagner [Training] |
||
---|---|---|---|
Line 21: | Line 21: | ||
==== Training ==== | ==== Training ==== | ||
If you would like to train Cinderella with your own classes, you can easily do it. | If you would like to train Cinderella with your own classes, you can easily do it. | ||
- | First you have to separate your good and bad classes into separate files. Create two folders, | + | First you have to separate your good and bad classes into separate files. Create two folders, |
Then specify the paths into a config file like this: | Then specify the paths into a config file like this: | ||
Line 33: | Line 33: | ||
" | " | ||
" | " | ||
- | "good_classes": " | + | "good_path": " |
- | "bad_classes": " | + | "bad_path": " |
" | " | ||
" | " | ||
Line 46: | Line 46: | ||
* **input_size**: | * **input_size**: | ||
* **batch_size**: | * **batch_size**: | ||
- | * **good_classes**: Path to folder with good classes. | + | * **good_path**: Path to folder with good classes. |
- | * **bad_classes**: Path to folder with bad classes. | + | * **bad_path**: Path to folder with bad classes. |
- | * **pretrained_weights**: | + | * **pretrained_weights**: |
* **saved_weights_name**: | * **saved_weights_name**: | ||
* **learning_rate**: | * **learning_rate**: | ||
* **nb_epoch**: | * **nb_epoch**: | ||
- | * **nb_early_stop**: | + | * **nb_early_stop**: |
The next step is to run the training: | The next step is to run the training: |