文件名称:OCR_tablenet:TableNet在Pytorch上的实现
文件大小:61.17MB
文件格式:ZIP
更新时间:2024-06-11 08:31:55
Python
表格网
该库由一个Pytorch实施 。
要进行培训或预测,您首先应该通过运行以下代码来安装需求:
pip install -r requirements.txt
要训练,只需要可以根据需要配置的train.py文件。 marmot.py和tablenet.py是Pytorch Lighting模块的继承:分别为LightningDataModule和LightningModule 。
为了进行预测,可以使用已经可用的预先训练的权重,并且应该在以下链接上下载:
python predict.py --model_weights= '
【文件预览】:
OCR_tablenet-master
----.flake8(31B)
----.gitignore(19B)
----predict.py(5KB)
----README.md(866B)
----mypy.ini(63B)
----train.py(2KB)
----requirements.txt(1KB)
----tablenet()
--------marmot.py(4KB)
--------__init__.py(155B)
--------tablenet.py(9KB)
----data()
--------data.zip(70.08MB)