From a74c830cfcfebe55d79e68f69c32cd439564b12e Mon Sep 17 00:00:00 2001 From: peteflorence Date: Mon, 19 Jun 2017 19:12:59 -0400 Subject: [PATCH] fix input to softmax_to_unary --- image_segmentation_conditional_random_fields.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/image_segmentation_conditional_random_fields.ipynb b/image_segmentation_conditional_random_fields.ipynb index bd7cfbc..d174ab1 100644 --- a/image_segmentation_conditional_random_fields.ipynb +++ b/image_segmentation_conditional_random_fields.ipynb @@ -764,7 +764,7 @@ "\n", "softmax = final_probabilities.squeeze()\n", "\n", - "softmax = processed_probabilities.transpose((2, 0, 1))\n", + "processed_probabilities = softmax.transpose((2, 0, 1))\n", "\n", "# The input should be the negative of the logarithm of probability values\n", "# Look up the definition of the softmax_to_unary for more information\n",