OpenCV模块
模块cv2: https://www.cnblogs.com/shizhengwen/p/8719062.html
opencv 的文档: https://docs.opencv.org/master/
常用方法:
一、读入图像
使用函数cv2.imread(filepath,flags)读入一副图片
- filepath:要读入图片的完整路径
- flags:读入图片的标志
- cv2.IMREAD_COLOR:默认参数,读入一副彩色图片,忽略alpha通道
- cv2.IMREAD_GRAYSCALE:读入灰度图片
- cv2.IMREAD_UNCHANGED:顾名思义,读入完整图片,包括alpha通道
import cv2 img = cv2.imread('1.jpg',cv2.IMREAD_GRAYSCALE)
二、显示图像
使用函数cv2.imshow(wname,img)显示图像,第一个参数是显示图像的窗口的名字,第二个参数是要显示的图像(imread读入的图像),窗口大小自动调整为图片大小
cv2.imshow('image',img) cv2.waitKey(0) cv2.destroyAllWindows() #dv2.destroyWindow(wname)
- cv2.waitKey顾名思义等待键盘输入,单位为毫秒,即等待指定的毫秒数看是否有键盘输入,若在等待时间内按下任意键则返回按键的ASCII码,程序继续运行。若没有按下任何键,超时后返回-1。参数为0表示无限等待。不调用waitKey的话,窗口会一闪而逝,看不到显示的图片。
- cv2.destroyAllWindow()销毁所有窗口
- cv2.destroyWindow(wname)销毁指定窗口
三、保存图像
使用函数cv2.imwrite(file,img,num)保存一个图像。第一个参数是要保存的文件名,第二个参数是要保存的图像。可选的第三个参数,它针对特定的格式:对于JPEG,其表示的是图像的质量,用0 - 100的整数表示,默认95;对于png ,第三个参数表示的是压缩级别。默认为3.
注意:
- cv2.IMWRITE_JPEG_QUALITY类型为 long ,必须转换成 int
- cv2.IMWRITE_PNG_COMPRESSION, 从0到9 压缩级别越高图像越小。
- img这个参数可以是点阵,可以参考下面那个例子。
cv2.imwrite('1.png',img, [int( cv2.IMWRITE_JPEG_QUALITY), 95]) cv2.imwrite('1.png',img, [int(cv2.IMWRITE_PNG_COMPRESSION), 9])
例子: (这个是一个人脸csv的data转换为照片的script)
将数据分离后,人脸像素数据全部存储在data.csv文件中,其中每行数据就是一张人脸。按行读取数据,利用opencv将每行的2304个数据恢复为一张48X48的人脸图片,并保存为jpg格式。在保存这些图片时,将第一行数据恢复出的人脸命名为0.jpg,第二行的人脸命名为1.jpg......,以方便与label[0]、label[1]......一一对应。
1 import cv2 2 import numpy as np 3 4 # 指定存放图片的路径 5 path = './/face' 6 # 读取像素数据 7 data = np.loadtxt('data.csv') 8 9 # 按行取数据 10 for i in range(data.shape[0]): 11 face_array = data[i, :].reshape((48, 48)) # reshape 12 cv2.imwrite(path + '//' + '{}.jpg'.format(i), face_array) # 写图片
四、图片操作
1、使用函数cv2.flip(img,flipcode)翻转图像,flipcode控制翻转效果。
- flipcode = 0:沿x轴翻转
- flipcode > 0:沿y轴翻转
- flipcode < 0:x,y轴同时翻转
imgflip = cv2.flip(img,1)
2、复制图像
imgcopy = img.copy()
3、颜色空间转换
#彩色图像转为灰度图像 img2 = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY) #灰度图像转为彩色图像 img3 = cv2.cvtColor(img,cv2.COLOR_GRAY2RGB) # cv2.COLOR_X2Y,其中X,Y = RGB, BGR, GRAY, HSV, YCrCb, XYZ, Lab, Luv, HLS
示例
读入一副图像,按’s’键保存后退出,其它任意键则直接退出不保存
import cv2 img = cv2.imread('1.jpg',cv2.IMREAD_UNCHANGED) cv2.imshow('image',img) k = cv2.waitKey(0) if k == ord('s'): # wait for 's' key to save and exit cv2.imwrite('1.png',img) cv2.destroyAllWindows() else: cv2.destroyAllWindows()
示例
读入一副图像,给图片加文本
import cv2 # img=cv2.imread('1.jpg',cv2.IMREAD_COLOR) img=cv2.imread('1.png',cv2.IMREAD_COLOR) # 打开文件 font = cv2.FONT_HERSHEY_DUPLEX # 设置字体 # 图片对象、文本、像素、字体、字体大小、颜色、字体粗细 imgzi = cv2.putText(img, "zhengwen", (1100, 1164), font, 5.5, (0, 0, 0), 2,) # cv2.imshow('lena',img) cv2.imwrite('5.png',img) # 写磁盘 cv2.destroyAllWindows() # 毁掉所有窗口 cv2.destroyWindow(wname) # 销毁指定窗口
配合画图
import numpy as np import cv2 np.set_printoptions(threshold='nan') # 创建一个宽512高512的黑色画布,RGB(0,0,0)即黑色 img=np.zeros((512,512,3),np.uint8) # 画直线,图片对象,起始坐标(x轴,y轴),结束坐标,颜色,宽度 cv2.line(img,(0,0),(311,511),(255,0,0),10) # 画矩形,图片对象,左上角坐标,右下角坐标,颜色,宽度 cv2.rectangle(img,(30,166),(130,266),(0,255,0),3) # 画圆形,图片对象,中心点坐标,半径大小,颜色,宽度 cv2.circle(img,(222,222),50,(255.111,111),-1) # 画椭圆形,图片对象,中心点坐标,长短轴,顺时针旋转度数,开始角度(右长轴表0度,上短轴表270度),颜色,宽度 cv2.ellipse(img,(333,333),(50,20),0,0,150,(255,222,222),-1) # 画多边形,指定各个点坐标,array必须是int32类型 pts=np.array([[10,5],[20,30],[70,20],[50,10]], np.int32) # -1表示该纬度靠后面的纬度自动计算出来,实际上是4 pts = pts.reshape((-1,1,2,)) # print(pts) # 画多条线,False表不闭合,True表示闭合,闭合即多边形 cv2.polylines(img,[pts],True,(255,255,0),5) #写字,字体选择 font=cv2.FONT_HERSHEY_SCRIPT_COMPLEX # 图片对象,要写的内容,左边距,字的底部到画布上端的距离,字体,大小,颜色,粗细 cv2.putText(img,"OpenCV",(10,400),font,3.5,(255,255,255),2) a=cv2.imwrite("picture.jpg",img) cv2.imshow("picture",img) cv2.waitKey(0) cv2.destroyAllWindows()