Spaces:
Runtime error
Runtime error
File size: 2,368 Bytes
4473449 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
import glob
import os
from PIL import Image
import numpy as np
import time
import PIL
import matplotlib.pyplot as plt
import argparse
patchsize = 296
stride = 236
pad=0
PIL.Image.MAX_IMAGE_PIXELS = 933120000
def CropImage(image_path,output_dir,pad):
image_name = os.path.split(image_path)[1].split('.')[0]
im = Image.open(image_path)
size = im.size
# new_size = (size[0]+40,size[1]+40)
new_size = size
if(pad):
new_im = Image.new("RGB", new_size) ## luckily, this is already black!
new_im.paste(im, ((new_size[0] - size[0]) // 2,
(new_size[1] - size[1]) // 2))
#plt.imshow(new_im)
#plt.show()
else:
new_im = im
width, height = new_im.size
x = 0
y = 0
right = 0
bottom = 0
while (bottom < height):
while (right < width):
left = x
top = y
right = left + patchsize
bottom = top + patchsize
if (right > width):
offset = right - width
right -= offset
left -= offset
if (bottom > height):
offset = bottom - height
bottom -= offset
top -= offset
im_crop = new_im.crop((left, top, right, bottom))
im_crop_name = image_name + "_" + str(left) + "_" + str(top) + ".png"
output_path = os.path.join(output_dir, im_crop_name)
im_crop.save(output_path)
x += stride
x = 0
right = 0
y += stride
start_time = time.time()
parser = argparse.ArgumentParser()
parser.add_argument("--image_path", help="path to image to crop",
default=r"F:\Datasets\DigestPath\safron\Benign\test\3\ablation_study_neg_46\neg_46.png")
parser.add_argument("--output_dir", help="path to output folder",
default=r"F:\Datasets\DigestPath\safron\Benign\test\3\ablation_study_neg_46\cropped_safron_patchadv")
parser.add_argument("--pad", type=int, default=0, help="pad the image borders")
args = parser.parse_args()
if not os.path.exists(args.output_dir):
os.makedirs(args.output_dir)
CropImage(args.image_path,args.output_dir,args.pad)
print("--- %s seconds ---" % (time.time() - start_time)) |