语义分割计算IOU，python实现

故事已经翻了几页

已于 2023-02-10 18:15:18 修改

阅读量5.4k

点赞数 1

分类专栏： Python图像处理文章标签： python 深度学习计算机视觉

于 2022-01-25 18:04:26 首次发布

本文链接：https://blog.csdn.net/gushiyi27/article/details/122689939

版权

话不多说，直接上代码

#计算IOU，本代码只适用于单通道图片，单类型分割，且分割出来物体的为白色，背景为黑色。
from PIL import Image
import numpy as np
import glob

def get_array(path):
    img = Image.open(path)
    img = img.getdata()
    img = np.array(img)
    return img

def get_IOU(maskpath,resultpath):
    mask = get_array(maskpath)
    result = get_array(resultpath)
    #计算iou
    S1 = 0 #交集
    S2 = 0 #并集