Update app.py

app.py

@@ -77,7 +77,7 @@ def save_image_and_show(image, folder_name="output_images"):
     st.success(f"Image saved to {image_path} (local path)")
 
     # Show the image to the user (not the path)
-    st.image(image, caption="Saved Image", channels="RGB")
+    st.image(image, caption="Saved Image", channels="BGR")
 
 
 
@@ -115,7 +115,7 @@ def rotation(img,n=5):
     for i in range(n):
     #center = (width//2, height//2)
         angle=np.random.choice([90, 180, -90])
-        scale=np.round(np.random.uniform(1,2),2)
+        scale=np.round(np.random.uniform(1,1.6),2)
         r_x=img.shape[0]//2
         r_y=img.shape[1]//2
         rm=cv2.getRotationMatrix2D((r_y,r_x),angle,scale)
@@ -124,15 +124,17 @@ def rotation(img,n=5):
         save_image_and_show(r_img)
         #st.image(img_rot_rgb, caption="Rotation Applied", use_container_width=True)
 
-def scaling(img):
-    sx=2
-    sy=2
-    tx=0
-    ty=0
-    sm=np.array([[sx,0,tx],[0,sy,ty]],dtype=np.float32)
-    s_img=cv2.warpAffine(img,sm,(img.shape[1],img.shape[0]))
-    img_scal_rgb=cv2.cvtColor(s_img,cv2.COLOR_BGR2RGB)
-    st.image(img_scal_rgb, caption="Scaling Applied", use_container_width=True)
+def scaling(img,n=5):
+    for i in range(n):
+        sx=np.round(np.random.uniform(1,1.6),2)
+        sy=np.round(np.random.uniform(1,1.6),2)
+        tx=np.random.randint(2,60)
+        ty=np.random.randint(1,50)
+        sm=np.array([[sx,0,tx],[0,sy,ty]],dtype=np.float32)
+        s_img=cv2.warpAffine(img,sm,(img.shape[1],img.shape[0]))
+        #img_scal_rgb=cv2.cvtColor(s_img,cv2.COLOR_BGR2RGB)
+        save_image_and_show(s_img)
+        #st.image(img_scal_rgb, caption="Scaling Applied", use_container_width=True)
 
 def shearing(img):
     sx=1