mirror of
https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
synced 2025-06-09 20:00:05 +02:00
updating train.py
This commit is contained in:
parent
dc356a5f42
commit
815e5a1d35
1 changed files with 2 additions and 0 deletions
2
train.py
2
train.py
|
@ -59,6 +59,8 @@ def config_params():
|
||||||
pretraining = False # Set to true to load pretrained weights of ResNet50 encoder.
|
pretraining = False # Set to true to load pretrained weights of ResNet50 encoder.
|
||||||
scaling_bluring = False # If true, a combination of scaling and blurring will be applied to the image.
|
scaling_bluring = False # If true, a combination of scaling and blurring will be applied to the image.
|
||||||
scaling_binarization = False # If true, a combination of scaling and binarization will be applied to the image.
|
scaling_binarization = False # If true, a combination of scaling and binarization will be applied to the image.
|
||||||
|
rotation = False # If true, a 90 degree rotation will be implemeneted.
|
||||||
|
rotation_not_90 = False # If true rotation based on provided angles with thetha will be implemeneted.
|
||||||
scaling_brightness = False # If true, a combination of scaling and brightening will be applied to the image.
|
scaling_brightness = False # If true, a combination of scaling and brightening will be applied to the image.
|
||||||
scaling_flip = False # If true, a combination of scaling and flipping will be applied to the image.
|
scaling_flip = False # If true, a combination of scaling and flipping will be applied to the image.
|
||||||
thetha = None # Rotate image by these angles for augmentation.
|
thetha = None # Rotate image by these angles for augmentation.
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue