mask iou iou 合并代码

AI算法网奇

已于 2025-05-27 20:50:02 修改

阅读量300

点赞数

CC 4.0 BY-SA版权

分类专栏： c++入门宝典文章标签：算法

于 2023-11-13 18:29:56 首次发布

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/134383669

c++入门宝典专栏收录该内容

359 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客介绍了IoU（Intersection over Union）合并算法在PyTorch和C++两种环境下的实现，包括PyTorch的调用方法和C++版的代码示例，特别地，C++代码限制了最多合并3个边界框。

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

mask iou

    def calculate_iou(mask1, mask2):
        # Convert masks to float tensors for calculations
        mask1 = mask1.to(torch.float32)
        mask2 = mask2.to(torch.float32)
        
        # Calculate intersection and union
        intersection = (mask1 * mask2).sum()
        union = mask1.sum() + mask2.sum() - intersection
        
        # Calculate IoU
        iou = intersection / union
        iou1=intersection / min(mask1.sum(),mask2.sum())
        return iou,iou1