2martens
/
masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed indexing 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-08-29 13:37:36 +02:00
parent cf132a89f0
commit 599f7d8957
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/ssd.py
View File

@ -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)