User Tools

Site Tools


auto2d_tutorial

Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Both sides previous revision Previous revision
Next revision
Previous revision
auto2d_tutorial [2019/12/10 08:59]
twagner
auto2d_tutorial [2020/08/28 07:36]
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, on containing good classes (e.g ''GOOD_CLASSES/'') and one contain bad classes (e.g ''BAD_CLASSES/''). Both folders can contain multiple .hdf / .mrcs files.+First you have to separate your good and bad classes into separate files. Create two folders, one containing good classes (e.g ''GOOD_CLASSES/'') and one contain bad classes (e.g ''BAD_CLASSES/''). Both folders can contain multiple .hdf / .mrcs files.
  
 Then specify the paths into a config file like this: Then specify the paths into a config file like this:
Line 28: Line 28:
 { {
  "model": {  "model": {
- "input_size": [75,75]+ "input_size": [64,64]
  },  },
  
  "train": {  "train": {
  "batch_size": 32,  "batch_size": 32,
- "good_classes": "GOOD_CLASSES/", + "good_path": "GOOD_CLASSES/", 
- "bad_classes": "BAD_CLASSES/",+ "bad_path": "BAD_CLASSES/",
  "pretrained_weights": "",  "pretrained_weights": "",
  "saved_weights_name": "my_model.h5",  "saved_weights_name": "my_model.h5",
Line 43: Line 43:
 } }
 </code> </code>
-The fields have the following meaning:+The fields in the section **model** have the following meaning:
   * **input_size**: This is the image size to which each class is resized to.   * **input_size**: This is the image size to which each class is resized to.
 +  * **mask_radius**: (Optional) Circular mask radius which is applied after resizing to the input size. If not given, it uses 0.4*input_size as default.
 +
 +The fields in the section **train** have the following meaning:
   * **batch_size**: How many classes are in one mini-batch. If you have memory problems, you can try to reduce this value.   * **batch_size**: How many classes are in one mini-batch. If you have memory problems, you can try to reduce this value.
   * **good_path**: Path to folder with good classes.   * **good_path**: Path to folder with good classes.
auto2d_tutorial.txt · Last modified: 2020/08/28 07:36 by twagner