(gt, pred, voxel_spacing)
| 16 | from medpy import metric |
| 17 | |
| 18 | def cal_metric(gt, pred, voxel_spacing): |
| 19 | if pred.sum() > 0 and gt.sum() > 0: |
| 20 | dice = metric.binary.dc(pred, gt) |
| 21 | hd95 = metric.binary.hd95(pred, gt, voxelspacing=voxel_spacing) |
| 22 | return np.array([dice, hd95]) |
| 23 | else: |
| 24 | return np.array([0.0, 50]) |
| 25 | |
| 26 | def each_cases_metric(gt, pred, voxel_spacing): |
| 27 | classes_num = gt.max() + 1 |