控制小车视觉循迹使用 OpenMV 往往是不够的。一般使用 OpenMV 对图像进行处理,将处理过后的数据使用串口发送给STM32,使用STM32控制小车行驶。本文主要讲解 OpenMV 模块与 STM32 间的串口通信以及两种循迹方案,分别是划分检测区域和线性回归。
一、OpenMV模块与STM32间的串口通信
1、硬件连接
OpenMV端:UART_RX---P5、UART_TX---P4
STM32端:USART_TX---PA9 、USART_RX---PA10
STM32的TX接OpenMV的RX、STM32的RX接OpenMV的TX。
2、OpenMV 端串口发送数据
uart = UART(3,115200,bits=8, parity=None, stop=1, timeout_char = 1000)
FH = bytearray([0x2C,0x12,cx,cy,ch,ci,0x5B])
uart.write(FH)
波特率为115200
0x2C、0x12为帧头,0x5B为桢尾
bytearray([ ])函数:用于把十六进制数据以字节形式存放到字节数组中,以便以数据帧的形式发送出去进行通信。
也可打包为函数形式:
def sending_data(cx,cy,cw,ch):
global uart;
data = ustruct.pack("<bbhhhhb",
0x2C,
0x12,
int(cx),
int(cy),
int(ch),
int(ci),
0x5B)
uart.write(data); #必须要传入一个字节数组
sending_data(cx,cy,ch,ci)
3、STM32串口以中断方式接收数据
static u8 Cx=0,Cy=0,Ch=0,Ci=0;
//USART1 全局中断服务函数
void USART1_IRQHandler(void)
{
u8 com_data;
u8 i;
static u8 RxCounter1=0;
static u16 RxBuffer1[10]={0};
static u8 RxState = 0;
static u8 RxFlag1 = 0;
if( USART_GetITStatus(USART1,USART_IT_RXNE)!=RESET) //接收中断
{
USART_ClearITPendingBit(USART1,USART_IT_RXNE); //清除中断标志
com_data = USART_ReceiveData(USART1);
if(RxState==0&&com_data==0x2C) //0x2c帧头
{
RxState=1;
RxBuffer1[RxCounter1++]=com_data;
OLED_Refresh();
}
else if(RxState==1&&com_data==0x12) //0x12帧头
{
RxState=2;
RxBuffer1[RxCounter1++]=com_data;
}
else if(RxState==2)
{
RxBuffer1[RxCounter1++]=com_data;
if(RxCounter1>=10||com_data == 0x5B) //RxBuffer1接受满了,接收数据结束
{
RxState=3; RxFlag1=1;
Cx=RxBuffer1[RxCounter1-5];
Cy=RxBuffer1[RxCounter1-4];
Ch=RxBuffer1[RxCounter1-3];
Ci=RxBuffer1[RxCounter1-2];
}
}
else if(RxState==3) //检测是否接受到结束标志
{
if(RxBuffer1[RxCounter1-1] == 0x5B)
{
USART_ITConfig(USART1,USART_IT_RXNE,DISABLE);//关闭DTSABLE中断
RxFlag1 = 0; RxCounter1 = 0; RxState = 0;
USART_ITConfig(USART1,USART_IT_RXNE,ENABLE);
}
else //接收错误
{
RxState = 0; RxCounter1=0;
for(i=0;i<10;i++)
{
RxBuffer1[i]=0x00; //将存放数据数组清零
}
}
}
else //接收异常
{
RxState = 0; RxCounter1=0;
for(i=0;i<10;i++)
{
RxBuffer1[i]=0x00; //将存放数据数组清零
}
}
}
}
二、视觉循迹方案一:划分检测区域
该方法和一般循迹模块的原理相同。使用五个红框对图像进行分割,对应五路循迹模块的五个输出。
为提高识别精度,将图像二值化后再进行黑线检测。(若要识别其他颜色的线则需要使用常规的设置阈值法)
import pyb, sensor, image, math, time
from pyb import UART
import ustruct
from image import SEARCH_EX, SEARCH_DS
sensor.set_contrast(1)
sensor.set_gainceiling(16)
clock = time.clock()
uart = UART(3,115200,bits=8, parity=None, stop=1, timeout_char = 1000)
# 划分五个区域
roi1 = [(0, 40, 20, 40), # x y w h 分别对应x、y坐标,框的宽度、高度
(35, 40, 20, 40),
(70, 40, 10, 10),
(105, 40, 20, 40),
(140, 40, 20, 40)]
sensor.reset()
sensor.set_pixformat(sensor.RGB565)
sensor.set_framesize(sensor.QQVGA)
sensor.skip_frames(time=2000)
sensor.set_auto_whitebal(True)
sensor.set_auto_gain(False)
sensor.set_vflip(False)
sensor.set_hmirror(False)
clock = time.clock()
GRAYSCALE_THRESHOLD = [(20,100)]
low_threshold = (0, 50)
while(True):
clock.tick()
blob1=None
blob2=None
blob4=None
blob5=None
cx=cy=cw=i=0
img = sensor.snapshot().lens_corr(strength = 1.7 , zoom = 1.0)
img.mean(2)
binary_img = img.binary(GRAYSCALE_THRESHOLD) # 二值化
blob1 = binary_img.find_blobs([low_threshold],roi=roi1[0])
blob2 = binary_img.find_blobs([low_threshold],roi=roi1[1])
blob3 = binary_img.find_blobs([low_threshold],roi=roi1[2])
blob4 = binary_img.find_blobs([low_threshold],roi=roi1[3])
blob5 = binary_img.find_blobs([low_threshold],roi=roi1[4])
if blob1:
cx = 1
if blob2:
cy = 1
if blob3:
cw = 1
if blob4:
ch = 1
if blob5:
ci = 1
FH = bytearray([0x2C,0x12,cx,cy,ch,ci,0x5B])
uart.write(FH)
for rec in roi1:
img.draw_rectangle(rec, color=(255,0,0))#绘制出roi区域
实验效果:
三、视觉循迹方案二:线性回归
将控制线的偏移距离与角度偏差作为PID控制对象,分别计算其控制输出,最后相加。
下面直接附上详细注释版的代码:
THRESHOLD = (21, 0, -77, 5, -110, 127)
#定义一个阈值元组用于图像二值化处理。这个阈值将用于将图像转换为二进制图像,用于线检测。
import sensor, image, time
#导入了用于图像处理的sensor、image模块,以及用于时间相关操作的time模块。
from pyb import LED
from pid import PID #从pid模块中导入了PID类,用于实现PID控制算法。
import time
from pyb import UART #从pyb模块中导入了UART类,用于串口通信。
import math
rho_pid = PID(p=0.37, i=0) #创建一个PID对象rho_pid,用于控制线的偏移距离。
theta_pid = PID(p=0.001, i=0) #创建一个PID对象theta_pid,用于控制线的角度偏差。
#rho 是从图像原点到直线的垂直距离, theta 是直线与垂直轴之间的角度。
LED(1).on()
LED(2).on()
LED(3).on()
uart = UART(3,19200) # 创建一个UART对象uart,用于串口通信,波特率为19200。
sensor.reset() # 重置图像传感器
sensor.set_vflip(True) # 设置图像传感器垂直翻转,即上下颠倒。
sensor.set_hmirror(True) # 设置图像传感器水平镜像,即左右翻转。
sensor.set_pixformat(sensor.RGB565) # 设置图像传感器的像素格式为RGB565。
sensor.set_framesize(sensor.QQQVGA) # 设置图像传感器的帧大小为QQQVGA。
sensor.skip_frames(time = 2000) # 跳过2000毫秒的图像帧,使传感器稳定。
clock = time.clock() # 创建一个时间对象clock,用于计时。
while(True):
clock.tick() #记录当前时间。
img = sensor.snapshot().binary([THRESHOLD])
#获取图像的快照,并将其转换为二值图像,根据之前定义的阈值进行二值化处理。
line = img.get_regression([(100,100)], robust = True)
#在二值图像中检测线段,返回一个线段对象line。
#参数[(100,100)]表示检测线段时使用的区域,robust=True表示使用鲁棒回归算法进行线段拟合。
if (line): #如果检测到了线段。
rho_err = abs(line.rho())-img.width()/2
#计算线段的偏移距离,即线段的rho值减去图像宽度的一半。
if line.theta()>90: #如果线段的角度大于90度。
theta_err = line.theta()-180 #计算线段的角度偏差,即线段的角度减去180度。
else:
theta_err = line.theta() # 小于等于90度 直接将线段的角度作为角度偏差。
img.draw_line(line.line(), color = 127) # 在图像上绘制检测到的线段。
print(rho_err,line.magnitude(),rho_err) # 打印线段的偏移距离、线段的长度和偏移距离。
if line.magnitude()>8: # 如果线段的长度大于8。
rho_output = rho_pid.get_pid(rho_err,1)
#使用PID控制算法计算线段偏移距离的控制输出。
theta_output = theta_pid.get_pid(theta_err,1)
#使用PID控制算法计算线段角度偏差的控制输出。
output = rho_output + theta_output #将线段偏移距离和角度偏差的控制输出相加。
if(output<0):
#控制输出小于0,STM32不好处理负数,所以要将计算出的负数取绝对值+100
#stm32通过判断其值是否大于100来判断其是否为负数。
output = abs(output) + 100 #将控制输出取绝对值并加上100。
OUTPUT = str(round(output))
#将控制输出四舍五入转换为字符串。因为UART的write()只接受字符串作为参数,不接受数值类型
uart.write(OUTPUT) #将控制输出通过串口发送。
uart.write('\r\n') #发送回车换行符。
print(OUTPUT) #打印控制输出。
pass
pid.py 文件见OpenMV官方源码:https://book.openmv.cc/project/follow-lines.html
实验效果: