diff --git a/StyleText/README.md b/StyleText/README.md index f978f43875f3994c7016fbf6cff1a1c0f2de4588..43c48208c345366fbd8c77c66203905b68de969f 100644 --- a/StyleText/README.md +++ b/StyleText/README.md @@ -138,7 +138,18 @@ We provide a general dataset containing Chinese, English and Korean (50,000 imag ``` bash python -m tools.synth_dataset.py -c configs/dataset_config.yml ``` - +We also provide example corpus and images in `example` folder. +
+ + +
+If you run the code above directly, you will get example output data in `output_data` folder. +You will get synthesis images and labels as below: +
+ +
+There will be some cache under the `label` folder. If the program exit unexpectedly, you can find cached labels there. +When the program finish normally, you will find all the labels in `label.txt` which give the final results. ### Applications diff --git a/StyleText/README_ch.md b/StyleText/README_ch.md index a8ab933bd38792bcd6aecd76c85fc3fd1f5f204f..c25290ae5fb311f6f60bbd26be7383edd86716f6 100644 --- a/StyleText/README_ch.md +++ b/StyleText/README_ch.md @@ -126,6 +126,19 @@ python3 -m tools.synth_image -c configs/config.yml --style_image examples/style_ ``` bash python -m tools.synth_dataset -c configs/dataset_config.yml ``` + 我们在example提供了样例图片和语料在example中。 +
+ + +
+ + 直接运行上述命令,可以在output_data中产生样例输出,包括图片和用于训练识别模型的标注文件: +
+ +
+ + 其中label目录下的标注文件为程序运行过程中产生的缓存,如果程序在中途异常终止,可以使用缓存的标注文件。 + 如果程序正常运行完毕,则会在output_data下生成label.txt,为最终的标注结果。 ### 四、应用案例 diff --git a/StyleText/configs/dataset_config.yml b/StyleText/configs/dataset_config.yml index e047489e5d82e4c561a835ccf4de1b385e4f5c08..aa4ec69b8ce383d16d505ec70ea6f75488a6fd17 100644 --- a/StyleText/configs/dataset_config.yml +++ b/StyleText/configs/dataset_config.yml @@ -33,7 +33,7 @@ Predictor: - 0.5 expand_result: false bg_generator: - pretrain: models/style_text_rec/bg_generator + pretrain: style_text_models/bg_generator module_name: bg_generator generator_type: BgGeneratorWithMask encode_dim: 64 @@ -43,7 +43,7 @@ Predictor: conv_block_dilation: true output_factor: 1.05 text_generator: - pretrain: models/style_text_rec/text_generator + pretrain: style_text_models/text_generator module_name: text_generator generator_type: TextGenerator encode_dim: 64 @@ -52,7 +52,7 @@ Predictor: conv_block_dropout: false conv_block_dilation: true fusion_generator: - pretrain: models/style_text_rec/fusion_generator + pretrain: style_text_models/fusion_generator module_name: fusion_generator generator_type: FusionGeneratorSimple encode_dim: 64 diff --git a/StyleText/examples/corpus/example.txt b/StyleText/examples/corpus/example.txt index 78451cc3d92a3353f5de0c74c2cb0a06e6197653..93ba35af3bae1c5a81166a22a6d69b460b9ff5e5 100644 --- a/StyleText/examples/corpus/example.txt +++ b/StyleText/examples/corpus/example.txt @@ -1,2 +1,2 @@ -PaddleOCR +Paddle 飞桨文字识别 diff --git a/StyleText/examples/style_images/12.png b/StyleText/examples/style_images/12.png new file mode 100644 index 0000000000000000000000000000000000000000..d53c96d31f6c4cafb79a7342baa8e673d649be53 Binary files /dev/null and b/StyleText/examples/style_images/12.png differ