当前位置: 移动技术网 > IT编程>脚本编程>Python > python实现超简单的视频对象提取功能

python实现超简单的视频对象提取功能

2018年08月20日  | 移动技术网IT编程  | 我要评论

赵正平视讯,男人雄起,永修租房

视频对象提取

与其说是视频对象提取,不如说是视频颜色提取,因为其本质还是使用了opencv的hsv颜色物体检测。下面话不多说了,来一起看看详细的介绍吧。

hsv介绍

hsv分别代表,色调(h:hue),饱和度(s:saturation),亮度(v:value),由a. r. smith在1978年创建的一种颜色空间, 也称六角锥体模型(hexcone model);

色调(h:hue):用角度度量,取值范围为0°~360°,从红色开始按逆时针方向计算,红色为0°,绿色为120°,蓝色为240°。它们的补色是:黄色为60°,青色为180°,品红为300°;(opencv中h的取值范围为0~180,8bit存储时);

饱和度(s:saturation):取值范围为0~255,值越大,颜色越饱和;

亮度(v:value):取值范围为0(黑色)~255(白色);

效果展示


实现思路

如上效果图所示,我们要做的就是把视频中的绿色的小猪佩奇识别出来即可,下面是的识别步骤:

  • 使用ps取的小猪佩奇颜色的hsb值,相当于opencv的hsv,不过ps的hsv(hsb)取值是:0~360、0~1、0~1,而opencv的hsv是:0~180、0~255、0~255,所以要对ps的hsv进行处理,h/2、sv*255;
  • 使用opencv位“与运算”提取hsv的颜色部分画面;
  • 使用高斯模糊优化图片;
  • 图片展示;

ps中工具栏右侧hsb显示:


完整代码

#coding=utf-8
#hsv转换(颜色提取)

import cv2
import numpy as np

cap = cv2.videocapture(0)

while (1):
 _, frame = cap.read()
 hsv = cv2.cvtcolor(frame, cv2.color_bgr2hsv)

 #在ps里用取色器的hsv
 pshsv = [112, 89, 52]
 diff = 40 #上下浮动值
 #因为ps的hsv(hsb)取值是:0~360、0~1、0~1,而opencv的hsv是:0~180、0~255、0~255,所以要对ps的hsv进行处理,h/2、sv*255
 lowerhsv = [(pshsv[0] - diff) / 2, (pshsv[1] - diff) * 255 / 100,
    (pshsv[2] - diff) * 255 / 100]
 upperhsv = [(pshsv[0] + diff) / 2, (pshsv[1] + diff) * 255 / 100,
    (pshsv[2] + diff) * 255 / 100]

 mask = cv2.inrange(hsv, np.array(lowerhsv), np.array(upperhsv))

 #使用位“与运算”提取颜色部分
 res = cv2.bitwise_and(frame, frame, mask=mask)
 #使用高斯模式优化图片
 res = cv2.gaussianblur(res, (5, 5), 1)

 cv2.imshow('frame', frame)
 # cv2.imshow('mask', mask)
 cv2.imshow('res', res)
 if cv2.waitkey(1) & 0xff == ord('q'):
  break

cv2.destroyallwindows()

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对移动技术网的支持。

如对本文有疑问,请在下面进行留言讨论,广大热心网友会与你互动!! 点击进行留言回复

相关文章:

验证码:
移动技术网