Share view computation among all policies

2021-09-11 14:52:27 +02:00
parent 3550b49669
commit 4ebd587553
5 changed files with 49 additions and 45 deletions
--- a/active_grasp/bbox.py
+++ b/active_grasp/bbox.py
@@ -9,6 +9,7 @@ class AABBox:
    def __init__(self, bbox_min, bbox_max):
        self.min = np.asarray(bbox_min)
        self.max = np.asarray(bbox_max)
+        self.center = 0.5 * (self.min + self.max)

    @property
    def corners(self):