python下对hsv颜色空间进行量化操作


Posted in Python onJune 04, 2020

更新:优化了代码,理由numpy的ufunc函数功能替换了之前的双重for循环,测试图片大小为692*1024*3,优化前运行时间为6.9s,优化后为0.8s。

由于工作需要,需要计算颜色直方图来提取颜色特征,但若不将颜色空间进行量化,则直方图矢量维数过高,不便于使用。但是看了opencv API后并未发现提供了相关函数能够在计算颜色直方图的同时进行量化,因此这部分功能只能自己实现。下面分为两个部分进行介绍:

一、颜色空间量化表

由于RGB模型不够直观,不符合人类视觉习惯,因此在进行颜色特征提取前,需要将照片从RGB颜色模型转换为更符合人类视觉的HSV模型。在提取颜色特征时,最常用的方法之一为颜色直方图法,但一张图片中出现的颜色一般特别多,导致直方图矢量的维数较高,因此需要对HSV空间进行量化。根据人眼对颜色的感知特性,采用较为常用的量化方法,即按照如下对应关系进行量化:

python下对hsv颜色空间进行量化操作

基于上述量化表,将各颜色分量按照下述公式合成为72维一维矢量:python下对hsv颜色空间进行量化操作

二、量化代码

代码使用纯python写成,效率偏低,处理388*500像素的照片用时1.45秒。在quantilize函数中,未使用if-else判断语句,因此至少节省了1/3的时间。但这个速度显然是无法令人满意的,用C++效率应该会更高点。如果有人有更好的想法,欢迎在下方评论交流。

#-*-coding:utf-8-*-
import cv2
import numpy as np
from datetime import datetime
from matplotlib import pyplot as plt

def colors(imagepath):
  img = cv2.imread(imagepath)
  hsv = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
  nhsv = np.zeros(hsv.shape[:2], dtype=np.uint8)
  t2 = datetime.now()
  for i in range(hsv.shape[0]):
    for j in range(hsv.shape[1]):
      nhsv[i][j] = quantilize(hsv[i][j])
  print datetime.now() - t2
  hist = cv2.calcHist([nhsv], [0], None, [72], [0,71]) # 40x faster than np.histogramfaster than np.histogram
  plt.plot(hist,color = 'r')
  plt.xlim([0, 72])
  plt.show()

def quantilize(value):
  '''hsv直方图量化
  value : [21, 144, 23] h, s, v
  opencv中,h-[0,180], s-[0,255], v-[0,255]
  '''
  # 
  value[0] = value[0] * 2
  hlist = [20, 40, 75, 155, 190, 270, 290, 316, 360]
  svlist = [21, 178, 255]
  for i in range(len(hlist)):
    if value[0] <= hlist[i]:
      h = i % 8
      break
  for i in range(len(svlist)):
    if value[1] <= svlist[i]:
      s = i
      break
  for i in range(len(svlist)):
    if value[2] <= svlist[i]:
      v = i
      break
  return 9 * h + 3 * s + v

以上,欢迎批评交流~

三、更新

#-*-coding:utf-8-*-
import cv2
import numpy as np
from datetime import datetime
from matplotlib import pyplot as plt

hlist = [20, 40, 75, 155, 190, 270, 290, 316, 360]
svlist = [21, 178, 255]

def quantilize(h, s, v):
  '''hsv直方图量化'''
  # value : [21, 144, 23] h, s, v
  h = h * 2
  for i in range(len(hlist)):
    if h <= hlist[i]:
      h = i % 8
      break
  for i in range(len(svlist)):
    if s <= svlist[i]:
      s = i
      break
  for i in range(len(svlist)):
    if v <= svlist[i]:
      v = i
      break
  return 9 * h + 3 * s + v

quantilize_ufunc = np.frompyfunc(quantilize, 3, 1) # 自定义ufunc函数,即将quantilize函数转化为ufunc函数,其输入参数为3个,输出参数为1个。

def colors(img):
  hsv = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
  nhsv = quantilize_ufunc(hsv[:,:,0], hsv[:,:,1], hsv[:,:,2]).astype(np.uint8) # 由于frompyfunc函数返回结果为对象,所以需要转换类型
  hist = cv2.calcHist([nhsv], [0], None, [72], [0,71]) # 40x faster than np.histogram
  hist = hist.reshape(1, hist.shape[0]).astype(np.int32).tolist()[0]
  return hist

if __name__ == '__main__':
  img_path = path + 'test.jpg'
  img = cv2.imread(img_path)
  colors(img)

结果

