Skip to content

Tian14267/PseNet_tensorflow

Repository files navigation

PseNet_tensorflow

说明

本代码是为PSENET检测网络的tensorflow版本。主要仅供学习该算法使用。代码主要参考了以下链接的代码:
链接一:https://github.com/whai362/PSENet
链接二:https://github.com/looput/PSENet-Tensorflow
链接一为原始链接,主要使用pytorch实现;链接二为tensorflow版本的代码。相比较两个链接的版本,通过本人使用和测试发现,链接一的pytorch效果略优于链接二tensorflow。
对于链接二,本人感觉其代码写的较为复杂,阅读学习性不太高,于是在这里改了一个简易版本的 Psenet_tensorflow

Data

训练数据为 ICPR 数据集,本文这里也公开所用到的数据集。
以下为训练所使用的训练数据:
链接:https://pan.baidu.com/s/1IRc-FAhb7cXmsycjAmO1cw 提取码:k8je

环境

python3
tensorflow 1.14.0

模型链接

链接:https://pan.baidu.com/s/1fRhuXWRfhsOT5FrhDBNV4g 提取码:pxuw
注:kernels = 3

训练细节

image

测试结果展示

image

image

About

基于tensorflow的PSENet简化版

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages