dataset_config.yml 1.5 KB
Newer Older
W
weishengyu 已提交
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
Global:
  output_num: 10
  output_dir: output_data
  use_gpu: false
  image_height: 32
  image_width: 320
  standard_font: fonts/en_standard.ttf
TextDrawer:
  fonts:
    en: fonts/en_standard.ttf
    ch: fonts/ch_standard.ttf
    ko: fonts/ko_standard.ttf
StyleSampler:
  method: DatasetSampler
  image_home: examples
  label_file: examples/image_list.txt
  with_label: true
CorpusGenerator:
  method: FileCorpus
  language: ch
  corpus_file: examples/corpus/example.txt
Predictor:
  method: StyleTextRecPredictor
  algorithm: StyleTextRec
  scale: 0.00392156862745098
  mean:
  - 0.5
  - 0.5
  - 0.5
  std:
  - 0.5
  - 0.5
  - 0.5
  expand_result: false
  bg_generator:
W
weishengyu 已提交
36
    pretrain: models/style_text_rec/bg_generator
W
weishengyu 已提交
37 38 39 40 41 42 43 44 45
    module_name: bg_generator
    generator_type: BgGeneratorWithMask
    encode_dim: 64
    norm_layer: null
    conv_block_num: 4
    conv_block_dropout: false
    conv_block_dilation: true
    output_factor: 1.05
  text_generator:
W
weishengyu 已提交
46
    pretrain: models/style_text_rec/text_generator
W
weishengyu 已提交
47 48 49 50 51 52 53 54
    module_name: text_generator
    generator_type: TextGenerator
    encode_dim: 64
    norm_layer: InstanceNorm2D
    conv_block_num: 4
    conv_block_dropout: false
    conv_block_dilation: true
  fusion_generator:
W
weishengyu 已提交
55
    pretrain: models/style_text_rec/fusion_generator
W
weishengyu 已提交
56 57 58 59 60 61 62 63 64
    module_name: fusion_generator
    generator_type: FusionGeneratorSimple
    encode_dim: 64
    norm_layer: null
    conv_block_num: 4
    conv_block_dropout: false
    conv_block_dilation: true
Writer:
  method: SimpleWriter