[0, 11, 31490, 100, 3, 32685, 0, 28, 313, 0, 0, 3268, 31, 0, 558364, 6, 1, 441, 0, 0, 2301, 92, 0, 34056, 0, 1, 396, 0, 0, 2682, 84, 5, 712, 0, 137, 55, 0, 0, 1215, 20, 2, 224, 0, 3, 0, 0, 0, 13838, 56, 0, 23474, 63, 23, 1, 0, 0, 4764, 0, 0, 2335, 0, 25, 27, 0, 0, 2302, 5, 0, 1676, 1, 59, 389]

以上这篇python下对hsv颜色空间进行量化操作就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python的Tornado框架实现图片上传及图片大小修改功能
Jun 30 Python
Python正则表达式实现截取成对括号的方法
Jan 06 Python
Python_LDA实现方法详解
Oct 25 Python
Python模拟随机游走图形效果示例
Feb 06 Python
tensorflow实现KNN识别MNIST
Mar 12 Python
python爬取cnvd漏洞库信息的实例
Feb 14 Python
python中pip的使用和修改下载源的方法
Jul 08 Python
使用python写的opencv实时监测和解析二维码和条形码
Aug 14 Python
python爬虫 线程池创建并获取文件代码实例
Sep 28 Python
opencv3/C++实现视频读取、视频写入
Dec 11 Python
解决windows上安装tensorflow时报错,“DLL load failed: 找不到指定的模块”的问题
May 20 Python
python 模块导入问题汇总
Feb 01 Python
Python-opencv实现红绿两色识别操作
Jun 04 #Python
Python基于pandas绘制散点图矩阵代码实例
Jun 04 #Python
Python使用plt.boxplot() 参数绘制箱线图
Jun 04 #Python
浅谈opencv自动光学检测、目标分割和检测(连通区域和findContours)
Jun 04 #Python
Python中操作各种多媒体,视频、音频到图片的代码详解
Jun 04 #Python
Python简单实现词云图代码及步骤解析
Jun 04 #Python
Python坐标轴操作及设置代码实例
Jun 04 #Python
You might like
解析yii数据库的增删查改
2013/06/20 PHP
深入理解PHP中的global
2014/08/19 PHP
php实现获取及设置用户访问页面语言类
2014/09/24 PHP
Laravel框架中实现使用阿里云ACE缓存服务
2015/02/10 PHP
PHP面向对象程序设计类的定义与用法简单示例
2016/12/27 PHP
浅谈PHP中如何实现Hook机制
2017/11/14 PHP
PHP实现链表的定义与反转功能示例
2018/06/09 PHP
新闻内页-JS分页
2006/06/07 Javascript
jquery select操作的日期联动实现代码
2009/12/06 Javascript
关于JavaScript中var声明变量作用域的推断
2010/12/16 Javascript
20款非常优秀的 jQuery 工具提示插件 推荐
2012/07/15 Javascript
巧用js提交表单轻松解决一个页面有多个提交按钮
2013/11/17 Javascript
node.js中的fs.closeSync方法使用说明
2014/12/17 Javascript
Bootstrap开发实战之第一次接触Bootstrap
2016/06/02 Javascript
微信小程序 wxapp内容组件 progress详细介绍
2016/10/31 Javascript
图片懒加载插件实例分享(含解析)
2017/01/09 Javascript
JavaScript错误处理操作实例详解
2019/01/04 Javascript
浅谈webpack devtool里的7种SourceMap模式
2019/01/14 Javascript
js实现随机div颜色位置 类似满天星效果
2019/10/24 Javascript
javascript设计模式 ? 抽象工厂模式原理与应用实例分析
2020/04/09 Javascript
js实现随机圆与矩形功能
2020/10/29 Javascript
Python使用win32com实现的模拟浏览器功能示例
2017/07/13 Python
Python:slice与indices的用法
2019/11/25 Python
Python将列表中的元素转化为数字并排序的示例
2019/12/25 Python
戴森英国官网:Dyson英国
2019/05/07 全球购物
中英文自我评价语句
2013/12/20 职场文书
陈安之励志演讲稿
2014/08/21 职场文书
先进班集体事迹材料
2014/12/25 职场文书
春节慰问信范文
2015/02/15 职场文书
雾霾停课通知
2015/04/24 职场文书
2015年幼儿园大班工作总结
2015/04/25 职场文书
2015年法律事务部工作总结
2015/07/27 职场文书
python通过函数名调用函数的几种方法总结
2021/06/07 Python
Pyqt5将多个类组合在一个界面显示的完整示例
2021/09/04 Python
浅谈克隆 JavaScript
2021/11/02 Javascript
一次Mysql update sql不当引起的生产故障记录
2022/04/01 MySQL