怎么把densenet换成其他分类网络？

YCG09 / chinese_ocr

CTPN + DenseNet + CTC based end-to-end Chinese OCR implemented using tensorflow and keras

Apache License 2.0

2.76k stars 1.08k forks source link

Open keithyh opened 4 years ago

keithyh commented 4 years ago

可以换成resnet，vgg等分类网络嘛？怎么换哦

我试过会报错

ValueError: Dimension must be 4 but is 3 for 'ctc/transpose_2' (op: 'Transpose') with input shapes:[?,1,?,5504], [3].

fire717 commented 4 years ago

当然可以换，但是效果不一定好。你要注意backnone的输出保持一致就行。