python 图像追踪_图像色彩追踪 python实现

色彩追踪指的是找出RGB图像中特定颜色在原图中所在的位置

思考：由于在RGB色彩空间中颜色有256³种，色彩追踪是比较困难的。所以我们考虑先将图像转换到HSV色彩空间中。

HSV变换，是将RGB变换到H(Hue：色相)、S(Saturation：饱和度)、V(Value：明度)的方法

关于HSV色彩空间的一些说明：

饱和度越小，图像越白，饱和度越大，颜色越浓烈，0<=S<=1;

明度数值越高图像越接近于白色，越低越接近于黑色，0<=V<=1;

色相：将颜色用0到360度表示，具体色相与数值对应关系可参考下图：

色相数值与颜色的对应关系

实验：追踪图像中的蓝色部分：

思考：为了追踪蓝色，可以将RGB图像转换为HSV图像后，提取色相在180°到240°之间的图像，将其值置为255(白色)。

import cv2

import numpy as np

import matplotlib.pyplot as plt

# BGR -> HSV

def BGR2HSV(_img):

img = _img.copy() / 255.

hsv = np.zeros_like(img, dtype=np.float32)

# get max and min

max_v = np.max(img, axis=