ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

二维图像Dice系数计算

2022-01-02 23:58:20  阅读:166  来源: 互联网

标签:__ 系数 Dice pred 二维 np true dice


项目场景:计算二维图像的Dice系数

在这里插入图片描述

问题描述:

通常我们用目标分割网络,预测结果后。为了得知网络的准确度,可以计算其Dice系数,通过比较其系数,可以得知网络的准确性。

import numpy as np
import cv2
from PIL import Image

if __name__ == '__main__':
    y_true_path = 'E:/AI-challenge/2021/results/label_show/'(这个是标签的文件地址)
    y_pred_path = 'E:/AI-challenge/2021/results/predict/'(这个是预测后的图像输出地址)
    dice_list = []
    for i in range(398):
        y_true = np.array(Image.open(y_true_path+str(i+1)+'.jpg'))
        y_pred = cv2.imread(y_pred_path+str(i+1)+'.jpg',cv2.IMREAD_GRAYSCALE)
        y_true = y_true/255
        y_pred = y_pred/255
        union = y_true * y_pred
        dice = 2*np.sum(union)/(np.sum(y_true)+np.sum(y_pred))
        dice_list.append(dice)
        print(i+1, ' ', dice, '\n')
    print(dice_list)
    print(np.mean(np.array(dice_list)))
        }

运行结果:

在这里插入图片描述
这样就可以求出每张图片对应的Dice系数。

标签:__,系数,Dice,pred,二维,np,true,dice
来源: https://blog.csdn.net/pidzhengding/article/details/122282714

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有