写一段python代码运用mask处理技术用opencv 读入给定的 乒乓mp4文件地址为CUsersJionieDesktopdazuoppmp4 编写代码分析每帧视频从视频画面中提取出乒乓球的遮罩然后再提取出乒乓球要求提取出的画面只有乒乓球
的运动轨迹,背景应该被过滤掉。
import cv2
# 读入视频
cap = cv2.VideoCapture('C:/Users/Jionie/Desktop/dazuo/pp.mp4')
# 定义背景减除器
fgbg = cv2.createBackgroundSubtractorMOG2()
# 循环处理每一帧
while True:
# 读取一帧
ret, frame = cap.read()
# 如果读取失败,退出循环
if not ret:
break
# 对当前帧进行背景减除
fgmask = fgbg.apply(frame)
# 对背景减除后的图像进行二值化处理
ret, thresh = cv2.threshold(fgmask, 127, 255, cv2.THRESH_BINARY)
# 查找轮廓
contours, hierarchy = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# 遍历所有轮廓
for cnt in contours:
# 计算轮廓面积
area = cv2.contourArea(cnt)
# 如果面积小于一定值,认为是噪点,忽略掉
if area < 100:
continue
# 计算轮廓的外接矩形
x, y, w, h = cv2.boundingRect(cnt)
# 在原图上绘制矩形框
cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
# 提取乒乓球
ball = frame[y:y+h, x:x+w]
# 显示提取出的乒乓球
cv2.imshow('ball', ball)
# 显示原图和处理后的图像
cv2.imshow('frame', frame)
cv2.imshow('fgmask', fgmask)
# 按下q键退出循环
if cv2.waitKey(30) & 0xff == ord('q'):
break
# 释放资源
cap.release()
cv2.destroyAllWindows()
``
原文地址: http://www.cveoy.top/t/topic/dQfB 著作权归作者所有。请勿转载和采集!