Added yet another empty check

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-08-29 15:43:08 +02:00
parent 25e39edbcd
commit b49087f415
1 changed files with 2 additions and 0 deletions

View File

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