在深度學習中,經常需要將一個完整的圖片分成多個patch進行處理,因為一個單獨的圖片分辨率可能太高。
python 隨機裁剪圖片得到多個patch
import cv2
import numpy as np
import os
from matplotlib import pyplot as plt
##創建新文件夾,如果該文件已經存在,則報錯
# os.makedirs('./hhh')
img=cv2.imread('./1.jpg')
##因為opencv讀取的圖片是BGR格式
# img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
plt.imshow(cv2.cvtColor(img,cv2.COLOR_BGR2RGB))
plt.show()
basename=os.path.basename('./1.jpg')[-1]
##裁剪塊的大小
patchsize=96
##裁剪的數量
num_patches=10
h,w,_=img.shape
for j in range(num_patches):
rr=np.random.randint(0,h-patchsize)
cc=np.random.randint(0,w-patchsize)
patchs=img[rr:rr+patchsize,cc:cc+patchsize,:]
cv2.imwrite(os.path.join('./hhh/',basename+'{}.png'.format(j)),patchs)