Update src/util.py

src/util.py

@@ -20,15 +20,6 @@ bucket = oss2.Bucket(oss2.Auth(access_key_id, access_key_secret), endpoint, buck
 oss_path = "hejunjie.hjj/TransferAnythingHF"
 oss_path_img_gallery = "hejunjie.hjj/TransferAnythingHF_img_gallery"
 
-
-def resize(image, short_side_length=768):
-    width, height = image.size
-    ratio = short_side_length / min(width, height)
-    new_width = int(width * ratio)
-    new_height = int(height * ratio)
-    resized_image = image.resize((new_width, new_height))
-    return resized_image
-
     
 def download_img_pil(index, img_url):
     # print(img_url)
@@ -60,13 +51,11 @@ def download_images(img_urls, batch_size):
 
 def upload_np_2_oss(input_image, name="cache.png", gallery=False):
     assert name.lower().endswith((".png", ".jpg")), name
-    if name.lower().endswith(".png"):
-        name = name[:-4] + ".jpg"
     imgByteArr = io.BytesIO()
     if name.lower().endswith(".png"):
-        resize(Image.fromarray(input_image)).save(imgByteArr, format="PNG")
+        Image.fromarray(input_image).save(imgByteArr, format="PNG")
     else:
-        resize(Image.fromarray(input_image)).save(imgByteArr, format="JPEG", quality=95)
+        Image.fromarray(input_image).save(imgByteArr, format="JPEG", quality=95)
     imgByteArr = imgByteArr.getvalue()
 
     if gallery: