如何比較兩個物件的兩點

我的目的是檢查這些立方體是否在一列中。無法正確檢測物件。不知道為什么數字會這樣放在那里。我的主要問題：

1. 如何比較兩個物件的兩點以檢查它們是否合適？
2. 如何改進檢測物件？

提前致謝。

import cv2

image = cv2.imread('test1.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
ret,thresh = cv2.threshold(gray,127,255,cv2.THRESH_BINARY)
contours, hierarchy = cv2.findContours(
    thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

cv2.imshow('Canny Edges', thresh)

for (i, c) in enumerate(contours):
    ((x, y), _) = cv2.minEnclosingCircle(c)
    cv2.putText(image, "#{}".format(i   1), (int(x)   0, int(y)   0),
        cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2, cv2.LINE_AA)
    cv2.drawContours(image, [c], -1, (0, 255, 0), 2)

cv2.imshow('Contours', image)
cv2.waitKey(0)

uj5u.com熱心網友回復：

由于盒子相互接觸，分水嶺演算法在這種情況下可以很好地提取盒子輪廓。那么矩形輪廓的x坐標位置可以用來表示它們是否對齊。

腳步：

1. 分水嶺演算法獲取標記。
2. 過濾掉框（矩形）
3. 使用所有框 x 坐標的標準差來查看其是否對齊。（可以相應地設定閾值）

代碼：

import cv2
import numpy as np


def watershed_algorithm(image):

    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY   cv2.THRESH_OTSU)

    # noise removal
    kernel = np.ones((5,5),np.uint8)
    opening = cv2.morphologyEx(thresh,cv2.MORPH_OPEN,kernel, iterations = 2)

    # sure background area
    sure_bg = cv2.dilate(opening,kernel,iterations=3)

    # Finding sure foreground area
    dist_transform = cv2.distanceTransform(opening,cv2.DIST_L2,5)
    ret, sure_fg = cv2.threshold(dist_transform,0.7*dist_transform.max(),255,0)

    # Finding unknown regio3
    sure_fg = np.uint8(sure_fg)
    unknown = cv2.subtract(sure_bg,sure_fg)

    # Marker labelling
    ret, markers = cv2.connectedComponents(sure_fg)

    # Add one to all labels so that sure background is not 0, but 1
    markers = markers 1

    # Now, mark the region of unknown with zero
    markers[unknown==255] = 0

    markers = cv2.watershed(image,markers)
    
    return markers

def is_rect_contour(contour):
    peri = cv2.arcLength(contour, True)
    approx = cv2.approxPolyDP(contour, 0.04 * peri, True)
    if len(approx) == 4:
        return True
    else:
        return False
    
def get_rectangle_contours(markers, shape):
    rectangle_contours = []
    
    for marker in np.unique(markers):
        if marker == 0 or marker==-1:
            continue
        mask = np.zeros(shape, dtype="uint8")
        mask[markers == marker] = 255
        cnts = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL,
            cv2.CHAIN_APPROX_SIMPLE)
        c = max(cnts[0], key=cv2.contourArea)
        if is_rect_contour(c):
            rectangle_contours.append(c)
    
    return rectangle_contours

def draw_and_display_contours(contours, image):
    for i,c in enumerate(contours):
        (x, y, w, h) = cv2.boundingRect(c)
        cv2.drawContours(image, [c], -1, (0, 255, 0), 2)
        cv2.putText(image, "#{}".format(i 1), (int(x) - 10, int(y)),
            cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 0, 255), 2)
    
    cv2.imshow('Contours', image)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    
def is_boxes_aligned(rectangle_contours):
    x_coord_list = []
    for c in rectangle_contours:
        (x, y, w, h) = cv2.boundingRect(c)
        x_coord_list.append(x)
    
    threshold_std_deviation_in_X_coordinates = 10
    if np.std(x_coord_list) < threshold_std_deviation_in_X_coordinates:
        print("Box are aligned in same column with a standard deviation of {}".format(np.std(x_coord_list)))
    else:
        print("Box are not aligned in same column")

    

image = cv2.imread("./cubes.jpg")
markers = watershed_algorithm(image)
rectangle_contours = get_rectangle_contours(markers, image.shape[:2])
draw_and_display_contours(rectangle_contours,image)
is_boxes_aligned(rectangle_contours)

輸出：

Box are aligned in same column with a standard deviation of 4.027681991198191

標籤：Python opencv 图像处理

上一篇：ArUco標記，為什么當我改變影像解析度時姿勢會改變？

下一篇：使用Node.js和Sharp從AWSS3調整影像大小