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

xiekai1116

于 2020-10-25 09:27:57 发布

阅读量447

点赞数 1

分类专栏：图像处理

本文链接：https://blog.csdn.net/csdnxiekai/article/details/109269980

版权

图像处理专栏收录该内容

11 篇文章

订阅专栏

根据刘博记录如何提取标记内的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)