import cv2
import glob
import os
import numpy as np
from medpy import metric
def calculate_metric_percase(pred, gt):
# dice = metric.binary.dc(pred, gt)
# jc = metric.binary.jc(pred, gt)
hd = metric.binary.hd95(pred, gt)
asd = metric.binary.asd(pred, gt)
return hd, asd
dice_mean = 0
hd_mean = 0
asd_mean = 0
hd = list()
asd =list()
predlist=glob.glob(os.path.join('./Dataset/data2/predict', "*"))#预测文件夹路径
gtlist=glob.glob(os.path.join('./Dataset/data2/val/M', "*"))#真值文件夹路径
for i in range(len(predlist)):
print(i, os.path.basename(predlist[i]))
img_ground = cv2.imread(predlist[i])
img_predict = cv2.imread(gtlist[i])
x = 0
y = 0
for w in range(256):
for h in range(256):
x += img_ground.item(w, h, 1)/255
y += img_predict.item(w, h, 1)/255
if x + y == 0 or x == 0 or y == 0:
current_hd=0
current_asd=0
else:
current_hd, current_asd = calculate_metric_percase(img_predict, img_ground)
print('hd', current_hd, 'asd', current_asd)
hd_mean += current_hd
hd.append((current_hd))
asd_mean += current_asd
asd.append((current_asd))
hd_mean /=len(predlist)
asd_mean /=len(predlist)
print('Dice',dice_mean)
print('hd',hd_mean,'asd',asd_mean)
Python测指标模板
最新推荐文章于 2023-09-28 17:08:46 发布