opencv沿斜线切割图片

opencv如果沿水平或者数值的线来切割图片，是很简单的。但是，如果沿着斜线，那么就不能直接进行切割了。

我的思路是，根据直线上两点来确定直线方程，然后，逐个遍历图像中元素，根据其坐标，判断其在直线下方还是上方。如果在下方，就把该点像素复制到另一个和原始图像大小相同的纯色图像中。在下方的话复制到另一个纯色图像中。

原理也很简单。

效果（图源百度，侵删）

原图：

切割后：

代码如下：

import cv2
import numpy as npdef line(h, w, h1, w1, h2, w2):k = (w2-w1)/(h2-h1)  # 直线斜率b = w1 - k*h1        # 偏移量return (h * k + b) <= w   # 返回bool值，如果该点在斜线下方，返回Falsedef cut(img, point_A, point_B):h1, w1 = point_A[0], point_A[1]h2, w2 = point_B[0], point_B[1]height, width, _ = img.shapeup = np.zeros((height, width, 3), dtype='uint8')down = np.zeros((height, width,3), dtype='uint8')   # 黑底, 一定要加uint8for i in range(height):for j in range(width):if line(i,j, h1, w1, h2, w2):up[i,j] = img[i,j]else:down[i,j] = img[i,j]return np.array(up), np.array(down)path = 'test.jpg'img = cv2.imread(path)# 点的坐标由用户来定，这是一个例子
point_A = (200, 200)
point_B = (600, 400)   up, down = cut(img, point_A, point_B)cv2.imwrite('up.jpg', up)
cv2.imwrite('down.jpg', down)
cv2.imshow('up', up)
cv2.imshow('down', down)
cv2.waitKey(0)
cv2.destroyAllWindows()