diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 4d5b627..07dfc57 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -541,6 +541,8 @@ def _get_observations(detections: Sequence[np.ndarray]) -> List[np.ndarray]: observations[i].append(observation_mean) final_observations.append(np.asarray(observations[i])) + if not len(observations[i]): + continue final_observations[i][:, -1] = -np.sum(final_observations[i][:, :-5] * np.log(final_observations[i][:, :-5]), axis=-1)