【table master mmocr】Windows下demo运行配置
运行TableMASTER-mmocr-master\table_recognition\demo\demo.py
文件目录:TableMASTER-mmocr-master\table_recognition\demo\demo.py
1.alphabet
把TableMASTER-mmocr-master\table_recognition\demo\alphabet
里面的文件复制到TableMASTER-mmocr-master\tools\data\alphabet
,tools里面的这个在跑demo的时候会用到
应在此处添加
2.预训练模型
从readme找到预训练模型下载,在TableMASTER-mmocr-master
的readme里,下载完后建议放到dataset\demo_model_v1
这样命名的文件夹,同时在这个文件夹下新建一个outputs
文件夹。
其中epoch_16_0.7767.pth
要改成tablemaster_best.pth
,这是根据demo原文件和readme推测的。
在源文件README.md里面,第一个pretrained模型,文件名epoch_16_0.7767.pth
,根据下面描述和demo.py需要路径推测,这就是训练出的某个模型作为tablemaster_best.pth
(最佳模型)。
原文描述如下:
The **TableMASTER** (TableMASTER_maxlength_500) pretrained model. In the validation set, the accuracy is **0.7767**.
[[Google Drive]](https://drive.google.com/file/d/1LSuVQJ0J8WFtXhLfcCKyzGqcCYmcwEk6/view?usp=sharing)
[[BaiduYun Drive]](https://pan.baidu.com/s/1G2tBpycZY6c6wzfE3V9khw) code:**irp6**
整个目录结构:
│TableMASTER-mmocr-master
│
│dataset
│
└─demo_model_v1
│ master_epoch_6.pth
│ pse_epoch_600.pth
│ tablemaster_best.pth
│
└─outputs
3.文件路径修改
修改demo.py的默认路径,改成你自己设置的dataset的路径。
原始路径都是/data_0/dataset
打头,我的配置是将其改为../dataset
即可(我的dataset
目录与TableMASTER-mmocr-master
同级)