利用scipy.ndimage.binary_fill_hoses提取标记轮廓的mask-20201025

颜色标记提取

最新推荐文章于 2021-08-19 15:59:36 发布

原创最新推荐文章于 2021-08-19 15:59:36 发布 · 467 阅读

1 ·

CC 4.0 BY-SA版权

图像处理专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种基于颜色范围的图像标记提取方法。通过定义特定颜色的阈值，利用Python中的NumPy和SciPy库来创建颜色掩膜(mask)，并进一步填充轮廓以完整提取目标区域。

根据刘博记录如何提取标记内的mask，中间是自己的理解过程，可删除精简。

import numpy as np
import matplotlib.pyplot as plt
import skimage.io as io
import scipy

def marker_color(color='yellow'):
    return [255, 242, 0]

img = io.imread('./8_2.png')[...,0:3]
mask1 = np.all(np.array([255, 242, 0]) - 40 < img, axis=-1)
plt.imshow(mask1)
plt.show()
mask2 = np.all(np.array([255, 242, 0]) + 40 > img, axis=-1)
plt.imshow(mask2)
plt.show()
mask = np.logical_and(mask1, mask2)
plt.imshow(mask)
plt.show()
mask = scipy.ndimage.binary_fill_holes(mask)
plt.imshow(mask)
plt.show()

fig,axes = plt.subplots(1,2,figsize=(10,5))
axes[0].imshow(img)
axes[1].imshow(mask, cmap='gray')
plt.show()
io.imsave('tmp.png', mask*255)