用python将照片做成数据集_图片数据集预处理，python

从公益诉讼进口Imageimport randomimport osimport cv2import shutil # opencv图像旋转和拦截def rotate_crop(=没有形象,角,中心,r = 20): # 1 #旋转,输入图像和旋转角度,r拦截半径,中心中心,规模放大(h, w) =(2): # 2如果没有一个中心:# 3 x = w \/ 2 y = h \/ 2中心= (x, y) # 4中心M =(中心、角、规模)# 5旋转=(形象,M, (w h)) # 6 rotationa = int (xr) # x startb = int x (x + r) # endc = int y(年)# startd = int (y + r) # y endcropImg =旋转(a, b, c, d)返回cropImg # 7 #×n度旋转并提取中心小图像def image_rotate_crop (image_dir save_dir n): #输入旋转角度,图像文件夹地址,并保存文件夹地址f (image_dir): image_path = (image_dir f)如果(image_path):角= 0 im = (image_path),而角< 360:re_str = str(角)+ ' jpg 'name = (jpg, re_str) save_path = (save_dir、名称)save_image = rotate_crop (im,角度)(save_path save_image)角+ = n #遍历指定目录,在目录中显示所有文件名def CropImage4File (filepath destpath, r): pathDir = (filepath) #列表中的所有路径或文件filepathfor allDir pathDir:孩子= (filepath, allDir) dest = (destpath allDir)如果(孩子):打印(孩子)图像=(孩子)sp = th e图像的形状sz1 = sp [0] # imagesz2 = sp的高度(行)[1]#宽度(科勒姆)# sz3 = sp[2] #图像像素值是由三个主要的颜色,在这里我们不使用#操作你想文件= int (sz1\/2-r) # x startb = int (sz1\/2 + r) # x endc = int (sz2\/2-r) # int y startd = (sz2\/2 + r) # y endcropImg =图像(a, b, c, d) #作物图像(桌子,cropImg) # #阅读所有的文件写目的地路径,迭代def read_file_all (data_dir_path save_dir):为f (data_dir_path): data_file_path = (data_dir_path f)如果(data_file_path): image_rotate (data_file_path save_dir) #打印(收集)其他:read_file_all (data_file_path) def image_rotate (image_path save_dir): #读取图像,旋转扩展数据集# im = (image_path,0) im = (image_path) #指定逆时针旋转的角度save_path = save_dir + random_name () +\u201Cjpg\u201D(save_path) (save_path, im) # im_rotate1 = (90) # save_path = save_dir + random_name () +\u201Cjpg\u201D# im_ (save_path) (save_path im_rotate1) # im_rotate2 = (180) # save_path = save_dir + random_name () +\u201Cjpg\u201D# im_ (save_path) # im_rotate3 = (270) # save_path = save_dir + random_name () +\u201Cjpg\u201D# im_ (save_path) def random_nam e ():#随机数a_list = [' 0 ', ' 1 ', ' 2 ', ' 3 ', ' 4 ', ' 5 ', ' 6 ', ' 7 ', ' 8 ', ' 9 ', ' ', ' b ', ' c ', ' d ', ' e '] name = (a_list 8) file_name = \# dst_dir = ' E: \/ pcb_image_data \/ 2018 _11_7_h \/ train_20 \/真实\/ # read_file_all (scr_dir dst_dir) #旋转和捕捉小图片# image_dir = ' E: \/ pcb_defect_image \/ 2018 _11_26 \/火车\/真正的# save_dir = ' E: \/ pcb_defect_image \/ 2018 _11_26 \/ image_13 \/火车\/真正的# image_rotate_crop (image_dir save_dir 30) #分类图像根据文件名# scr_dir = ' E: \/ pcb_image_data \/ 2018 _11_8_h_hu \/ \/真实的# dst_dir = ' E: \/ pcb_image_data \/ 2018 _11_8_h_hu \/ \/ true_class\u201D# image_class (scr_dir dst_dir)