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

code extended for testing purpose

Browse files
Files changed (1) hide show
  1. 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()