【深度学习】【实例分割】mask的IoU的代码分析

最新推荐文章于 2023-12-26 16:46:59 发布

Hanawh

最新推荐文章于 2023-12-26 16:46:59 发布

阅读量4.9k

点赞数 6

分类专栏：深度学习文章标签：深度学习

本文链接：https://blog.csdn.net/qq_36530992/article/details/102549006

版权

本文深入探讨了深度学习中的实例分割，特别是针对mask的IoU（Intersection over Union）计算。通过示例代码，详细解释了如何进行IoU的求解过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

实例分割中mask的IoU的代码分析）

(话不多说上代码）

def mask_iou(mask1, mask2):
    """
    mask1: [m1,n] m1 means number of predicted objects 
    mask2: [m2,n] m2 means number of gt objects
    Note: n means image_w x image_h
    """
    intersection = torch.matmul(mask1, mask2.t())
    area1 = torch.sum(mask1, dim=1).view(1, -1)