fix heatmap bugs

bubbliiiing · Apr 18, 2022 · f371ab8 · f371ab8
1 parent 71c94df
commit f371ab8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolo.py b/yolo.py
@@ -304,7 +304,7 @@ def sigmoid(x):
             b, c, h, w = np.shape(sub_output)
             sub_output = np.transpose(np.reshape(sub_output, [b, 3, -1, h, w]), [0, 3, 4, 1, 2])[0]
             score      = np.max(sigmoid(sub_output[..., 4]), -1)
-            score      = cv2.resize(score, (image.size[1], image.size[0]))
+            score      = cv2.resize(score, (image.size[0], image.size[1]))
             normed_score    = (score * 255).astype('uint8')
             mask            = np.maximum(mask, normed_score)