Spaces:
Sleeping
Sleeping
code extended for testing purpose
Browse files- extractor.py +2 -0
extractor.py
CHANGED
@@ -27,6 +27,8 @@ def get_card_xy(model_path, image_path):
|
|
27 |
image_np = np.true_divide(image_np, 255, dtype=np.float32)
|
28 |
image_np = image_np[np.newaxis, :]
|
29 |
|
|
|
|
|
30 |
# inference
|
31 |
interpreter.set_tensor(input_details[0]['index'], image_np)
|
32 |
interpreter.invoke()
|
|
|
27 |
image_np = np.true_divide(image_np, 255, dtype=np.float32)
|
28 |
image_np = image_np[np.newaxis, :]
|
29 |
|
30 |
+
st.write(f"{image_np.shape}")
|
31 |
+
|
32 |
# inference
|
33 |
interpreter.set_tensor(input_details[0]['index'], image_np)
|
34 |
interpreter.invoke()
|