From 599f7d8957d1a8bb6a2a5d165d467f5feb6ec53a Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 29 Aug 2019 13:37:36 +0200 Subject: [PATCH] Fixed indexing Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 545888e..afe2d62 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -411,7 +411,7 @@ def _apply_entropy_threshold(observations: Sequence[np.ndarray], entropy_thresho final_observations.append(observations[i]) continue - filtered_image_observations = observations[observations[i][:, -1] < entropy_threshold] + filtered_image_observations = observations[i][observations[i][:, -1] < entropy_threshold] final_image_observations = np.copy(filtered_image_observations[:, -8:-1]) final_image_observations[:, 0] = np.argmax(filtered_image_observations[:, :-5], axis=-1) final_image_observations[:, 1] = np.amax(filtered_image_observations[:, :-5], axis=-1)