argmax(output, dim=1) # convert with argmax to shape of target # first compute statistics for true positives, false positives, false negative and # true negative "pixels" tp, fp, fn, tn = smp.metrics.get_stats(output, target, mode='multiclass') # compute metric iou_score = smp.metr...