Add setup.py file
This commit is contained in:
36
scripts/run.py
Normal file
36
scripts/run.py
Normal 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()
|
Reference in New Issue
Block a user