nbv_sim/scripts/run.py

import argparse
import rospy

from active_grasp.controller import GraspController
from active_grasp.policy import make, registry


def create_parser():
    parser = argparse.ArgumentParser()
    parser.add_argument("--policy", type=str, choices=registry.keys())
    return parser


def main():
    rospy.init_node("active_grasp")
    parser = create_parser()
    args = parser.parse_args()
    policy = make(args.policy)
    controller = GraspController(policy)

    while True:
        controller.run()


if __name__ == "__main__":
    main()