diff --git a/src/active_grasp/nbv.py b/src/active_grasp/nbv.py index 2bd4763..f897093 100644 --- a/src/active_grasp/nbv.py +++ b/src/active_grasp/nbv.py @@ -80,7 +80,7 @@ class NextBestView(MultiViewPolicy): i = np.argmax(utilities) nbv, gain = views[i], gains[i] - if gain < self.min_gain: + if gain < self.min_gain and len(self.views) > self.T: self.done = True self.x_d = nbv