lstm hidden reset

rc_car/camera.py

@@ -132,6 +132,8 @@ class AttentionCamera(BaseCamera):
         super(AttentionCamera, self).__init__(cam_path)
         self.det_frame = None
         self.agent = load_attention_agent(model_path, model)
+        if hasattr(self.agent, "reset"):
+            self.agent.reset()
 
     def get_frame(self):
         with self.frame_lock:

rc_car/policy_controller.py

@@ -140,6 +140,10 @@ class AttentionPolicyController(BasePolicyController):
 
     def drive(self):
         self.stop.value = False
+        if hasattr(self.agent, "reset"):
+            self.agent.reset()
+            self.cam.agent.reset()
+
         alpha = 0.9
         while True:
             start = time.time()