Update app.py

app.py

@@ -64,18 +64,29 @@ def translation(img):
     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_rgb = cv2.cvtColor(trans_img, cv2.COLOR_BGR2RGB)
-    st.image(img_rgb, caption="Translation Applied", use_container_width=True)
+    img_trans_rgb = cv2.cvtColor(trans_img, cv2.COLOR_BGR2RGB)
+    st.image(img_trans_rgb, caption="Translation Applied", use_container_width=True)
 
 def rotation(img):
     #center = (width//2, height//2)
     angle=90
     r_x=img.shape[0]//2
     r_y=img.shape[1]//2
-    rm=cv2.getRotationMatrix2D((r_x,r_y),angle,1)
+    rm=cv2.getRotationMatrix2D((r_y,r_x),angle,1)
     r_img=cv2.warpAffine(img,rm,(img.shape[1],img.shape[0]))
-    img_rgb = cv2.cvtColor(r_img, cv2.COLOR_BGR2RGB)
-    st.image(img_rgb, caption="Rotation Applied", use_container_width=True)
+    img_rot_rgb = cv2.cvtColor(r_img, cv2.COLOR_BGR2RGB)
+    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)
+    
 
 
 if uploaded_file is not None:
@@ -93,6 +104,7 @@ if uploaded_file is not None:
         rotation(img)
         transformation.append("Rotation")
     if scal:
+        scaling(img)
         transformation.append("Scaling")
     if shearing:
         transformation.append("Shearing")