批量随机裁剪图片 python pil

最新推荐文章于 2022-09-21 14:47:49 发布

原创

最新推荐文章于 2022-09-21 14:47:49 发布 · 3.4k 阅读

6 ·

CC 4.0 BY-SA版权

#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=ran