debug: 修复episode首帧图像不正确的问题；修复前2个episode帧重复的问题

2026-02-26 16:17:54 +08:00
parent 40c40695dd
commit 4e0add4e1d
3 changed files with 12 additions and 1 deletions
--- a/roboimi/envs/double_pos_ctrl_env.py
+++ b/roboimi/envs/double_pos_ctrl_env.py
@@ -72,6 +72,10 @@ class DualDianaMed_Pos_Ctrl(DualDianaMed):
        self.mj_data.joint('red_box_joint').qpos[5] = 0.0
        self.mj_data.joint('red_box_joint').qpos[6] = 0.0
        super().reset()
+        self.top = None
+        self.angle = None
+        self.r_vis = None
+        self.front = None
        self.cam_flage = True
        t=0
        while self.cam_flage: