基於鼠標點擊的圖像分割

Question

使用python框架，我們可以創建圖像段，如附件所示。現在，基於鼠標點擊圖像片段，我們需要使用特定的顏色來突出顯示片段。

基於鼠標點擊我可以得到特定位置的x/y座標。請告訴我如何檢查座標屬於哪個圖像段？

以下是代碼片段：

from skimage.segmentation import felzenszwalb, slic,quickshift 
from skimage.segmentation import mark_boundaries 
from skimage.segmentation import find_boundaries 
from skimage.util import img_as_float 
from skimage import io 

import matplotlib.pyplot as plt 

from skimage import measure 
from skimage import restoration 
from skimage import img_as_float 

import numpy as np 

coords = [] 

def find_nearest(array,value): 

    idx = (np.abs(array-value)).argmin() 

    return array[idx] 

def onclick(event): 

    global ix, iy 

    ix, iy = event.xdata, event.ydata 

    print ('ix ',ix) 

    print ("iy ",iy) 

    color = np.float64([1,0,1]) # red color 

    image[segments == 14] = color 

    mark_boundaries(image, segments) 

    ax.imshow(mark_boundaries(image, segments)) 

    coords.append((ix, iy)) 

    return 

image=img_as_float(io.imread("amazon.jpg")) 

segments = quickshift(image, ratio=1.0, kernel_size=20, max_dist=10, 
      return_tree=False, sigma=0, convert2lab=True, random_seed=42) 

fig = plt.figure("Superpixels -- %d segments" % (500)) 

ax = fig.add_subplot(1, 1, 1) 

fig,ax = plt.subplots() 

color = np.float64([1,0,0]) 

image[segments == 14] = color # desired segment to be colored 

fig.canvas.mpl_connect('button_press_event', onclick) 

ax.imshow(mark_boundaries(image, segments)) 

plt.axis("off") 

plt.show() 

Answer 1

根據the documentation，quickshift返回指示段標籤的整數掩模。如果您知道用戶點擊了哪個像素，則可以檢查變量segments中該像素的值以獲取片段編號。

在點擊處理時，可以使用 clicked_segment = segments[event.xdata, event.ydata]