Update app.py

app.py

@@ -107,8 +107,8 @@ def translation(img,n=5):
         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)
-        save_image_and_show(img_trans_rgb)
+        #img_trans_rgb = cv2.cvtColor(trans_img, cv2.COLOR_BGR2RGB)
+        save_image_and_show(trans_img)
     #st.image(img_trans_rgb, caption="Translation Applied", use_container_width=True)
 
 def rotation(img,n=5):
@@ -128,8 +128,8 @@ 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,40)
-        ty=np.random.randint(1,30)
+        tx=np.random.randint(1,16)
+        ty=np.random.randint(1,12)
         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)
@@ -140,10 +140,10 @@ 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)
+        tx=np.random.randint(1,15)
         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)
+        ty=np.random.randint(1,20)
         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(img_all)