-
Notifications
You must be signed in to change notification settings - Fork 55
/
Copy pathmjsynth.yml
37 lines (35 loc) · 1.18 KB
/
mjsynth.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
dataset_builder: &ds_builder
table_path: 'example/table.txt'
# If change image height(32), change the net.
# If image width is not null, the image will be distorted.
img_shape: [32, null, 3]
# The image that width greater than max img_width will be dropped.
# Only work with image width is null.
max_img_width: 300
ignore_case: true
train:
dataset_builder:
<<: *ds_builder
train_ann_paths:
- '/datasets/MJSynth/mnt/ramdisk/max/90kDICT32px/annotation_train.txt'
- '/datasets/MJSynth/mnt/ramdisk/max/90kDICT32px/annotation_val.txt'
val_ann_paths:
- '/datasets/MJSynth/mnt/ramdisk/max/90kDICT32px/annotation_test.txt'
batch_size_per_replica: 256
# Number of epochs to train.
epochs: 20
lr_schedule:
initial_learning_rate: 0.0001
decay_steps: 600000
alpha: 0.01
# TensorBoard Arguments
# https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/TensorBoard#arguments_1
tensorboard:
histogram_freq: 1
profile_batch: 0
eval:
dataset_builder:
<<: *ds_builder
ann_paths:
- '/datasets/ICDAR/2013/Challenge2_Test_Task3_gt.txt'
batch_size: 128