标签: loss-function resnet transfer-learning cnn ctc
我想使用预先训练的CNN模型(ResNet50,VGG等)提取特征,并使用具有CTC损失功能的特征。
我想将其构建为文本识别模型。
任何人我该如何实现?
答案 0 :(得分:0)
我不确定您是要微调预训练的模型还是将模型用于特征提取。要进行后者冻结定型的模型权重(在PyTorch中有几种方法可以做到这一点,最简单的方法是在模型上调用.eval()),并将logit从模型的最后一层馈送到新的输出头。有关更深入的指导,请参阅PyTorch教程here。