【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同级)

4.修改待测图片文件名

5.这样就能顺利跑出结果了

posted @ 2023-01-06 23:32  faf4r  阅读(436)  评论(0)    收藏  举报