Text-Recognition-on-Cross-Domain-Datasets

Text-Recognition-on-Cross-Domain-Datasets copied to clipboard

导出模型为onnx的时候遇到问题

2

我是使用的您的ResNet_CRNN_IAM模型进行训练手写汉字的数据集，但是将模型导出为onnx的时候遇到了问题，我看您在网络中定义了forward和inference（您的inferrence应该是笔误）两个方法，我给的输入dummy_input 是torch.randn(1, 3, 192, 2048)的时候，会报 x, rec_targets, rec_lengths = input_dict['images'], \ IndexError: too many indices for tensor of dimension 4，这应该跟您在forward中定义需要读入input_dict 的字典作为输入参数有关，但我将dummy_input 换成这个字典的参数，还是无法正常生成。请问我应该怎么修改呢

cigegege