Vahe commited on
Commit
3ca16b0
·
1 Parent(s): bcd524f

input details index get back

Browse files
Files changed (1) hide show
  1. extractor.py +1 -1
extractor.py CHANGED
@@ -28,7 +28,7 @@ def get_card_xy(model_path, image_path):
28
  image_np = image_np[np.newaxis, :]
29
 
30
  # inference
31
- interpreter.set_tensor(int(input_details[0]['index']+1), image_np)
32
  interpreter.invoke()
33
 
34
  # Obtaining output results
 
28
  image_np = image_np[np.newaxis, :]
29
 
30
  # inference
31
+ interpreter.set_tensor(input_details[0]['index'], image_np)
32
  interpreter.invoke()
33
 
34
  # Obtaining output results