Augmentation
- class plugins.train.trainer.trainer_config.Augmentation(helptext: str)
Bases:
GlobalSectiontrainer.augmentation section
Attributes Summary
Attributes Documentation
- Parameters:
helptext (str)
- color_ab = ConfigItem(datatype=<class 'int'>, default=8, group='color augmentation', info="Percentage amount to randomly alter the 'a' and 'b' colors of the L*a*b* color space of each training image.\nNB: This is ignored if the 'no-augment-color' option is enabled", choices=[], gui_radio=False, min_max=(0, 50), rounding=1, fixed=True, _value=8, _name='')
- Return type:
- color_clahe_chance = ConfigItem(datatype=<class 'int'>, default=50, group='color augmentation', info="Percentage chance to perform Contrast Limited Adaptive Histogram Equalization on each training image.\nNB: This is ignored if the 'no-augment-color' option is enabled", choices=[], gui_radio=False, min_max=(0, 75), rounding=1, fixed=False, _value=50, _name='')
- Return type:
- color_clahe_max_size = ConfigItem(datatype=<class 'int'>, default=4, group='color augmentation', info="The grid size dictates how much Contrast Limited Adaptive Histogram Equalization is performed on any training image selected for clahe. Contrast will be applied randomly with a grid-size of 0 up to the maximum. This value is a multiplier calculated from the training image size.\nNB: This is ignored if the 'no-augment-color' option is enabled", choices=[], gui_radio=False, min_max=(1, 8), rounding=1, fixed=True, _value=4, _name='')
- Return type:
- color_lightness = ConfigItem(datatype=<class 'int'>, default=30, group='color augmentation', info="Percentage amount to randomly alter the lightness of each training image.\nNB: This is ignored if the 'no-augment-color' option is enabled", choices=[], gui_radio=False, min_max=(0, 75), rounding=1, fixed=True, _value=30, _name='')
- Return type:
- flip_chance = ConfigItem(datatype=<class 'int'>, default=50, group='image augmentation', info="Percentage chance to randomly flip each training image horizontally.\nNB: This is ignored if the 'no-flip' option is enabled", choices=[], gui_radio=False, min_max=(0, 75), rounding=1, fixed=True, _value=50, _name='')
- Return type:
- helptext: str = 'Data Augmentation Options.\nWARNING: The defaults for augmentation will be fine for 99.9% of use cases. Only change them if you absolutely know what you are doing!'
- mask_color = ConfigItem(datatype=<class 'str'>, default='#ff0000', group='evaluation', info='The RGB hex color to use for the mask overlay in the training preview.', choices='colorchooser', gui_radio=False, min_max=None, rounding=-1, fixed=True, _value='#ff0000', _name='')
- Return type:
- mask_opacity = ConfigItem(datatype=<class 'int'>, default=30, group='evaluation', info='The opacity of the mask overlay in the training preview. Lower values are more transparent.', choices=[], gui_radio=False, min_max=(0, 100), rounding=2, fixed=True, _value=30, _name='')
- Return type:
- preview_images = ConfigItem(datatype=<class 'int'>, default=14, group='evaluation', info='Number of sample faces to display for each side in the preview when training.', choices=[], gui_radio=False, min_max=(2, 16), rounding=2, fixed=True, _value=14, _name='')
- Return type:
- rotation_range = ConfigItem(datatype=<class 'int'>, default=10, group='image augmentation', info='Percentage amount to randomly rotate each training image.', choices=[], gui_radio=False, min_max=(0, 25), rounding=1, fixed=True, _value=10, _name='')
- Return type:
- shift_range = ConfigItem(datatype=<class 'int'>, default=5, group='image augmentation', info='Percentage amount to randomly shift each training image horizontally and vertically.', choices=[], gui_radio=False, min_max=(0, 25), rounding=1, fixed=True, _value=5, _name='')
- Return type: