opencv-python基础操作汇总——1（读取、画线、平移，旋转缩放、翻转和裁剪等操作）

前言

最近中日合作活动中要求使用opencv-python，于是我就复习一下，做个小总结，😄。

1. 读取并显示

注意：opencv读取时，默认是BGR格式，且是numpy.ndarray类型

import cv2
import matplotlib.pyplot as plt

image = cv2.imread('./im0.png') # 默认读取BGR格式
print('图片的高:', image.shape[0])
print('图片的宽:', image.shape[1])
print('图片的通道数:', image.shape[2])
plt.imshow(image)
# plt.axis('off') # 不显示坐标
plt.show()

效果如下：

2. 转为RGB图片并保存

import cv2
import matplotlib.pyplot as plt

image = cv2.imread('./im0.png') # 默认读取BGR格式
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
plt.imshow(image)
# plt.axis('off') # 不显示坐标
plt.show()

cv2.imwrite('new_img.jpg')

效果如下：

3. 画线

3.1 直线

可以使用cv2.line
第一个参数是图片的各个维度的值， 第二个参数是起始坐标，第三个参数是终止坐标， 第四个参数是画线颜色的RBG的值， 第五个参数是线的宽度。

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = np.zeros((224, 224, 3), dtype=np.uint8)
g = (0, 255, 0)
cv2.line(image, (0, 0), (224, 224), g, 5)
plt.imshow(image)
plt.show()

效果如下：

3.2 矩形

使用cv2.rectangle，第一个参数是图片的各个维度的值， 第二个参数是矩形左上角坐标，第三个参数是矩形右下角坐标， 第四个参数是画线颜色的RBG的值， 第五个参数是线的宽度，若是-1，则会将矩形填充。

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = np.zeros((224, 224, 3), dtype=np.uint8)
g = (0, 255, 0)
# cv2.line(image, (0, 0), (224, 224), g, 5)
cv2.rectangle(image, (30, 30), (100, 100), g, 3)
cv2.rectangle(image, (60, 60), (200, 200), g, -1)
plt.imshow(image)
plt.show()

效果如下：

3.3 圆形

使用cv2.circle， 第一个参数是图片的各个维度的值， 第二个参数是圆心坐标，第三个参数是半径， 第四个参数是画线颜色的RBG的值， 第五个参数是线的宽度，若是-1，则会将矩形填充。

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = np.zeros((224, 224, 3), dtype=np.uint8)
g = (0, 255, 0)
cv2.circle(image, (112, 112), 112, g, 4)
cv2.circle(image, (112, 112), 60, g, -1)
plt.imshow(image)
plt.show()

效果如下：

4. 图像平移

使用的是cv2.warpAffine，第一个参数是图片的各个维度的值， 第二个参数是方向，第三个是图片大小

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('im0.png')

a = np.float32([
    [1, 0, 250], # 向右平移250个像素点
    [0, 1, 200] # 向下平移200个像素点
])
image = cv2.warpAffine(image, a, (image.shape[1], image.shape[0]))
plt.imshow(image)
plt.show()

5. 图像旋转与缩放

使用的是cv2.getRotationMatrix2D和cv2.warpAffine结合，cv2.getRotationMatrix2D第一个参数是旋转中心坐标，第二个参数是旋转角度，第三个是缩放程度。

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('im0.png')

h, w = image.shape[0], image.shape[1]
x, y = w/2, h/2
a = cv2.getRotationMatrix2D((x, y), 45, 1.5)
image = cv2.warpAffine(image, a, (w, h))
plt.imshow(image)
plt.show()

效果如下：

6. 图像调整大小（resize）

使用的是cv2.resize，第一个参数是图片，第二个参数是调整大小的宽和高，第三个参数是插值方法

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('im0.png')

n_w, n_h = 224, 224
image = cv2.resize(image, (n_w, n_h))
plt.imshow(image)
plt.show()

7. 翻转

使用的是cv2.flip，第一个参数是图片，第二个参数是翻转方式

7.1 水平翻转

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('im0.png')

image = cv2.flip(image, 1)
plt.imshow(image)
plt.show()

7.2 垂直翻转

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('im0.png')

image = cv2.flip(image, 0)
plt.imshow(image)
plt.show()

7.3 水平+垂直

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('im0.png')

image = cv2.flip(image, -1)
plt.imshow(image)
plt.show()

8. 裁剪

裁剪主要是对x和y轴上的取值做变化，如下：
在x轴上取0~500，y轴取100到最后

import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('im0.png')

image = image[0:500, 100:]
plt.imshow(image)
plt.show()