简单粗暴,不多说,直接代码吧:
import os
import random
from PIL import Image
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
import cv2
import numpy as np
def random_crop(imagepath):
# image = Image.open(imagepath)
# len_rand = random.randint(0, min(image.size[0], image.size[1])-1)
# x_rand = random.randint(0, image.size[0]-len_rand-1)
# y_rand = random.randint(0, image.size[1]-len_rand-1)
image = cv2.imread(imagepath)
len_rand = random.randint(0, min(image.shape[1], image.shape[0])-1)
x_rand = random.randint(0, image.shape[1]-len_rand-1)
y_rand = random.randint(0, image.shape[0]-len_rand-1)
box = (x_rand, y_rand, x_rand+len_rand, y_rand+len_rand)
# return image.crop(box)
return image[box[1]:box[3], box[0]:box[2], :]
save_dir = 'data_prepare/net_negative'
if os.path.exists(save_dir)==False:
os.makedirs(save_dir)
neg_img_dir = 'imagenet_selected'
for file in os.walk(neg_img_dir):
filenames = file[2]
for idx, filename in enumerate(filenames):
filepath = neg_img_dir + '/' + filename
for i in range(33):
image_crop = random_crop(filepath)
savepath = save_dir + '/' + str(i) + filename
# image_crop_12 = image_crop.resize((15,15))
if image_crop is None:
continue
if image_crop.shape[0]<10 and image_crop.shape[1]<10:
continue
image_crop = cv2.resize(image_crop, (12, 12))
# image_crop_12.save(savepath, 'jpeg')
cv2.imwrite(savepath, np.array(image_crop))
print (idx)