From efe6e9d1bdfa78623cc0ae2271e541744e664943 Mon Sep 17 00:00:00 2001 From: Michel Breyer Date: Sun, 12 Sep 2021 16:48:31 +0200 Subject: [PATCH] Clear best grasp in case of no predictions --- active_grasp/policy.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/active_grasp/policy.py b/active_grasp/policy.py index e87d91b..6ebe1be 100644 --- a/active_grasp/policy.py +++ b/active_grasp/policy.py @@ -152,6 +152,9 @@ class MultiViewPolicy(Policy): self.best_grasp = grasps[0] self.vis.grasps(self.base_frame, grasps, scores, smin, smax) self.vis.best_grasp(self.base_frame, grasps[0], scores[0], smin, smax) + else: + self.best_grasp = None + # TODO clear grasp markers def compute_error(x_d, x):