Single view baselines

2021-08-25 18:29:10 +02:00
parent fc8d9a8791
commit 5501c2ae42
6 changed files with 89 additions and 207 deletions
--- a/active_grasp/init.py
+++ b/active_grasp/init.py
@@ -1,10 +1,6 @@
 from .policy import register
 from .baselines import *
-from .nbv import *

-register("single-view", SingleView)
-register("top", TopView)
-register("random", RandomView)
-register("fixed-trajectory", FixedTrajectory)
-register("alignment", AlignmentView)
-register("nbv", NextBestView)
+register("initial-view", InitialView)
+register("front-view", FrontView)
+register("top-view", TopView)