Add alignment baseline

2021-07-08 10:54:29 +02:00
parent 44190330fe
commit 8772e80321
3 changed files with 44 additions and 9 deletions
--- a/active_grasp/init.py
+++ b/active_grasp/init.py
@@ -1,7 +1,8 @@
 from .policy import register
 from .baselines import *

-register("single-view", SingleViewBaseline)
-register("top", TopBaseline)
-register("random", RandomBaseline)
-register("fixed-trajectory", FixedTrajectoryBaseline)
+register("single-view", SingleView)
+register("top", TopView)
+register("random", RandomView)
+register("fixed-trajectory", FixedTrajectory)
+register("alignment", AlignmentView)