python/line_detection.py · github-mysync/opencv_tutorial

代码拉取完成,页面将自动刷新

扫描微信二维码支付

取消

支付完成

richgiteeai

Watch

不关注关注所有动态仅关注版本发行动态关注但不提醒动态

1 Star 0 Fork 3

github-mysync/opencv_tutorial

forked from mirrors_gloomyfish1998/opencv_tutorial

代码 Wiki 统计流水线

服务

加入 Gitee

与超过 1400万开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)

免费加入

已有帐号? 立即登录

文件

master

分支 (1)

管理

master

该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。

项目仓库所选许可证以仓库主分支所使用许可证为准

克隆/下载

HTTPS SSH SVN SVN+SSH 下载ZIP

提示

下载代码请复制以下命令到终端执行

为确保你提交的代码身份被 Gitee 正确识别,请执行以下命令完成配置

git config --global user.name userName 
git config --global user.email userEmail

初次使用 SSH 协议进行代码克隆、推送等操作时,需按下述提示完成 SSH 配置

1 生成 RSA 密钥

2 获取 RSA 公钥内容,并配置到 SSH公钥中

在 Gitee 上使用 SVN,请访问使用指南

使用 HTTPS 协议时,命令行会出现如下账号密码验证步骤。基于安全考虑,Gitee 建议配置并使用私人令牌替代登录密码进行克隆、推送等操作

Username for 'https://gitee.com': userName

Password for 'https://userName@gitee.com': # 私人令牌

分支 1

标签 0

opencv_tutorial

python

line_detection.py

line_detection.py 2.31 KB

####################################################
# 作者:zhigang,
# 微博:https://weibo.com/u/3181256271
####################################################
import cv2 as cv
import numpy as np

class LineDetector:
 def __init__(self):
 self.lines = []

def find_lines(self, frame):
 h, w, ch = frame.shape
 # 二值化图像
 print("start to detect lines...\n")
 gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
 ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_OTSU)
 cv.imshow("binary image", binary)
 cv.imwrite("D:/binary.png", binary)

dist = cv.distanceTransform(binary, cv.DIST_L1, cv.DIST_MASK_PRECISE)
 cv.imshow("distance", dist/15)
 dist = dist / 15
 result = np.zeros((h, w), dtype=np.uint8)
 ypts = []
 for row in range(h):
 cx = 0
 cy = 0
 max_d = 0
 for col in range(w):
 d = dist[row][col]
 if d > max_d:
 max_d = d
 cx = col
 cy = row
 result[cy][cx] = 255
 ypts.append([cx, cy])

xpts = []
 for col in range(w):
 cx = 0
 cy = 0
 max_d = 0
 for row in range(h):
 d = dist[row][col]
 if d > max_d:
 max_d = d
 cx = col
 cy = row
 result[cy][cx] = 255
 xpts.append([cx, cy])

cv.imshow("lines", result)
 cv.imwrite("D:/skeleton.png", result)

frame = self.line_fitness(ypts, image=frame)
 frame = self.line_fitness(xpts, image=frame, color=(255, 0, 0))

cv.imshow("fit-lines", frame)
 cv.imwrite("D:/fitlines.png", frame)
 return self.lines

def line_fitness(self, pts, image, color=(0, 0, 255)):
 h, w, ch = image.shape
 [vx, vy, x, y] = cv.fitLine(np.array(pts), cv.DIST_L1, 0, 0.01, 0.01)
 y1 = int((-x * vy / vx) + y)
 y2 = int(((w - x) * vy / vx) + y)
 cv.line(image, (w - 1, y2), (0, y1), color, 2)
 return image

if __name__ == "__main__":
 src = cv.imread("D:/javaopencv/two_lines.jpg")
 ld = LineDetector()
 lines = ld.find_lines(src)
 cv.waitKey(0)
 cv.destroyAllWindows()

一键复制编辑原始数据按行查看历史

OpenCV学堂提交于 2018年04月15日 13:31 +08:00 . fit lines

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78

####################################################
# 作者:zhigang,
# 微博:https://weibo.com/u/3181256271
####################################################
import cv2 as cv
import numpy as np


class LineDetector:
 def __init__(self):
 self.lines = []

 def find_lines(self, frame):
 h, w, ch = frame.shape
 # 二值化图像
 print("start to detect lines...\n")
 gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
 ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_OTSU)
 cv.imshow("binary image", binary)
 cv.imwrite("D:/binary.png", binary)

 dist = cv.distanceTransform(binary, cv.DIST_L1, cv.DIST_MASK_PRECISE)
 cv.imshow("distance", dist/15)
 dist = dist / 15
 result = np.zeros((h, w), dtype=np.uint8)
 ypts = []
 for row in range(h):
 cx = 0
 cy = 0
 max_d = 0
 for col in range(w):
 d = dist[row][col]
 if d > max_d:
 max_d = d
 cx = col
 cy = row
 result[cy][cx] = 255
 ypts.append([cx, cy])

 xpts = []
 for col in range(w):
 cx = 0
 cy = 0
 max_d = 0
 for row in range(h):
 d = dist[row][col]
 if d > max_d:
 max_d = d
 cx = col
 cy = row
 result[cy][cx] = 255
 xpts.append([cx, cy])

 cv.imshow("lines", result)
 cv.imwrite("D:/skeleton.png", result)

 frame = self.line_fitness(ypts, image=frame)
 frame = self.line_fitness(xpts, image=frame, color=(255, 0, 0))

 cv.imshow("fit-lines", frame)
 cv.imwrite("D:/fitlines.png", frame)
 return self.lines

 def line_fitness(self, pts, image, color=(0, 0, 255)):
 h, w, ch = image.shape
 [vx, vy, x, y] = cv.fitLine(np.array(pts), cv.DIST_L1, 0, 0.01, 0.01)
 y1 = int((-x * vy / vx) + y)
 y2 = int(((w - x) * vy / vx) + y)
 cv.line(image, (w - 1, y2), (0, y1), color, 2)
 return image


if __name__ == "__main__":
 src = cv.imread("D:/javaopencv/two_lines.jpg")
 ld = LineDetector()
 lines = ld.find_lines(src)
 cv.waitKey(0)
 cv.destroyAllWindows()