We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ed7a680 commit 925c76dCopy full SHA for 925c76d
src/main.py
@@ -64,6 +64,14 @@ def main():
64
current_obs = robot.get_observation()
65
frame = current_obs["front"]
66
if get_robot_status() == RobotStatus.FIND_BUCKET:
67
+ gripper_pos = current_obs.get('arm_gripper.pos', 5)
68
+ is_gripper_holding = gripper_pos > 25
69
+
70
+ if is_gripper_holding is False:
71
+ set_robot_status(RobotStatus.SEARCH)
72
+ reset_robot()
73
+ continue
74
75
result = get_red_bucket_local(frame) # 找桶的算法
76
else:
77
result = yolo_infer(frame) # 找球的算法
0 commit comments