leakyrelu commited on
Commit
70ff432
·
1 Parent(s): 58a527c

try new recog model

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -9,11 +9,12 @@ DECODE_DICT = {i:char for i, char in enumerate(CHARS)}
9
  interpreter = tf.lite.Interpreter(model_path='detection.tflite')
10
  interpreter.allocate_tensors()
11
  recog_interpreter = tf.lite.Interpreter(model_path='recognition.tflite')
 
 
 
12
  recog_interpreter.allocate_tensors()
13
  input_details = interpreter.get_input_details()
14
  output_details = interpreter.get_output_details()
15
- recog_input_details = recog_interpreter.get_input_details()
16
- recog_output_details = recog_interpreter.get_output_details()
17
 
18
 
19
 
 
9
  interpreter = tf.lite.Interpreter(model_path='detection.tflite')
10
  interpreter.allocate_tensors()
11
  recog_interpreter = tf.lite.Interpreter(model_path='recognition.tflite')
12
+ recog_input_details = recog_interpreter.get_input_details()
13
+ recog_output_details = recog_interpreter.get_output_details()
14
+ recog_interpreter.resize_tensor_input(recog_input_details[0]['index'], (1, 24, 94, 3))
15
  recog_interpreter.allocate_tensors()
16
  input_details = interpreter.get_input_details()
17
  output_details = interpreter.get_output_details()
 
 
18
 
19
 
20