如何计算scikit-image中超像素的平均颜色

时间:2017-01-10 21:10:44

标签: python image-processing computer-vision scikit-image superpixels

我使用scikit图片库中的quickshift方法对图像进行了分割。如何计算超像素的平均颜色和面积?如何解释quickshift()方法的返回值?文档说返回值为"整数掩码表示段标签"但这对我来说并不清楚。如何在原始图像的形状中创建一个布尔数组,其中填充了特定超像素的位置,在此演示文稿中,我的生活将更容易(我曾经在OpenCV中使用这种类型的蒙版)。你能帮帮我吗?我的代码(来自scikit-image网站的简化示例):

from skimage.data import astronaut
from skimage.segmentation import felzenszwalb, slic, quickshift
from skimage.segmentation import mark_boundaries
from skimage.util import img_as_float

img = img_as_float(astronaut()[::2, ::2])
segments_quick = quickshift(img, kernel_size=3, max_dist=6, ratio=0.5)

print("Quickshift number of segments: %d" % len(np.unique(segments_quick)))
plt.imshow(mark_boundaries(img, segments_quick))

plt.show()

3 个答案:

答案 0 :(得分:6)

skimage.measure.regionprops函数返回标记区域的属性,包括您感兴趣的区域:平均值和面积。您可以按如下方式使用它:

from skimage import measure
regions = measure.regionprops(segments_quick, intensity_image=img)

regions是一个列表,每个条目都是一个表示超像素的RegionProperty对象。您可以按如下方式查询区域属性:

print([r.area for r in regions])
print([r.mean_intensity for r in regions])

答案 1 :(得分:3)

这是一个简单,直接且通用的代码,用于表示每个像素的超像素平均值。

这里使用SLIC获得标签数组,可以使用任何其他方法来生成标签。

import numpy as np
import cv2
from skimage import segmentation
from skimage.data import astronaut

img=cv2.cvtColor(astronaut(),cv2.COLOR_BGR2RGB)

enter image description here

label=segmentation.slic(img,compactness=10, n_segments=1000)

def mean_image(image,label):

im_rp=image.reshape((image.shape[0]*image.shape[1],image.shape[2]))
sli_1d=np.reshape(label,-1)    
uni=np.unique(sli_1d)
uu=np.zeros(im_rp.shape)
for i in uni:
    loc=np.where(sli_1d==i)[0]
    #print(loc)
    mm=np.mean(im_rp[loc,:],axis=0)
    uu[loc,:]=mm
oo=np.reshape(uu,[image.shape[0],image.shape[1],image.shape[2]]).astype('uint8')
cv2.imshow('img',oo)

output=mean_image(img,label) # displays the output image.

enter image description here

答案 2 :(得分:0)

对我来说最简单的方法是:

from skimage import segmentation, color
from skimage.io import imread
from skimage.future import graph
from matplotlib import pyplot as plt
img = imread('test.jpeg')

img_segments = segmentation.slic(img, compactness=20, n_segments=500)
superpixels = color.label2rgb(img_segments, img, kind='avg')