Upload 7 files

SessionState.py

@@ -0,0 +1,60 @@
+from streamlit.report_thread import get_report_ctx
+import streamlit as st
+
+class SessionState(object):
+    def __init__(self, **kwargs):
+        """A new SessionState object.
+
+        Parameters
+        ----------
+        **kwargs : any
+            Default values for the session state.
+
+        Example
+        -------
+        >>> session_state = SessionState(user_name='', favorite_color='black')
+        >>> session_state.user_name = 'Mary'
+        ''
+        >>> session_state.favorite_color
+        'black'
+
+        """
+        for key, val in kwargs.items():
+            setattr(self, key, val)
+
+
+@st.cache(allow_output_mutation=True)
+def get_session(id, **kwargs):
+    return SessionState(**kwargs)
+
+
+def get(**kwargs):
+    """Gets a SessionState object for the current session.
+
+    Creates a new object if necessary.
+
+    Parameters
+    ----------
+    **kwargs : any
+        Default values you want to add to the session state, if we're creating a
+        new one.
+
+    Example
+    -------
+    >>> session_state = get(user_name='', favorite_color='black')
+    >>> session_state.user_name
+    ''
+    >>> session_state.user_name = 'Mary'
+    >>> session_state.favorite_color
+    'black'
+
+    Since you set user_name above, next time your script runs this will be the
+    result:
+    >>> session_state = get(user_name='', favorite_color='black')
+    >>> session_state.user_name
+    'Mary'
+
+    """
+    ctx = get_report_ctx()
+    id = ctx.session_id
+    return get_session(id, **kwargs)

helper_fns/gan_utils.py

@@ -0,0 +1,30 @@
+import os
+from PIL import Image
+
+def make_square(image, min_size=512, fill_color=(255, 255, 255, 0)):
+    '''
+        Make a square image with signature in the center and black (transparent)
+        strips on top and bottom. Cycle GAN is trained with images of this format.    
+    '''
+    x, y = image.size
+    size = max(min_size, x, y)
+    new_im = Image.new('RGBA', (size, size), fill_color)
+    new_im.paste(image, (int((size - x) / 2), int((size - y) / 2)))
+    new_im = new_im.resize((512, 512))
+    return new_im
+
+def resize_images(path):
+    '''
+        Resize all the images present in path that matches the ips used in cyclegan
+        training
+    '''
+    dirs = os.listdir(path)
+    for item in dirs:
+        if os.path.isfile(path+item):
+            image = Image.open(path+item)
+            image = make_square(image)
+            image = image.convert('RGB')
+            image.save(path+item)
+
+# test_path = 'runs/detect/exp/crops/DLSignature/'
+# resize_images(test_path)