Reload scene config

2021-09-09 21:19:30 +02:00
parent 35bb9eaff0
commit 15f120f207
4 changed files with 93 additions and 110 deletions
--- a/active_grasp/policy.py
+++ b/active_grasp/policy.py
@@ -78,8 +78,7 @@ class Policy:
        return grasps[indices], scores[indices]

    def score_fn(self, grasp):
-        # return grasp.quality
-        return grasp.pose.translation[2]
+        return grasp.quality

    def update(self, img, pose):
        raise NotImplementedError