Add setup.py file

This commit is contained in:
Michel Breyer
2021-07-07 16:29:50 +02:00
parent f29d861fd4
commit 6658b8c7f0
11 changed files with 61 additions and 65 deletions

36
scripts/run.py Normal file
View File

@@ -0,0 +1,36 @@
import argparse
import rospy
from active_grasp.controller import GraspController
from active_grasp.policies import get_policy
def create_parser():
parser = argparse.ArgumentParser()
parser.add_argument(
"--policy",
type=str,
choices=[
"single-view",
"top",
"alignment",
"random",
"fixed-trajectory",
],
)
return parser
def main():
rospy.init_node("active_grasp")
parser = create_parser()
args = parser.parse_args()
policy = get_policy(args.policy)
controller = GraspController(policy)
while True:
controller.run()
if __name__ == "__main__":
main()