search

chineseocr / table-detect

table detect(yolo) , table line(unet)
MIT License
236 stars 86 forks source link

unet主干网络sigmoid选用 #12

Closed dcdethan closed 3 years ago

dcdethan commented 3 years ago

作者您好，感谢您对于表格识别的杰出工作，我这里现在有个问题 classify = Conv2D(num_classes, (1, 1), activation='sigmoid')(up0a) 这里既然num_classes是2，多分类问题了，为啥选用sigmoid作为激活函数呢？

dcdethan commented 3 years ago

作者您好，主干网络出来的结果shape（1024,1024,2）中的2为啥自动分为横线和竖线，如下 vpred = pred[...,1]>vprob##竖线 hpred = pred[...,0]>hprob##横线 请问这个计算原理是什么，是算两个分类吗