Spaces:
Sleeping
Sleeping
another test option
Browse files- 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}")
|