brainai-spaces commited on
Commit
f1a2fac
·
verified ·
1 Parent(s): 76bdd26

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
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("boxes")
 
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