Giters

opconty / pytorch_ctpn

This is a pytorch implementation of CTPN(Detecting Text in Natural Image with Connectionist Text Proposal Network). You may want to finetune from: https://drive.google.com/open?id=1JHhI4sEIXfs5gDa1I9AgJBY477HTzAd0

Home Page:https://mp.weixin.qq.com/s/VO42GzwwJBOabpPJOWVn4g

Repository from Github https://github.comopconty/pytorch_ctpnRepository from Github https://github.comopconty/pytorch_ctpn

为什么您在model的forward函数把输入的channel first和channel last来回倒？一直channel first不行吗？

HandsomeBrotherShuaiLi opened this issue · comments

commented

pytorch_ctpn/ctpn_model.py

Line 106 in d1723c6

x1 = x.permute(0,2,3,1).contiguous() # channels last

commented

我懂了，是因为LSTM输入的问题。