-
Notifications
You must be signed in to change notification settings - Fork 9
/
main.py
executable file
·34 lines (24 loc) · 889 Bytes
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#!/usr/bin/python
import argparse
p = argparse.ArgumentParser()
p.add_argument('--test', action='store_true', help='Verify Denavit-Hartenberg for two link case')
p.add_argument('--train', action='store_true', help='Train network to learn IK')
p.add_argument('--infer', action='store_true', help='Run inference test on existing network model')
p.add_argument('--table', action='store_true', help='Produce table of determinant values')
p.add_argument('--ik', action='store_true', help='Run inverse kinematic test using sympy solver')
args = p.parse_args()
if args.test:
from tests.test import Test
Test().test()
if args.train:
from src.train import Train
Train().train()
if args.infer:
from src.infer import Infer
Infer().infer()
if args.table:
from tests.table import Table
Table().table()
if args.ik:
from tests.ik import IK
IK().ik()