一、所用工具
1、芯片:STM32F407ZGT6
2、CubeMX
3、KEIL5
4、OpenMV
5、舵机
二、实现功能
利用由两个自由舵机组装而成的二维云台来控制OpenMV的位置,以实现追踪指定阈值色块的效果。
三、CubeMX配置
3.1 初始化配置
3.2 定时器配置(PWM波输出)
这里我使用TIM3的通道1和TIM4的通道1分别实现对两个舵机的控制,由于我购买的舵机所需要的频率为50Hz(如下图)所以下面的PWM频率均是按照50Hz配置的。
TIM4的配置同理。
3.3 串口通信配置(与OpenMV和电脑通信)
为了方便后续的调试,所以建议同时开启与OpenMV和电脑的通信,即打开两个串口。USART1是用来和电脑通信的,USART2是用来与OpenMV通信的。
USART2的配置同理。
四、OpenMV配置
这是OpenMV中寻找色块,并通过串口PA4,PA5发送中心点位置以及长度和宽度的代码,可以直接复制到OpenMV IDE中。需要注意的是,在OpenMV中UART_RX—P5 ------ UART_TX—P4。
import time
import sensor
import math
import image
import ustruct
from pyb import UART
uart = UART(3, 115200, timeout_char=200)
uart.init(115200, bits=8, parity=None, stop=1) # init with given parameters
threshold_index = 0 # 0 for red, 1 for green, 2 for blue
thresholds = [
(30, 100, 15, 127, 15, 127), # generic_red_thresholds
(30, 100, -64, -8, -32, 32), # generic_green_thresholds
(0, 30, 0, 64, -128, 0), # generic_blue_thresholds
(82, 100, 75, -49, -22, 31), # generic_white_thresholds
(21, 83, 32, 65, 31, 63),
]
sensor.reset()
sensor.set_pixformat(sensor.RGB565)
sensor.set_framesize(sensor.QVGA)# QVGA的中心坐标:160,120
sensor.skip_frames(time=2000) # 跳过2000毫秒的帧让相机图像在改变相机设置后稳定下来
sensor.set_auto_gain(False) # 必须关闭才能进行颜色跟踪
sensor.set_auto_whitebal(False) # 必须关闭才能进行颜色跟踪
clock = time.clock()
def find_max(blobs):
max_size=0
for blob in blobs:
if blob.pixels() > max_size:
max_blob = blob
max_size = blob.pixels()
return max_blob
def send_data(x,y,w,h):
global uart;
FH = bytearray([0xb3,0xb3]) # 帧头
uart.write(FH) # 写到串口
uart.write(str(x))
uart.write(bytearray([0x20])) # 发送空格
uart.write(str(y))
uart.write(bytearray([0x20]))
uart.write(str(w))
uart.write(bytearray([0x20]))
uart.write(str(h))
uart.write(bytearray([0x20]))
FH = bytearray([0x0d,0x0a]) # 帧尾,换行和回车的ascll
uart.write(FH)
while True:
clock.tick()
img = sensor.snapshot()
blobs = img.find_blobs([thresholds[threshold_index]])
#如果找到了目标颜色
if blobs:
max_blob = find_max(blobs)
cx=max_blob[5]
cy=max_blob[6]
cw=max_blob[2]
ch=max_blob[3]
# 这些值取决于max_blob不是圆形的,否则它们将不稳定.
# 检查max_blob是否显著偏离圆形
if max_blob.elongation() > 0.5:
img.draw_edges(max_blob.min_corners(), color=(255, 0, 0))
img.draw_line(max_blob.major_axis_line(), color=(0, 255, 0))
img.draw_line(max_blob.minor_axis_line(), color=(0, 0, 255))
# 这些值始终是稳定的。
# img.draw_rectangle(max_blob.rect())
img.draw_rectangle(160,120,35,35)
img.draw_cross(cx, cy)
# 注意-max_blob旋转仅限于0-180。
img.draw_keypoints(
[(cx, cy, int(math.degrees(max_blob.rotation())))], size=20
)
send_data(cx,cy,cw,ch) # 发送数据
print(cx,cy,cw,ch)
print(clock.fps())
五、KEIL代码修改
5.1 串口重定义
在usart.c的最后加入重定义代码。
/* USER CODE BEGIN 1 */
int fputc(int ch, FILE *f)
{
HAL_UART_Transmit(&huart1, (uint8_t *)&ch, 1, 0xffff);
return ch;
}
int fgetc(FILE *f)
{
uint8_t ch = 0;
HAL_UART_Receive(&huart1, &ch, 1, 0xffff);
return ch;
}
/* USER CODE END 1 */
在usart.h的开头加入include。
/* USER CODE BEGIN Includes */
#include <stdio.h>
/* USER CODE END Includes */
5.2 串口回调函数
用到的变量初始化,大家自行放到main.c中的相应位置。
#include "string.h"
#include "stdio.h"
#include "stdlib.h"
#define RXBUFFERSIZE 256
char RxBuffer[RXBUFFERSIZE],rx_buf[RXBUFFERSIZE];
uint8_t aRxBuffer;
uint8_t Uart1_Rx_Cnt = 0;
int flag=0;
printf("Hello World!\r\n");
HAL_Delay(200);
HAL_UART_Receive_IT(&huart2, (uint8_t *)&aRxBuffer, 1);
这段代码放在main.c中,这里为了方便确定OpenMV与STM32之间有数据传输,加入了翻转LED灯,大家可以有需要可以自己再配置一下LED,不用的话可自行删除。
/* USER CODE BEGIN 4 */
void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart)
{
UNUSED(huart);
if(huart==&huart2){
HAL_GPIO_TogglePin(GPIOF,LED_Pin); //有数据则翻转LED灯
RxBuffer[Uart1_Rx_Cnt] = aRxBuffer;
Uart1_Rx_Cnt++;
if((RxBuffer[Uart1_Rx_Cnt-1] == 0xb3)&&(RxBuffer[Uart1_Rx_Cnt-2] == 0xb3)) flag=1; //帧头判定
else if((RxBuffer[Uart1_Rx_Cnt-2] == 0x0d)&&(RxBuffer[Uart1_Rx_Cnt-1] == 0x0a)) flag=2; //帧尾判定
else flag=0;
switch (flag)
{
case 1:
Uart1_Rx_Cnt = 0;
memset(RxBuffer,0x00,sizeof(RxBuffer));
break;
case 2:
RxBuffer[Uart1_Rx_Cnt-1] = '\0';
RxBuffer[Uart1_Rx_Cnt-2] = '\0';
strcpy(rx_buf,RxBuffer);
printf("%s\r\n",rx_buf);
while(HAL_UART_GetState(&huart2) == HAL_UART_STATE_BUSY_TX);
Uart1_Rx_Cnt = 0;
memset(RxBuffer,0x00,sizeof(RxBuffer));
break;
default:break;
}
HAL_UART_Receive_IT(&huart2, (uint8_t *)&aRxBuffer, 1);
}
}
/* USER CODE END 4 */
5.3 加入pid控制文件
5.3.1 pid.c
由于两个舵机分别控制横向和纵向的移动,所以此处定义了两个句柄和两个pid函数,分别对应两个舵机,大家可以自行调试其中Kp,Ki,Kd的值。
#include "pid.h"
pid_typedef pid1;
pid_typedef pid2;
void PID_init(void)
{
pid1.SetPosition=0;
pid1.ActualPosition=0.0;
pid1.err=0.0;
pid1.err_last=0.0;
pid1.out=0.0;
pid1.integral=0.0;
pid1.Kp=0.025;
pid1.Ki=0;
pid1.Kd=0.017;
pid2.SetPosition=0;
pid2.ActualPosition=0.0;
pid2.err=0.0;
pid2.err_last=0.0;
pid2.out=0.0;
pid2.integral=0.0;
pid2.Kp=0.025;
pid2.Ki=0;
pid2.Kd=0.017;
}
float PIDx_realize(float ActualPosition,float SetPosition)
{
pid1.ActualPosition=ActualPosition;
pid1.SetPosition=SetPosition;
pid1.err=pid1.SetPosition-pid1.ActualPosition;
pid1.integral+=pid1.err;
pid1.out=pid1.Kp*pid1.err+pid1.Ki*pid1.integral+pid1.Kd*(pid1.err-pid1.err_last);
pid1.err_last=pid1.err;
return pid1.out;
}
float PIDy_realize(float ActualPosition,float SetPosition)
{
pid2.ActualPosition=ActualPosition;
pid2.SetPosition=SetPosition;
pid2.err=pid2.ActualPosition-pid2.SetPosition;
pid2.integral+=pid2.err;
pid2.out=pid2.Kp*pid2.err+pid2.Ki*pid2.integral+pid2.Kd*(pid2.err-pid2.err_last);
pid2.err_last=pid2.err;
return pid2.out;
}
5.3.2 pid.h
#ifndef __PID_H
#define __PID_H
#include "stm32f4xx.h"
typedef struct
{
float SetPosition;//设定值
float ActualPosition;//实际值
float err;
float err_last;
float Kp;
float Ki;
float Kd;
float out;//执行器的变量
float integral;//积分值
}pid_typedef;
void PID_init(void);
float PIDx_realize(float ActualPosition,float SetPosition);
float PIDy_realize(float ActualPosition,float SetPosition);
#endif
添加好文件之后不要忘记添加pid.h的目录 。
5.4 实现对PWM波占空比的控制
放置在前面的初始化代码。
#include "pid.h"
double motor1=25;
double motor2=35;
int cx,cy;
PID_init();
HAL_Delay(100) ;
HAL_TIM_PWM_Start(&htim3, TIM_CHANNEL_1);//开启PWM波
HAL_TIM_PWM_Start(&htim4, TIM_CHANNEL_1);
__HAL_TIM_SET_COMPARE(&htim3, TIM_CHANNEL_1, motor1);//占空比初始化,为25\1000=2.5%
__HAL_TIM_SET_COMPARE(&htim4, TIM_CHANNEL_1, motor2);
放在中断回调函数中的代码,可以选择放在case2中的printf下面,因为在OpenMV中设定的QVGA的中心坐标为160,120,所以下列的设定横纵坐标分别为160和120。其中舵机2我设定的占空比在3.5%和6%之间是因为我的装置结构问题,如果上面的电机往下太多会导致数据线与底座接触,造成电机卡住,大家可以根据自己的装配情况自行更改占空比的上下范围。
sscanf(rx_buf, "%d %d", &cx, &cy);//提取rx_buf中的前两个数字,即为横纵坐标
//对下面对应横坐标的舵机进行控制
motor1=motor1+PIDx_realize(cx,160);//第一参数为实际坐标,第二个参数为设定坐标
if(motor1<25)motor1=25;//防止超过舵机的工作范围内的占空比
if(motor1>125)motor1=125;
TIM3->CCR1=motor1;
//对上面对应纵坐标的舵机进行控制
motor2=motor2+PIDy_realize(cy,120);
if(motor2<35)motor2=35;
if(motor2>60)motor2=60;
TIM4->CCR1=motor2;
六、成果展示
由于实验室器材有限,无奈拿了一个开学典礼的灯作为底座,所以在舵机运行起来时会不稳,建议大家把舵机固定在重一点且高一点的东西上,可以避免OpenMV的线被卡住。总体来说,摄像头的跟踪效果还是不错的。
色块跟踪
七、结语
大家如果想要购买我的同款舵机,可以私信我,但是这个舵机的装配比较麻烦,需要自己用工具把材料修剪到合适的大小。然后这是我写的第一份博客,内容可能有不足的地方,大家都可以指出,最后为大家分享几篇我在做的过程中参考的文章,其中第一篇来自我同实验室的同学。