Update app.py

app.py

@@ -4,6 +4,7 @@ import pandas as pd
 import os
 import zipfile
 
+
 import cv2
 
 st.markdown("""
@@ -97,24 +98,28 @@ def save_image_and_show(image, folder_name="output_images"):
 
 
 
-def translation(img):
-    tx=30 #np.random.randint(2,60)
-    ty=45 #np.random.randint(2,55)
-    tm=np.array([[1,0,tx],[0,1,ty]],dtype=np.float32)
-    trans_img=cv2.warpAffine(img,tm,dsize=(img.shape[1],img.shape[0]),borderMode=cv2.BORDER_REFLECT)
-    img_trans_rgb = cv2.cvtColor(trans_img, cv2.COLOR_BGR2RGB)
-    save_image_and_show(img_trans_rgb)
+def translation(img,n):
+    for i in range(n):
+        tx=np.random.randint(2,60)
+        ty=np.random.randint(2,55)
+        tm=np.array([[1,0,tx],[0,1,ty]],dtype=np.float32)
+        trans_img=cv2.warpAffine(img,tm,dsize=(img.shape[1],img.shape[0]),borderMode=cv2.BORDER_REFLECT)
+        img_trans_rgb = cv2.cvtColor(trans_img, cv2.COLOR_BGR2RGB)
+        save_image_and_show(img_trans_rgb)
     #st.image(img_trans_rgb, caption="Translation Applied", use_container_width=True)
 
-def rotation(img):
+def rotation(img,n=5):
+    for i in range(n):
     #center = (width//2, height//2)
-    angle=90
+    angle=np.random.choice([90, 180, -90])
+    scale=np.round(np.random.uniform(1,2),2)
     r_x=img.shape[0]//2
     r_y=img.shape[1]//2
-    rm=cv2.getRotationMatrix2D((r_y,r_x),angle,1)
+    rm=cv2.getRotationMatrix2D((r_y,r_x),angle,scale)
     r_img=cv2.warpAffine(img,rm,(img.shape[1],img.shape[0]))
-    img_rot_rgb = cv2.cvtColor(r_img, cv2.COLOR_BGR2RGB)
-    st.image(img_rot_rgb, caption="Rotation Applied", use_container_width=True)
+    #img_rot_rgb = cv2.cvtColor(r_img, cv2.COLOR_BGR2RGB)
+    save_image_and_show(r_img)
+    #st.image(img_rot_rgb, caption="Rotation Applied", use_container_width=True)
 
 def scaling(img):
     sx=2
@@ -162,6 +167,7 @@ if uploaded_file is not None:
     #  # Convert BGR to RGB as streamlit expects RGB colorspace
     # img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     transformation = []
+    
     if trans and btn:
         translation(img)
         transformation.append("Translation")