OpenCV模块

Posted 2021-03-25 ian-learning

模块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()