about crnn predict.py - Githubissues

Rid7 / Table-OCR

Recognize tables from images and restore them into word.

GNU General Public License v3.0

271 stars 69 forks source link

Closed dcdethan closed 3 years ago

dcdethan commented 4 years ago

你好，请教一下，在predict.py中的56行，读入的一张图的tensor，怎么读入多张图一个batch作为tensor去做预测呢？谢谢！

Rid7 commented 4 years ago

你好，请教一下，在predict.py中的56行，读入的一张图的tensor，怎么读入多张图一个batch作为tensor去做预测呢？谢谢！

可以用torch的Dataloader类实现，类似于训练过程

dcdethan commented 4 years ago

多谢多谢。我根据demo改好了多batch的。