Vahe commited on
Commit
d737333
1 Parent(s): 2357905

another test option

Browse files
Files changed (1) hide show
  1. extractor.py +1 -0
extractor.py CHANGED
@@ -25,6 +25,7 @@ def get_card_xy(model_path, image_path):
25
 
26
  image_np = np.array(image_resized) #
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}")
 
25
 
26
  image_np = np.array(image_resized) #
27
  image_np = np.true_divide(image_np, 255, dtype=np.float32)
28
+ image_np = np.moveaxis(image_np, -1, 0)
29
  image_np = image_np[np.newaxis, :]
30
 
31
  st.write(f"{image_np.shape}")