dataset_config.yml 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. Global:
  2. output_num: 10
  3. output_dir: output_data
  4. use_gpu: false
  5. image_height: 32
  6. image_width: 320
  7. standard_font: fonts/en_standard.ttf
  8. TextDrawer:
  9. fonts:
  10. en: fonts/en_standard.ttf
  11. ch: fonts/ch_standard.ttf
  12. ko: fonts/ko_standard.ttf
  13. StyleSampler:
  14. method: DatasetSampler
  15. image_home: examples
  16. label_file: examples/image_list.txt
  17. with_label: true
  18. CorpusGenerator:
  19. method: FileCorpus
  20. language: ch
  21. corpus_file: examples/corpus/example.txt
  22. Predictor:
  23. method: StyleTextRecPredictor
  24. algorithm: StyleTextRec
  25. scale: 0.00392156862745098
  26. mean:
  27. - 0.5
  28. - 0.5
  29. - 0.5
  30. std:
  31. - 0.5
  32. - 0.5
  33. - 0.5
  34. expand_result: false
  35. bg_generator:
  36. pretrain: style_text_models/bg_generator
  37. module_name: bg_generator
  38. generator_type: BgGeneratorWithMask
  39. encode_dim: 64
  40. norm_layer: null
  41. conv_block_num: 4
  42. conv_block_dropout: false
  43. conv_block_dilation: true
  44. output_factor: 1.05
  45. text_generator:
  46. pretrain: style_text_models/text_generator
  47. module_name: text_generator
  48. generator_type: TextGenerator
  49. encode_dim: 64
  50. norm_layer: InstanceNorm2D
  51. conv_block_num: 4
  52. conv_block_dropout: false
  53. conv_block_dilation: true
  54. fusion_generator:
  55. pretrain: style_text_models/fusion_generator
  56. module_name: fusion_generator
  57. generator_type: FusionGeneratorSimple
  58. encode_dim: 64
  59. norm_layer: null
  60. conv_block_num: 4
  61. conv_block_dropout: false
  62. conv_block_dilation: true
  63. Writer:
  64. method: SimpleWriter