Skip to content

Commit f6c3e25

Browse files
Merge pull request #440 from JdeRobot/issue-439
Fixed brain for memory based models with 3-image input and previous V
2 parents 964870d + d03c507 commit f6c3e25

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

behavior_metrics/brains/CARLA/brain_carla_bird_eye_deep_learning_x3_previous_v.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -160,13 +160,12 @@ def execute(self):
160160
self.image_1 = self.image_2
161161
self.image_2 = self.image_1
162162
self.image_3 = img
163-
163+
164164
velocity_dim = np.full((150, 50), self.previous_speed/30)
165-
self.image_1 = np.dstack((self.image_1, velocity_dim))
166-
self.image_2 = np.dstack((self.image_2, velocity_dim))
167-
self.image_3 = np.dstack((self.image_3, velocity_dim))
168-
169-
img = [self.image_3, self.image_2 , self.image_1]
165+
image_1 = np.dstack((self.image_1, velocity_dim))
166+
image_2 = np.dstack((self.image_2, velocity_dim))
167+
image_3 = np.dstack((self.image_3, velocity_dim))
168+
img = [image_3, image_2 , image_1]
170169

171170
img = np.expand_dims(img, axis=0)
172171
start_time = time.time()
@@ -179,7 +178,6 @@ def execute(self):
179178
speed = self.vehicle.get_velocity()
180179
vehicle_speed = 3.6 * math.sqrt(speed.x**2 + speed.y**2 + speed.z**2)
181180
self.previous_speed = vehicle_speed
182-
183181
if vehicle_speed > 300:
184182
self.motors.sendThrottle(0)
185183
self.motors.sendSteer(steer)

0 commit comments

Comments
 (0)