opencv沿斜线切割图片

opencv沿斜线切割图片

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

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

原理也很简单。

效果（图源百度，侵删）

原图：

切割后：

代码如下：

import cv2
import numpy as np

def line(h, w, h1, w1, h2, w2):
    k = (w2-w1)/(h2-h1)  # 直线斜率
    b = w1 - k*h1        # 偏移量
    return (h * k + b) <= w   # 返回bool值，如果该点在斜线下方，返回False
    
def 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.shape
    
    up = np.zeros((height, width, 3), dtype='uint8')
    down = np.zeros((height, width,3), dtype='uint8')   # 黑底, 一定要加uint8
    
    for 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()