Update app.py

app.py

@@ -103,8 +103,8 @@ def zip_file(folder_name="output_images"):
 
 def translation(img,n):
     for i in range(n):
-        tx=np.random.randint(2,60)
-        ty=np.random.randint(2,55)
+        tx=np.random.randint(2,50)
+        ty=np.random.randint(1,45)
         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)
@@ -128,30 +128,38 @@ 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)
+        tx=np.random.randint(2,40)
+        ty=np.random.randint(1,30)
         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
-    shx=0.1
-    tx=0
-    shy=0.1
-    sy=1
-    ty=0
-    shm=np.array([[sx,shx,tx],[shy,sy,ty]],dtype=np.float32)
-    img_all=cv2.warpAffine(img,shm,(img.shape[1],img.shape[0]))
-    img_shear_rgb=cv2.cvtColor(img_all,cv2.COLOR_BGR2RGB)
-    st.image(img_shear_rgb, caption="Shearing Applied", use_container_width=True)
-
-def cropping(img):
-    img_crop=img[20:120,50:155]
-    img_crop_rgb=cv2.cvtColor(img_crop,cv2.COLOR_BGR2RGB)
-    st.image(img_crop_rgb, caption="Shearing Applied", use_container_width=True)
+def shearing(img,n=5):
+    for i in range(n):
+        sx=np.round(np.random.uniform(1,1.5),2)
+        shx=np.round(np.random.uniform(0,0.5),2)
+        tx=np.random.randint(2,40)
+        shy=np.round(np.random.uniform(0,0.5),2)
+        sy=np.round(np.random.uniform(0,0.5),2)
+        ty=np.random.randint(1,30)
+        shm=np.array([[sx,shx,tx],[shy,sy,ty]],dtype=np.float32)
+        img_all=cv2.warpAffine(img,shm,(img.shape[1],img.shape[0]))
+        save_image_and_show(imag_all)
+        #img_shear_rgb=cv2.cvtColor(img_all,cv2.COLOR_BGR2RGB)
+        #st.image(img_shear_rgb, caption="Shearing Applied", use_container_width=True)
+
+def cropping(img,n=5):
+    for i in range(n):
+        x_1=np.random.randint(20,60)
+        y_1=np.random.randint(110,160)
+        x_2=np.random.randint(20,60)
+        y_2=np.random.randint(110,160)
+        img_crop=img[x_1:y_1,x_2:y_2]
+        save_image_and_show(img_crop)
+        # img_crop_rgb=cv2.cvtColor(img_crop,cv2.COLOR_BGR2RGB)
+        # st.image(img_crop_rgb, caption="Shearing Applied", use_container_width=True)
 
 # def save_image(image, folder_name="output_images"):
 #     if not os.path.exists(folder_name):