hofee/nbv_sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix init of baselines

Browse Source
This commit is contained in:
Michel Breyer 2021-08-10 11:02:56 +02:00
parent 0beca41c39
commit 020adebfd3
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
active_grasp/baselines.py
View File

@ -42,8 +42,8 @@ class RandomView(BasePolicy):
Move the camera to a random viewpoint on a circle centered above the target. Move the camera to a random viewpoint on a circle centered above the target.
""" """
def __init__(self, intrinsic): def __init__(self, rate, filter_grasps):
super().__init__(intrinsic) super().__init__(rate, filter_grasps)
self.r = 0.06 # radius of the circle self.r = 0.06 # radius of the circle
self.h = 0.3 # distance above bbox center 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. Follow a pre-defined circular trajectory centered above the target object.
""" """
def __init__(self, intrinsic): def __init__(self, rate, filter_grasps):
super().__init__(intrinsic) super().__init__(rate, filter_grasps)
self.r = 0.08 self.r = 0.08
self.h = 0.3 self.h = 0.3
self.duration = 6.0 self.duration = 6.0