reset state

This commit is contained in:
Josh Hawkins 2025-04-07 19:21:07 -05:00
parent 9def823576
commit 4505986ed5

View File

@ -94,6 +94,10 @@ class ONNXModelRunner:
if self.type == "ov": if self.type == "ov":
infer_request = self.interpreter.create_infer_request() infer_request = self.interpreter.create_infer_request()
# This ensures the model starts with a clean state for each sequence
# Important for RNN models like PaddleOCR recognition
infer_request.reset_state()
outputs = infer_request.infer(input) outputs = infer_request.infer(input)
return outputs return outputs