add files needed for training

This commit is contained in:
b-vr103 2019-12-05 14:05:07 +01:00
parent 4601237427
commit 226330535d
4 changed files with 575 additions and 0 deletions

24
train/config_params.json Normal file
View file

@ -0,0 +1,24 @@
{
"n_classes" : 2,
"n_epochs" : 2,
"input_height" : 448,
"input_width" : 896,
"weight_decay" : 1e-6,
"n_batch" : 1,
"learning_rate": 1e-4,
"patches" : true,
"pretraining" : true,
"augmentation" : false,
"flip_aug" : false,
"elastic_aug" : false,
"blur_aug" : false,
"scaling" : false,
"binarization" : false,
"scaling_bluring" : false,
"scaling_binarization" : false,
"rotation": false,
"weighted_loss": true,
"dir_train": "/home/vahid/textline_gt_images/train_light",
"dir_eval": "/home/vahid/textline_gt_images/eval",
"dir_output": "/home/vahid/textline_gt_images/output"
}