hofee/nbv_sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix issue with nans

Browse Source
This commit is contained in:
Michel Breyer 2021-08-10 17:17:37 +02:00
parent 020adebfd3
commit 5a33561abb
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
active_grasp/policy.py
View File

@ -85,6 +85,7 @@ class BasePolicy(Policy):
# The next line prints a warning since some voxels have no grasp # The next line prints a warning since some voxels have no grasp
# predictions resulting in empty slices. # predictions resulting in empty slices.
qual = np.mean(self.qual_hist[:T, ...], axis=0, where=mask) 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) qual = threshold_quality(qual, 0.9)
index_list = select_local_maxima(qual, 3) index_list = select_local_maxima(qual, 3)