使用OpenCV2批量裁剪圖片,並將裁剪後的圖片儲存至指定資料夾
阿新 • • 發佈:2019-01-09
在做影象復原的相關工作中,由於資料集的缺失,我們通常需要將圖片裁剪成比較小的crop。比如在做深度學習影象去噪的相關研究的時候,由於影象去噪不太關注整體的影象內容,所以可以先將大的圖片裁剪成小的crop,以次來增加資料量。
具體程式碼如下:
import glob import cv2 '''cut the img_1024*1024 into img_46*46,the sampling interval is 30''' src_dir="C:\\Users\\ASUS\\Desktop\\src" #the scr_img 1024*1024 filepath=glob.glob(src_dir+'./*jpg') #return a list count=0 for i in range(len(filepath)): img_src= cv2.imread(filepath[i]) for x in range (0,(1024-46),30): for y in range (0,(1024-46),30): img_cut = img_src[x:x + 46, y:y + 46] save_dir = "C:\\Users\\ASUS\\Desktop\\crop_46_46\\crop{}.jpg".format(count) count += 1 cv2.imwrite(save_dir, img_cut)