如何从 HFS 模型输出中抓取轮廓?我正在尝试检测地板。任何帮助将不胜感激。
由于地板有特定的颜色范围,我们可以使用 cv2.inRange()
来设置颜色阈值。我们将图像转换为 HSV 格式,然后使用下限和上限阈值生成二进制分段掩码
lower = np.array([0, 31, 182])
upper = np.array([57, 75, 209])
要找到地板轮廓,我们可以只在蒙版图像上找到轮廓。这是地板以绿色突出显示的结果
nhập numpy dưới dạng np
nhập khẩu cv2
# Color threshold
hình ảnh = cv2.imread('1.jpg')
original = image.copy()
hsv = cv2.cvtColor(hình ảnh, cv2.COLOR_BGR2HSV)
lower = np.array([0, 31, 182])
upper = np.array([57, 75, 209])
mask = cv2.inRange(hsv, lower, upper)
result = cv2.bitwise_and(original,original,mask=mask)
# Find floor contour on mask
cnts = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnts = cnts[0] if len(cnts) == 2 else cnts[1]
for c in cnts:
cv2.drawContours(original,[c], -1, (36,255,12), 2)
cv2.imshow('mask', mask)
cv2.imshow('result', result)
cv2.imshow('original', original)
cv2.waitKey()
Tôi là một lập trình viên xuất sắc, rất giỏi!