up 10

preprocessing.py

@@ -66,20 +66,20 @@ def get_mask(image_mask: np.ndarray) -> np.ndarray:
     return mask
 
 
-def get_image() -> np.ndarray:
-
-    """Get the image from the session state.
-    Returns:
-        np.ndarray: image
-    """
-    if 'initial_image' in st.session_state and st.session_state['initial_image'] is not None:
-        initial_image = st.session_state['initial_image']
-        if isinstance(initial_image, Image.Image):
-            return np.array(initial_image.resize((WIDTH, HEIGHT)))
-        else:
-            return np.array(Image.fromarray(initial_image).resize((WIDTH, HEIGHT)))
-    else:
-        return None
+# def get_image() -> np.ndarray:
+#
+#     """Get the image from the session state.
+#     Returns:
+#         np.ndarray: image
+#     """
+#     if 'initial_image' in st.session_state and st.session_state['initial_image'] is not None:
+#         initial_image = st.session_state['initial_image']
+#         if isinstance(initial_image, Image.Image):
+#             return np.array(initial_image.resize((WIDTH, HEIGHT)))
+#         else:
+#             return np.array(Image.fromarray(initial_image).resize((WIDTH, HEIGHT)))
+#     else:
+#         return None
 
 
 # def make_enhance_config(segmentation, objects=None):