michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-06-21 00:50:30 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

openvino: legacy gpu crash fix for text recognition

Browse Source
This commit is contained in:
Koushik Dutta
2025-11-16 18:13:12 -08:00
parent c4a12fe493
commit b43fdf83e2
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
plugins/openvino/src/ov/text_recognition.py
View File

@@ -29,7 +29,10 @@ class OpenVINOTextRecognition(TextRecognition):
)
if "vgg" in model:
model = self.plugin.core.read_model(xmlFile)
model.reshape([1, 1, 64, 384])
# this reshape causes a crash on GPU but causes a crash if NOT used with NPU...
# on older systems skipping the reshape does not crash, but does throw na exception which is recoverable.
if "NPU" in self.plugin.mode:
model.reshape([1, 1, 64, 384])
return self.plugin.core.compile_model(model, self.plugin.mode)
else:
model = self.plugin.core.read_model(xmlFile)