CNN与CTC损失

Question

我想使用预先训练的CNN模型(ResNet50，VGG等)提取特征，并使用CTC损失函数来使用这些特征。

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

有谁知道我该如何做到这一点吗？

Answer 1

我不确定您是希望优化预先训练的模型还是使用这些模型进行特征提取。要执行后一种操作，请冻结石化模型权重(在PyTorch中有几种方法可以做到这一点，最简单的方法是在模型上调用.eval() )，并将逻辑从模型的最后一层提供给新的输出头。有关更深入的指南，请参阅PyTorch tutorial here。