qurator-spk
/
sbb_pixelwise_segmentation
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git
1
0
Fork 0
Code Issues Releases Wiki Activity

updating train.py

Browse Source
pull/18/head
vahidrezanezhad 7 months ago
parent dc356a5f42
commit 815e5a1d35
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File

2
train.py
Unescape Escape View File

@ -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.

Write Preview
Loading…
Cancel
Save