xuewenyuan / TGRNet

TGRNet: A Table Graph Reconstruction Network for Table Structure Recognition

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

数据集优有一点小问题

hsdjkfnsfc opened this issue · comments

作者您的数据集当中,在pickle文件以及text文件当中使用多个是您本机的绝对路径,希望您更改一下。

同时我在运行您的test文件的时候发现Cell Spatial Location的结果与您的论文一致,但是Cell Logical Location完全不一致
image
image

@hsdjkfnsfc 谢谢反馈,请问您是从哪个链接下载的数据集和模型,我看一下

可以写个脚本自行修改一下根目录,我排查了一下,是每个目录下的train.txt,val.txt, test.txt, 还有gt文件夹下的每个pkl里面的image_path字段,这些地方的路径都可以替换为相对路径,或者自己的绝对路径

@EmperorKaiser 谢谢。
@hsdjkfnsfc 抱歉回复比较晚。我在cuda10.2下复现了你的问题,但是切换到cuda9.2后,结果正常。所以,很可能是不同cuda版本下,软件的影响,特别时torch geometric。
我的测试环境:

  • cuda 9.2
  • torch 1.7.0+torchvision 0.8.0
  • torch-cluster 1.5.9
  • torch-geometric 1.6.3
  • torch-scatter 2.0.6
  • torch-sparse 0.6.9
  • torch-spline-conv 1.2.1

@hsdjkfnsfc @xuewenyuan @EmperorKaiser 数据集是不是少了一个graph_edge的文件夹,只有gt、graph_node和graph_target文件夹

您好,数据集graph_target缺少这个怎么解决呢? @xuewenyuan