From 4986bec5d319c33baea3949ff2378ad371da0a15 Mon Sep 17 00:00:00 2001 From: yalaudah Date: Fri, 5 Jun 2020 17:53:02 +0000 Subject: [PATCH] bug fix to model predictions (#345) --- experiments/interpretation/dutchf3_patch/local/test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experiments/interpretation/dutchf3_patch/local/test.py b/experiments/interpretation/dutchf3_patch/local/test.py index 09d5ebf3..d417dfac 100644 --- a/experiments/interpretation/dutchf3_patch/local/test.py +++ b/experiments/interpretation/dutchf3_patch/local/test.py @@ -236,7 +236,7 @@ def _patch_label_2d( # save image: image_to_disk(np.array(batch[i, 0, :, :]), path_prefix + "_img.png") # dump model prediction: - mask_to_disk(model_output[i, :, :, :].argmax(dim=1).numpy(), path_prefix + "_pred.png", num_classes) + mask_to_disk(model_output[i, :, :, :].argmax(dim=0).numpy(), path_prefix + "_pred.png", num_classes) # dump model confidence values for nclass in range(num_classes): image_to_disk(