机械臂控制逻辑入口
“dofbot_color_stacking/scripts/main.py”与“dofbot_garbage_yolov5/dofbot_garbage_yolov5/main.py”均为机械臂的运行控制入口,启动后调用各个功能包模块代码,实现机械臂的分拣与堆叠。核心代码片段示例如下。
# 当摄像头正常打开的情况下循环执行
while capture.isOpened():
# 读取相机的每一帧
ret, img = capture.read()
print("read image from camera successfully:", ret)
# 统一图像大小
img = cv.resize(img, (640, 480))
dp = np.fromfile(dp_cfg_path, dtype=np.int32)
if DP_PRINT:
print("dp has dtype:", dp.dtype)
print("dp has shape:", dp.shape)
dp = dp.reshape(4,2)
if DP_PRINT:
print("After reshape, dp has shape:", dp.shape)
print("dp now is:", dp)
img = calibration.Perspective_transform(dp, img)
img, msg = target.garbage_run(img)
print("Model is warming up at stage:", warm_up_count)
if warm_up_count != 0 and last_num == warm_up_count:
last_count += 1
if last_count > 5:
warm_up_count = 0
last_count = 0
last_num = warm_up_count
父主题: 代码实现