Sanster / text_renderer

Generate text images for training deep learning ocr model
MIT License
1.39k stars 385 forks source link

使用自己的语料数据来生成图片程序卡住,求助! #120

Open anonymous530 opened 11 months ago

anonymous530 commented 11 months ago

Hi 作者大大,我这里自己生成了上万条文字数据,想对这个文字数据合成图片数据。目前尝试使用一部分预料数据(10条)来生成,目前看速度非常慢,请问是什么问题呢?最后一直卡住了。

执行命令:python main.py --corpus_dir=corpus_dir --corpus_mode=list --gpu

程序输出:

Total fonts num: 1
Background num: 1
Loading corpus from: corpus_dir
Load corpus: corpus_dir/com_dataset_part.txt
Total lines: 10
Generate text images in ./output/default
start_index:  0

机器信息:16核CPU

Kagi217 commented 2 months ago

我在生成時也完全卡住 程序輸出: Total fonts num: 1 Background num: 2 Loading corpus from: ./data/corpus Loading chn corpus: 1/1 Total fonts num: 1 Background num: 2 也不報錯,不知道是哪裡出了問題

Sanster commented 2 months ago

我在生成時也完全卡住 程序輸出: Total fonts num: 1 Background num: 2 Loading corpus from: ./data/corpus Loading chn corpus: 1/1 Total fonts num: 1 Background num: 2 也不報錯,不知道是哪裡出了問題

卡主应该是这里一直在 retry,这里的 retry 没打异常 https://github.com/Sanster/text_renderer/blob/d048a9488760b20840135d8d0c170bf3a9cd5e13/textrenderer/renderer.py#L500 可以把 retry 去了试下

Kagi217 commented 2 months ago

我在生成時也完全卡住程式輸出:Total fonts num: 1 Background num: 2 Loading corpus from: ./data/corpus Loading chn corpus: 1/1 Total fonts num: 1 Background num: 2報錯,也不知道是哪裡生長問題

卡主應該是這裡一直在重試,這裡的重試沒打異常

https://github.com/Sanster/text_renderer/blob/d048a9488760b20840135d8d0c170bf3a9cd5e13/textrenderer/renderer.py#L500

可以把重試去試下

問題解決了,感謝