config.yml 1.2 KB

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