opencv

https://segmentfault.com/a/1190000044071469

cv2.imread(path, flag)

• flag：

[return]    type: numpy.ndarray
            size: (H, W, C), C -> (BGR)

ImageToTensor

image = cv2.imread(path)
image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

image = torch.from_numpy(image_rgb).transpose(1, 2).transpose(0, 1)
image = image.float().div(255)

cv2.imwrite(path, image)

TensorToImage

image = image.mul(255).byte()
image = image.transpose(0, 1).transpose(1, 2)
image = image.cpu().numpy()

通道处理

b, g, r = cv2.split(image)
image = cv2.merge((b, g, r))

图像处理与计算机视觉基础

二值化

• cv2.threshold()

边缘检测

• cv2.Canny()

图像滤波

• cv2.filter2D()
• cv2.blur()
• cv2.GaussianBlur()

图像形态学操作

• cv2.erode() 腐蚀
• cv2.dilate() 膨胀
• cv2.morphologyEx() 开闭运算、形态学梯度、顶帽、黑帽等