nlf 孔洞处理

AI算法网奇

于 2025-05-23 11:52:46 发布

阅读量98

点赞数

分类专栏： python基础文章标签： python 人工智能开发语言

本文链接：https://blog.csdn.net/jacke121/article/details/148161971

版权

python基础专栏收录该内容

772 篇文章

订阅专栏

https://github.com/isarandi/stcnbuf/blob/1c309eb4d6c7e70f9a3bc98ee432664e828da139/src/stcnbuf/morph.py#L40

def process_mask_inplace(rle):
    rle.dilate5x5(inplace=True)
    rle.dilate5x5(inplace=True)
    rle.erode5x5(inplace=True)
    rle.erode5x5(inplace=True)
    # rle.remove_small_components(100, inplace=True)
    # rle.fill_small_holes(100, inplace=True)
    bbox = rle.largest_connected_component().bbox()
    bbox = [bbox[0] - bbox[2] * 0.15, bbox[1] - bbox[3] * 0.15, bbox[2] * 1.3, bbox[3] * 1.3]
    rle &= rlemasklib.RLEMask.from_bbox(bbox, rle.shape)
    return rle


def process_mask_dict(d):
    rle = rlemasklib.RLEMask.from_dict(d)
    rle = process_mask_inplace(rle)
    return rle.to_dict()