Fix issue with nans

2021-08-10 17:17:37 +02:00 · 2021-08-10 17:17:37 +02:00 · 5a33561abb
commit 5a33561abb
parent 020adebfd3
1 changed files with 1 additions and 0 deletions
--- a/active_grasp/policy.py
+++ b/active_grasp/policy.py
@ -85,6 +85,7 @@ class BasePolicy(Policy):
            # The next line prints a warning since some voxels have no grasp
            # predictions resulting in empty slices.
            qual = np.mean(self.qual_hist[:T, ...], axis=0, where=mask)
+            qual = np.nan_to_num(qual, copy=False)
            qual = threshold_quality(qual, 0.9)
            index_list = select_local_maxima(qual, 3)