search

looput / PSENet-Tensorflow

TensorFlow implementation of PSENet text detector (Shape Robust Text Detection with Progressive Scale Expansion Networkt)
49 stars 10 forks source link

关于模型测试的问题 #12

Open Tian14267 opened 4 years ago

Tian14267 commented 4 years ago

作者你好，又来麻烦你了。我看了你的代码，感觉有点复杂，根据你的代码和whai362的代码于是写了一个简化版的psenet。新代码可以正常训练，但是测试无法通过。始终无法识别出Box。不知道作者有没有空，能不能向你请教下这方面问题？ 链接：https://github.com/Tian14267/PseNet_tensorflow 期待作者能够给个回复 其他的小伙伴如果感兴趣，也可以过来看看。

looput commented 4 years ago

你好，代码可能还需要你再检查下。

  1. 你说的训练正常是指什么? 是loss 下降正常，还是分割图的可视化是正常的
  2. 测试有问题具体是什么表现？score map可视化如何？另外建议检查下：数据预处理是否和训练保持一致；模型加载是否成功，特别是bn的参数
Tian14267 commented 4 years ago

作者你好，训练正常，是指loss正常（不过没有原作者那样下降的漂亮）。但是测试的话，box无法正确画出来，准确的说是，box是完全在瞎画。目前表现为，仅 loss 下降和 acc 正常。训练数据的预处理应该是没问题的，训练参数目前也没发现那里不对。 不知道作者有没有空，可以指点下吗？链接附在上面了 @looput