Update utils.py
Browse files
utils.py
CHANGED
|
@@ -6,7 +6,8 @@ core = ov.Core()
|
|
| 6 |
model = core.read_model(model='models/horizontal-text-detection-0001.xml')
|
| 7 |
compiled_model = core.compile_model(model = model, device_name="CPU")
|
| 8 |
input_layer = compiled_model.input(0)
|
| 9 |
-
output_layer = compiled_model.output(
|
|
|
|
| 10 |
|
| 11 |
def preprocess(image, input_layer):
|
| 12 |
N, C, H, W = input_layer.shape
|
|
|
|
| 6 |
model = core.read_model(model='models/horizontal-text-detection-0001.xml')
|
| 7 |
compiled_model = core.compile_model(model = model, device_name="CPU")
|
| 8 |
input_layer = compiled_model.input(0)
|
| 9 |
+
output_layer = compiled_model.output(0)
|
| 10 |
+
#output_layer = compiled_model.output("boxes")
|
| 11 |
|
| 12 |
def preprocess(image, input_layer):
|
| 13 |
N, C, H, W = input_layer.shape
|