(outputs: np.array, labels: np.array)
| 229 | |
| 230 | |
| 231 | def iou(outputs: np.array, labels: np.array): |
| 232 | |
| 233 | SMOOTH = 1e-6 |
| 234 | intersection = (outputs & labels).sum((1, 2)) |
| 235 | union = (outputs | labels).sum((1, 2)) |
| 236 | |
| 237 | iou = (intersection + SMOOTH) / (union + SMOOTH) |
| 238 | |
| 239 | |
| 240 | return iou.mean() |
| 241 | |
| 242 | class DiceCoeff(Function): |
| 243 | """Dice coeff for individual examples""" |