12
社区成员
发帖
与我相关
我的任务
分享图像识别,图像分类,目前最有效的就是AI了, 不过要准备些样本,做个训练
我觉得可以用霍夫变换来识别图像中的线(我不确定这个办法是否可行,不过可以一试)。
1.将图像转换为灰度图像。
2.使用边缘检测算法检测图像中的边缘。
3.使用霍夫变换将边缘点映射到霍夫空间。
4.在霍夫空间中寻找具有足够多投票的点。
import cv2
# 将图像转换为灰度图像
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# 使用边缘检测算法检测图像中的边缘
edges = cv2.Canny(gray_image, 100, 200)
# 使用霍夫变换将边缘点映射到霍夫空间
lines = cv2.HoughLinesP(edges, 1, np.pi / 180, 100, minLineLength=100, maxLineGap=10)
# 在霍夫空间中寻找具有足够多投票的点
for line in lines:
x1, y1, x2, y2 = line[0]
cv2.line(image, (x1, y1), (x2, y2), (0, 0, 255), 2)
# 显示图像
cv2.imshow('Image', image)
cv2.waitKey(0)
尝试下AI