Skip to content

ChanChiChoi/SynthText_unicode_version

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 

Repository files navigation

SynthText_unicode_version

this project borrowed from Ankush, some code and setting referenced from SynthText_Chinese_version
this project can run on windows and linux environment.

requirement

python 3.5+
pygame
tensorflow 1.4.0+
cv2 3.2.0+

1 - if you want to use your text corpus

1.1 put the corpus in "SynthText_unicode_version/data/corpus/"
1.2 then you need update the file "SynthText_unicode_version/data/models/char_freq.cp", so run the command:

python tools/update_freq.py --corpusPath data/corpus --outputPath data/models

2 - if you want to use your font

2.1 put the font in "SynthText_unicode_version/data/fonts/"
2.2 then you need update the file "SynthText_unicode_version/data/models/font_px2pt.cp", so run the command:

 python tools/font_pixel2pt.py --fontDirPath data/fonts --outputPath data/models

PS: the file "SynthText_unicode_version/data/models/colors_new.cp" could not be update, because this is Color-model (foreground/background text color model), learnt from the IIIT-5K word dataset. the file is used to render the text when the program choose an foreground string from text corpus, then put the text onto a rander color background, so it's ok to use the file.

3 - if you want to add new background image

3.1 get the images' depth
put the new background images in "SynthText_unicode_version/data/bgi/", then

cd SynthText_unicode_version/tools/depthPrediction_tensorflow/code

read this or just directly download NYU_FCRN.ckpt, then decompress it into "SynthText_unicode_version/tools/depthPrediction_tensorflow/ckpt". directory structure like this:  

tree SynthText_unicode_version/tools/depthPrediction_tensorflow

.
├── ckpt
│   ├── NYU_FCRN.ckpt.data-00000-of-00001
│   ├── NYU_FCRN.ckpt.index
│   └── NYU_FCRN.ckpt.meta
├── code
│   ├── models
│   │   ├── fcrn.py
│   │   ├── init.py
│   │   └── network.py
│   └── predict.py
└── README.md
then run the command:

python predict.py --model_path ../ckpt/NYU_FCRN.ckpt --image_paths ../../../data/bgi --img_type jpg --output_path ../depth

file "SynthText_unicode_version/tools/depthPrediction_tensorflow/depth/depth.h5" is the depth prediction of this new background images

bgi1
origin image of bgi1.jpg
   
depth prediction of big1.png
 

PS:because depth prediction program will resize the image into (height=228,width=304), so it partly looks unlike the origin image

3.2 get the images' segmentation masks
the origin image should be splited into some blocks by edge lines, each block is a region. the image mask has blocks and lines, lines have value 0, each block has it's order value, e.g. from left to right, from up to down, the first region has value 1, the second region has value 2,...

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published