Skip to content

Commit 3299d8e

Browse files
Formatted Q matrix in obstacle class
Changed global transform and infering orientation from velocity flags to true
1 parent 5af5d49 commit 3299d8e

File tree

2 files changed

+9
-54
lines changed

2 files changed

+9
-54
lines changed

kf_hungarian_tracker/kf_hungarian_tracker/kf_hungarian_node.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,8 @@ def __init__(self):
4949
("vel_filter", [0.1, 2.0]),
5050
("height_filter", [-2.0, 2.0]),
5151
("cost_filter", 1.0),
52-
("transform_to_global_frame", False),
53-
("infer_orientation_from_velocity", False),
52+
("transform_to_global_frame", True),
53+
("infer_orientation_from_velocity", True),
5454
],
5555
)
5656
self.global_frame = self.get_parameter("global_frame")._value

kf_hungarian_tracker/kf_hungarian_tracker/obstacle_class.py

Lines changed: 7 additions & 52 deletions
Original file line numberDiff line numberDiff line change
@@ -103,58 +103,13 @@ def predict(self, dt):
103103
dt2 = dt**2
104104
dt3 = dt * dt2
105105
dt4 = dt2**2
106-
Q = np.array(
107-
[
108-
[
109-
dt4 * self.process_noise_cov[0] / 4,
110-
0,
111-
0,
112-
dt3 * self.process_noise_cov[0] / 2,
113-
0,
114-
0,
115-
],
116-
[
117-
0,
118-
dt4 * self.process_noise_cov[1] / 4,
119-
0,
120-
0,
121-
dt3 * self.process_noise_cov[1] / 2,
122-
0,
123-
],
124-
[
125-
0,
126-
0,
127-
dt4 * self.process_noise_cov[2] / 4,
128-
0,
129-
0,
130-
dt3 * self.process_noise_cov[2] / 2,
131-
],
132-
[
133-
dt3 * self.process_noise_cov[0] / 2,
134-
0,
135-
0,
136-
dt2 * self.process_noise_cov[0],
137-
0,
138-
0,
139-
],
140-
[
141-
0,
142-
dt3 * self.process_noise_cov[1] / 2,
143-
0,
144-
0,
145-
dt2 * self.process_noise_cov[1],
146-
0,
147-
],
148-
[
149-
0,
150-
0,
151-
dt3 * self.process_noise_cov[2] / 2,
152-
0,
153-
0,
154-
dt2 * self.process_noise_cov[2],
155-
],
156-
]
157-
).astype(np.float32)
106+
107+
Q = np.array([[dt4*self.process_noise_cov[0]/4, 0, 0, dt3*self.process_noise_cov[0]/2, 0, 0],
108+
[0, dt4*self.process_noise_cov[1]/4, 0, 0, dt3*self.process_noise_cov[1]/2, 0],
109+
[0, 0, dt4*self.process_noise_cov[2]/4, 0, 0, dt3*self.process_noise_cov[2]/2],
110+
[dt3*self.process_noise_cov[0]/2, 0, 0, dt2*self.process_noise_cov[0], 0, 0],
111+
[0, dt3*self.process_noise_cov[1]/2, 0, 0, dt2*self.process_noise_cov[1], 0],
112+
[0, 0, dt3*self.process_noise_cov[2]/2, 0, 0, dt2*self.process_noise_cov[2]]]).astype(np.float32)
158113

159114
self.kalman.transitionMatrix = F
160115
self.kalman.processNoiseCov = Q

0 commit comments

Comments
 (0)