入门|图像增强技术

#需要用到os和cv2库（安装请安装opencv，导入请导入cv2） imagelist=os.listdir(path)#读取path目录下所有文件的名字 for i in imagelist: if(i.endswith('.jpg' or '.png')): #如果文件以jpg或者png结尾 image=cv2.imread(path+'/'+i)#读取图片

#裁剪(crop) Image=cv2.imread(‘某张图片地址’) #读取图片 sp=image.shape #读取图像的形状信息并通过数组输出[图像的高，图像的宽，图像通道数] h=sp[0] w=sp[1] #截取图像某个区域，可配合random库随机截取，也可以自己设置截取范围 y1=int((1/3)*h) y2=int((2/3)*h) x1=int((1/3)*w) x2=int((2/3)*w) img=image[y1:y2,x1:x2] #将图片左上角视为原点，截取y1到y2、x1到x2的区域 #旋转(roating) h,w=image.shape[:2] center=(h//2,w//2)#旋转中心 #图片旋转 #degree旋转角度，size旋转后的大小，填1为等比例旋转。 M = cv2.getRotationMatrix2D(center, degree, size) #执行旋转操作，roated为旋转后的图片 roated = cv2.warpAffine(image, M, (w, h)) #镜像翻转(flip) image=cv2.imread(path) image=cv2.flip(image,1)#图片翻转 #缩放 image=cv2.imread(path) image=cv2.resize(image,(192,192))#括号内填缩放后的大小

cv2.imwrite('保存地址',image)

data=[]#该数组用于存储图片 label=[]#用于储存图片类型，看一个实例 pathone='D:\\imgCut\\Archive\\resize\\nosmoke'#无烟集图片地址 pathtwo="D:\\imgCut\\Archive\\resize\\smoke"#有烟集图片地址 #0代表无烟图片，1代表有烟图片 def saveZ(pathone,pathtwo): data=[] label=[] imagelist_one=os.listdir(pathone)#读取目录下所有文件的名字 imagelist_two=os.listdir(pathtwo) for i in imagelist_one: if(i.endswith('.jpg' or '.png')): #如果文件以jpg结尾 image=cv2.imread(pathone+'\\'+i) data.append(image)#将图片加入数组,这里存入的是图片本身而不是一个引用。 label.append(0) for i in imagelist_two: if(i.endswith('.jpg' or '.png')): #如果文件以jpg结尾 image=cv2.imread(pathtwo+'\\'+i) data.append(image) label.append(1) np.savez('arc.npz', data=data, label=label) #文件以npz形式储存，里面有两个数组data,label

arc=np.load('arc.npz')#导入这个文件data=arc['data']#读取里面的data数组和label数组label=arc['label']