From 020adebfd3791705e6d11c35eb5ed372a6bd1ee6 Mon Sep 17 00:00:00 2001 From: Michel Breyer Date: Tue, 10 Aug 2021 11:02:56 +0200 Subject: [PATCH] Fix init of baselines --- active_grasp/baselines.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/active_grasp/baselines.py b/active_grasp/baselines.py index ea16176..81a7f07 100644 --- a/active_grasp/baselines.py +++ b/active_grasp/baselines.py @@ -42,8 +42,8 @@ class RandomView(BasePolicy): Move the camera to a random viewpoint on a circle centered above the target. """ - def __init__(self, intrinsic): - super().__init__(intrinsic) + def __init__(self, rate, filter_grasps): + super().__init__(rate, filter_grasps) self.r = 0.06 # radius of the circle self.h = 0.3 # distance above bbox center @@ -68,8 +68,8 @@ class FixedTrajectory(BasePolicy): Follow a pre-defined circular trajectory centered above the target object. """ - def __init__(self, intrinsic): - super().__init__(intrinsic) + def __init__(self, rate, filter_grasps): + super().__init__(rate, filter_grasps) self.r = 0.08 self.h = 0.3 self.duration = 6.0