Gmgge / TrOCR-Seal-Recognition

基于transformer的ocr识别,在公章(印章识别, seal recognition)拓展应用
121 stars 24 forks source link

运行报错 #14

Closed yqj0919 closed 7 months ago

yqj0919 commented 8 months ago

Traceback (most recent call last): File "onnx_test.py", line 135, in
res = model.run(img)
File "onnx_test.py", line 103, in run
decoder_output = self.decoder(input_ids=input_ids,
File "onnx_test.py", line 75, in call
onnx_output = self.model.run(['logits'], onnx_inputs)
File "D:\anaconda\envs\fastapi\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 200, in run
return self._sess.run(output_names, input_feed, run_options)
onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Unexpected input data type. Actual: (tensor(int32)) , expected: (tensor(int64))

Gmgge commented 8 months ago

经排查在windows平台可能会出现数据类型不匹配情况,已经修复,请拉取最新的代码查看