Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
C
cs598mp-fall2021-proj
Manage
Activity
Members
Labels
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Deploy
Releases
Model registry
Analyze
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
chsieh16
cs598mp-fall2021-proj
Commits
85362be9
Commit
85362be9
authored
2 years ago
by
chsieh16
Browse files
Options
Downloads
Patches
Plain Diff
Rename ctd back to cte fpr consistency
parent
0102a733
No related branches found
Branches containing commit
No related tags found
No related merge requests found
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
agbot_stanley_teacher.py
+10
-10
10 additions, 10 deletions
agbot_stanley_teacher.py
gem_stanley_teacher.py
+6
-6
6 additions, 6 deletions
gem_stanley_teacher.py
with
16 additions
and
16 deletions
agbot_stanley_teacher.py
+
10
−
10
View file @
85362be9
...
...
@@ -55,8 +55,8 @@ class DTreeAgBotStanleyGurobiTeacher(DTreeGurobiTeacherBase):
return
new_x
,
new_y
,
new_yaw
def
g
(
perc
):
prcv_ct
d
,
prcv_heading
=
perc
error
=
prcv_heading
+
np
.
arctan
(
K_P
*
prcv_ct
d
/
FORWARD_VEL
)
prcv_ct
e
,
prcv_heading
=
perc
error
=
prcv_heading
+
np
.
arctan
(
K_P
*
prcv_ct
e
/
FORWARD_VEL
)
# Calculate controller output
ang_vel
=
error
/
CYCLE_TIME
if
ang_vel
>
ANG_VEL_LIM
:
...
...
@@ -72,8 +72,8 @@ class DTreeAgBotStanleyGurobiTeacher(DTreeGurobiTeacherBase):
return
self
.
PERC_GT
@
state
def
v
(
gt_perc
)
->
float
:
ct
d
,
psi
=
gt_perc
return
abs
(
psi
+
np
.
arctan
((
K_P
*
ct
d
)
/
FORWARD_VEL
))
ct
e
,
psi
=
gt_perc
return
abs
(
psi
+
np
.
arctan
((
K_P
*
ct
e
)
/
FORWARD_VEL
))
def
spec
(
state
,
perc
)
->
bool
:
v_old
=
v
(
m_star
(
state
))
...
...
@@ -95,20 +95,20 @@ class DTreeAgBotStanleyGurobiTeacher(DTreeGurobiTeacherBase):
m
=
self
.
_gp_model
old_x
,
old_y
,
old_yaw
=
self
.
_old_state
.
tolist
()
new_x
,
new_y
,
new_yaw
=
self
.
_new_state
.
tolist
()
ct
d
,
psi
=
self
.
_percept
.
tolist
()
ct
e
,
psi
=
self
.
_percept
.
tolist
()
ang_vel
,
=
self
.
_control
.
tolist
()
# Controller
K_ct
d
_V
=
m
.
addVar
(
name
=
"
K*dhat/Vf
"
,
lb
=-
K_CTE_V_LIM
,
ub
=
K_CTE_V_LIM
)
m
.
addConstr
(
K_ct
d
_V
==
K_P
*
ct
d
/
FORWARD_VEL
)
atan_K_ct
d
_V
=
m
.
addVar
(
name
=
"
atan(K*dhat/Vf)
"
,
K_ct
e
_V
=
m
.
addVar
(
name
=
"
K*dhat/Vf
"
,
lb
=-
K_CTE_V_LIM
,
ub
=
K_CTE_V_LIM
)
m
.
addConstr
(
K_ct
e
_V
==
K_P
*
ct
e
/
FORWARD_VEL
)
atan_K_ct
e
_V
=
m
.
addVar
(
name
=
"
atan(K*dhat/Vf)
"
,
lb
=-
ATAN_K_CTE_V_LIM
,
ub
=
ATAN_K_CTE_V_LIM
)
m
.
addGenConstrTan
(
xvar
=
atan_K_ct
d
_V
,
yvar
=
K_ct
d
_V
)
m
.
addGenConstrTan
(
xvar
=
atan_K_ct
e
_V
,
yvar
=
K_ct
e
_V
)
# Clip angular velocity
raw_ang_vel
=
m
.
addVar
(
name
=
"
(φhat+atan(K*dhat/V))/T
"
,
lb
=-
RAW_ANG_VEL_LIM
,
ub
=
RAW_ANG_VEL_LIM
)
m
.
addConstr
(
raw_ang_vel
==
(
psi
+
atan_K_ct
d
_V
)
/
CYCLE_TIME
)
m
.
addConstr
(
raw_ang_vel
==
(
psi
+
atan_K_ct
e
_V
)
/
CYCLE_TIME
)
ang_vel
=
m
.
addVar
(
name
=
"
ω
"
,
lb
=-
ANG_VEL_LIM
,
ub
=
ANG_VEL_LIM
)
m
.
addGenConstrPWL
(
name
=
"
clip
"
,
xvar
=
raw_ang_vel
,
yvar
=
ang_vel
,
...
...
This diff is collapsed.
Click to expand it.
gem_stanley_teacher.py
+
6
−
6
View file @
85362be9
...
...
@@ -46,8 +46,8 @@ class DTreeGEMStanleyGurobiTeacher(DTreeGurobiTeacherBase):
assert
len
(
ex
)
==
self
.
state_dim
+
self
.
perc_dim
def
g
(
perc
):
ct
d
,
psi
=
perc
error
=
psi
+
np
.
arctan
(
K_P
*
ct
d
/
FORWARD_VEL
)
ct
e
,
psi
=
perc
error
=
psi
+
np
.
arctan
(
K_P
*
ct
e
/
FORWARD_VEL
)
steer
=
np
.
clip
(
error
,
-
STEERING_LIM
,
STEERING_LIM
)
return
(
steer
,)
...
...
@@ -85,12 +85,12 @@ class DTreeGEMStanleyGurobiTeacher(DTreeGurobiTeacherBase):
m
=
self
.
_gp_model
old_x
,
old_y
,
old_yaw
=
self
.
_old_state
.
tolist
()
new_x
,
new_y
,
new_yaw
=
self
.
_new_state
.
tolist
()
ct
d
,
psi
=
self
.
_percept
.
tolist
()
ct
e
,
psi
=
self
.
_percept
.
tolist
()
steering
,
=
self
.
_control
.
tolist
()
# Controller
K_cte_V
=
m
.
addVar
(
name
=
"
K*d/Vf
"
,
lb
=-
K_CTE_V_LIM
,
ub
=
K_CTE_V_LIM
)
m
.
addConstr
(
K_cte_V
==
K_P
*
ct
d
/
FORWARD_VEL
)
m
.
addConstr
(
K_cte_V
==
K_P
*
ct
e
/
FORWARD_VEL
)
atan_K_cte_V
=
m
.
addVar
(
name
=
"
atan(K*d/Vf)
"
,
lb
=-
ATAN_K_CTE_V_LIM
,
ub
=
ATAN_K_CTE_V_LIM
)
m
.
addGenConstrTan
(
xvar
=
atan_K_cte_V
,
yvar
=
K_cte_V
)
...
...
@@ -197,10 +197,10 @@ class GEMStanleySymPyTeacher(SymPyTeacherBase):
# Variable Aliases
old_x
,
old_y
,
old_yaw
=
self
.
_old_state
new_x
,
new_y
,
new_yaw
=
self
.
_new_state
ct
d
,
psi
=
self
.
_percept
ct
e
,
psi
=
self
.
_percept
steering
,
=
self
.
_control
err
=
psi
+
sympy
.
atan2
(
ct
d
*
self
.
K_P
,
self
.
FORWARD_VEL
)
err
=
psi
+
sympy
.
atan2
(
ct
e
*
self
.
K_P
,
self
.
FORWARD_VEL
)
clipped_err
=
sympy
.
Piecewise
(
(
self
.
STEERING_LIM
,
err
>
self
.
STEERING_LIM
),
(
-
self
.
STEERING_LIM
,
err
<
-
self
.
STEERING_LIM
),
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment