Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ DECODE_DICT = {i:char for i, char in enumerate(CHARS)}
|
|
| 8 |
|
| 9 |
interpreter = tf.lite.Interpreter(model_path='detection.tflite')
|
| 10 |
interpreter.allocate_tensors()
|
| 11 |
-
recog_interpreter = tf.lite.Interpreter(model_path='
|
| 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))
|
|
@@ -46,7 +46,7 @@ def execute_text_recognition_tflite( boxes, frame, interpreter, input_details, o
|
|
| 46 |
return text,cv2.resize(save_frame,(94,24))
|
| 47 |
|
| 48 |
def greet(image):
|
| 49 |
-
resized = cv2.resize(image, (
|
| 50 |
input_data = resized.astype(np.float32) # Set as 3D RGB float array
|
| 51 |
input_data /= 255. # Normalize
|
| 52 |
input_data = np.expand_dims(input_data, axis=0) # Batch dimension (wrap in 4D)
|
|
|
|
| 8 |
|
| 9 |
interpreter = tf.lite.Interpreter(model_path='detection.tflite')
|
| 10 |
interpreter.allocate_tensors()
|
| 11 |
+
recog_interpreter = tf.lite.Interpreter(model_path='recognition2.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))
|
|
|
|
| 46 |
return text,cv2.resize(save_frame,(94,24))
|
| 47 |
|
| 48 |
def greet(image):
|
| 49 |
+
resized = cv2.resize(image, (320,320), interpolation=cv2.INTER_CUBIC)
|
| 50 |
input_data = resized.astype(np.float32) # Set as 3D RGB float array
|
| 51 |
input_data /= 255. # Normalize
|
| 52 |
input_data = np.expand_dims(input_data, axis=0) # Batch dimension (wrap in 4D)
|