批量随机裁剪图片 python pil

#coding=utf-8

import cv2
import random
import os
import numpy as np
from tqdm import tqdm

def solve():
	cur_src="E:/CCF_competition/remote_sensing_image/unet_buildings/unet_buildings/newsrc/jpg/"
	cur_label="E:/CCF_competition/remote_sensing_image/unet_buildings/unet_buildings/newlabel/"
	for i in range(1,2):
		count = 1
		cur_url="{}{}.jpg".format(cur_src,i)
		cur_url_label = "{}{}.png".format(cur_label,i)
		print(cur_url)
		now = cv2.imread(cur_url)
		now_label = cv2.imread(cur_url_label)
		w,h,_ = now.shape
		for k in range(w//256):
			for kk in range(h//256):
				s_img = now[k*256:(k+1)*256,kk*256:(kk+1)*256,:]
				l_img = now_label[k*256:(k+1)*256,kk*256:(kk+1)*256,:]
				cv2.imwrite("{}{}/{}_{}.jpg".format(cur_src,"cut",i,count),s_img)
				cv2.imwrite("{}{}/{}_{}.png".format(cur_label,"cut",i,count),l_img)
				count +=1
		while count<=2500:
			j = random.randint(0,5)
			if j<=3:
				x_min=random.randint(0,w-512)
				y_min=random.randint(0,h-512)
				s_img = now[x_min:x_min+512,y_min:y_min+512,:]
				l_img = now_label[x_min:x_min+512,y_min:y_min+512,:]
				cv2.imwrite("{}{}/{}_{}.jpg".format(cur_src,"cut",i,count),s_img)
				cv2.imwrite("{}{}/{}_{}.png".format(cur_label,"cut",i,count),l_img)	
			else:
				x_min=random.randint(0,w-1024)
				y_min=random.randint(0,h-1024)
				s_img = now[x_min:x_min+1024,y_min:y_min+1024,:]
				l_img = now_label[x_min:x_min+1024,y_min:y_min+1024,:]
				cv2.imwrite("{}{}/{}_{}.jpg".format(cur_src,"cut",i,count),s_img)
				cv2.imwrite("{}{}/{}_{}.png".format(cur_label,"cut",i,count),l_img)
			count += 1
if __name__ == '__main__':
	solve()