CNN with CTC loss

Question

我想使用预训练的CNN模型(ResNet50、VGG等(提取特征，并使用具有CTC损失函数的特征。

我想把它构建成一个文本识别模型。

有人问我该如何做到这一点吗？

Answer 1

我不确定您是想微调预训练的模型，还是想使用这些模型进行特征提取。为了实现后者，冻结petrained模型权重(PyTorch中有几种方法可以做到这一点，最简单的方法是在模型上调用.eval(((，并将模型最后一层的logits提供给新的输出头。请参阅此处的PyTorch教程，以获得更深入的指南。