Add robot configuration to the state

2021-09-12 14:40:17 +02:00
parent 027a925693
commit 8ba015b1ef
4 changed files with 15 additions and 13 deletions
--- a/active_grasp/nbv.py
+++ b/active_grasp/nbv.py
@@ -73,7 +73,7 @@ class NextBestView(MultiViewPolicy):
            if self.is_feasible(view):
                self.view_candidates.append(view)

-    def update(self, img, x):
+    def update(self, img, x, q):
        if len(self.views) > self.max_views or self.best_grasp_prediction_is_stable():
            self.done = True
        else: