Clear best grasp in case of no predictions

This commit is contained in:
Michel Breyer 2021-09-12 16:48:31 +02:00
parent 9f06bb27be
commit efe6e9d1bd

View File

@ -152,6 +152,9 @@ class MultiViewPolicy(Policy):
self.best_grasp = grasps[0] self.best_grasp = grasps[0]
self.vis.grasps(self.base_frame, grasps, scores, smin, smax) self.vis.grasps(self.base_frame, grasps, scores, smin, smax)
self.vis.best_grasp(self.base_frame, grasps[0], scores[0], 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): def compute_error(x_d, x):