在opencv OpenCV中打开相机摄像头,用分水岭算法实时实现图像的分割与提取

python 复制代码
import cv2
import numpy as np

# 定义回调函数
def callback(x):
    pass

# 打开摄像头
cap = cv2.VideoCapture(0)

# 创建窗口和控件
cv2.namedWindow('image')
cv2.createTrackbar('threshold', 'image', 0, 255, callback)

# 初始化参数
bgdModel = np.zeros((1, 65), np.float64)
fgdModel = np.zeros((1, 65), np.float64)
rect = (0, 0, 1, 1)

while True:
    # 获取当前帧
    ret, frame = cap.read()

    # 转换为灰度图像
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # 获取阈值
    threshold = cv2.getTrackbarPos('threshold', 'image')

    # 进行前景和背景的分割
    mask = np.zeros(gray.shape, np.uint8)
    mask[gray >= threshold] = 1
    cv2.imshow('mask', mask)

    # 对分割后的图像进行分水岭处理
    markers = cv2.watershed(frame, mask)

    # 绘制分割结果
    frame[markers == -1] = [255, 0, 0]

    # 显示结果
    cv2.imshow('image', frame)

    # 按下ESC键退出程序
    if cv2.waitKey(1) == 27:
        break

# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()

cv2.VideoCapture()函数打开摄像头,然后创建一个新窗口并添加一个用于调整阈值的滑动条。在每一帧图像中,我们通过cv2.cvtColor()将其转换为灰度图像,然后根据阈值获取前景和背景的分割结果。接下来,我们将分割结果传递给cv2.watershed()函数进行分水岭处理,然后将分水岭处理后的分割结果绘制到原图像中。最后,我们使用cv2.imshow()显示结果,并在用户按下ESC键时退出程序。

相关推荐
格林威4 分钟前
Linux使用-MySQL的使用
linux·运维·人工智能·数码相机·mysql·计算机视觉·视觉检测
Jelena技术达人10 分钟前
淘宝/天猫按图搜索(拍立淘)item_search_img API接口实战指南
算法·图搜索算法
Adorable老犀牛20 分钟前
阿里云-基于通义灵码实现高效 AI 编码 | 8 | 上手实操:LeetCode学习宝典,通义灵码赋能算法高效突破
学习·算法·leetcode
望获linux26 分钟前
【实时Linux实战系列】规避缺页中断:mlock/hugetlb 与页面预热
java·linux·服务器·数据库·chrome·算法
菜就多练,以前是以前,现在是现在27 分钟前
Codeforces Round 1048 (Div. 2)
数据结构·c++·算法
林木辛1 小时前
LeetCode 热题 160.相交链表(双指针)
算法·leetcode·链表
野生的编程萌新1 小时前
【C++深学日志】从0开始的C++生活
c语言·开发语言·c++·算法
ulias2122 小时前
单元最短路问题
数据库·c++·算法·动态规划
崎岖Qiu2 小时前
leetcode380:RandomizedSet - O(1)时间插入删除和获取随机元素(数组+哈希表的巧妙结合)
java·数据结构·算法·leetcode·力扣·散列表
ZLRRLZ2 小时前
【数据结构】图
数据结构·算法·图论