是否可以整合成一套框架

MrZilinXiao / Hyper-Table-OCR

A carefully-designed OCR pipeline for universal boarded table recognition and reconstruction.

https://mrxiao.net/hyper-table-ocr.html

168 stars 43 forks source link

Open huwei1024 opened 3 years ago

huwei1024 commented 3 years ago

是否可以整合成一套框架?把tf代码转成pytorch?

MrZilinXiao commented 3 years ago

您好，感谢对我工作的兴趣。 UNet本身是很好用torch实现的，但由于UNet分割线训练数据缺失，此部分需要载入第三方权重，因此没有做推理框架上的整合，见谅。

huwei1024 commented 3 years ago

后期有没有可能整合呢？装几套框架太麻烦了。

MrZilinXiao commented 3 years ago

如果您能提供标注好的表格数据集，我很乐意做一版只使用一套框架的Hyper-Table-OCR；如果您的这个需求十分迫切，您可以考虑将tf的权重转换为onnx后用torch载入，或者直接考虑转换为torch权重。