用Jetson Nano的串口给STM32F4‘下命令’：打造一个简单的边缘AI控制节点
Jetson Nano与STM32F4的串口通信：构建边缘AI控制系统的实践指南
在智能硬件开发领域，将AI推理能力与实时控制相结合的需求日益增长。想象一下，当你的摄像头识别到特定手势时，机械臂立即做出响应；或者当语音识别模块捕捉到指令时，智能小车马上调整方向——这些场景都需要一个可靠的通信桥梁，连接负责AI计算的"大脑"和执行具体动作的"四肢"。本文将带你深入探索如何利用Jetson Nano的串口与STM32F4构建这样一个高效、可靠的控制系统。
1. 系统架构设计
边缘AI控制系统通常由三个核心部分组成：感知层、决策层和执行层。在我们的方案中，Jetson Nano充当决策中枢，负责运行YOLO等AI模型进行实时分析；STM32F4则作为执行单元，控制电机、舵机等物理设备。
为什么选择串口通信？
- 硬件资源占用少：相比以太网或USB，串口不需要复杂的协议栈
- 实时性高：适合传输简单的控制指令
- 开发简单：大多数微控制器都内置硬件UART模块
- 可靠性好：在短距离通信中表现稳定
典型的系统连接方式如下：
| 设备 | 角色 | 接口配置 |
|---|---|---|
| Jetson Nano | 主控制器 | /dev/ttyTHS1 |
| STM32F4 | 从设备 | UART2 |
| USB转串口 | 调试接口 | 连接STM32的UART1 |
2. 通信协议设计
一个健壮的通信协议需要考虑指令格式、校验机制和错误处理。我们设计了一种基于字符串的轻量级协议，具有以下特点：
- 人类可读：采用明文指令，便于调试
- 结构化：使用键值对形式组织参数
- 可扩展：支持未来添加新指令类型
指令格式示例
MOTOR:LEFT,SPEED:80,DURATION:1000对应的Python生成代码：
def generate_motor_command(side, speed, duration_ms): return f"MOTOR:{side.upper()},SPEED:{speed},DURATION:{duration_ms}\r\n"校验机制实现
为提高可靠性，可以在指令末尾添加简单的校验和：
def add_checksum(command): checksum = sum(ord(c) for c in command) % 256 return f"{command}CHECKSUM:{checksum}\r\n"3. Jetson Nano端的实现
Jetson Nano作为系统的大脑，需要完成AI推理和指令生成两大任务。我们使用Python的pyserial库进行串口通信。
基础通信设置
import serial class JetsonSerialController: def __init__(self, port='/dev/ttyTHS1', baudrate=115200): self.serial = serial.Serial( port=port, baudrate=baudrate, timeout=1, write_timeout=1 ) def send_command(self, command): try: self.serial.write(command.encode('utf-8')) self.serial.flush() except serial.SerialTimeoutException: print("发送超时，请检查连接") except serial.SerialException as e: print(f"串口错误: {e}") def close(self): self.serial.close()与AI模型集成示例
假设我们有一个简单的物体检测系统，检测到特定物体时控制电机：
import cv2 from yolov5 import detect # 假设使用YOLOv5 def ai_control_loop(): ser = JetsonSerialController() cap = cv2.VideoCapture(0) while True: ret, frame = cap.read() if not ret: break results = detect.detect(frame) # 获取检测结果 for obj in results: if obj['class'] == 'person' and obj['confidence'] > 0.7: x_center = obj['x'] + obj['width']/2 if x_center < frame.shape[1]/3: ser.send_command(generate_motor_command("left", 70, 500)) elif x_center > 2*frame.shape[1]/3: ser.send_command(generate_motor_command("right", 70, 500)) cap.release() ser.close()4. STM32F4端的实现
STM32端需要可靠地解析来自Jetson Nano的指令，并将其转换为具体的硬件操作。我们使用HAL库开发，采用状态机的方式处理指令。
串口接收配置
在STM32CubeIDE中配置UART2：
- 波特率：115200
- 字长：8位
- 停止位：1位
- 无校验
- 启用接收中断
指令解析状态机
typedef enum { WAIT_START, READ_KEY, READ_VALUE, CHECK_END } ParserState; typedef struct { char key[20]; char value[20]; uint8_t key_index; uint8_t value_index; ParserState state; } CommandParser; void parse_command_char(CommandParser* parser, char c) { switch(parser->state) { case WAIT_START: if(isalpha(c) || c == ':') { parser->state = READ_KEY; parser->key_index = 0; memset(parser->key, 0, sizeof(parser->key)); } break; case READ_KEY: if(c == ':') { parser->state = READ_VALUE; parser->value_index = 0; memset(parser->value, 0, sizeof(parser->value)); } else if(isalnum(c)) { parser->key[parser->key_index++] = c; } break; case READ_VALUE: if(c == ',' || c == '\r') { parser->state = (c == ',') ? READ_KEY : CHECK_END; // 处理键值对 execute_key_value(parser->key, parser->value); } else { parser->value[parser->value_index++] = c; } break; case CHECK_END: if(c == '\n') { // 完整指令处理完成 parser->state = WAIT_START; } break; } }电机控制实现
void execute_key_value(const char* key, const char* value) { if(strcmp(key, "MOTOR") == 0) { if(strcmp(value, "LEFT") == 0) { HAL_GPIO_WritePin(MOTOR_LEFT_GPIO_Port, MOTOR_LEFT_Pin, GPIO_PIN_SET); } else if(strcmp(value, "RIGHT") == 0) { HAL_GPIO_WritePin(MOTOR_RIGHT_GPIO_Port, MOTOR_RIGHT_Pin, GPIO_PIN_SET); } } else if(strcmp(key, "SPEED") == 0) { int speed = atoi(value); set_motor_speed(speed); } }5. 系统优化与调试技巧
构建稳定可靠的通信系统需要考虑许多细节。以下是一些实用技巧：
提高通信可靠性
超时重传机制：
- Jetson端发送指令后等待ACK
- 如果500ms内未收到响应，重发指令
- 连续3次失败后报错
数据完整性检查：
- 除了校验和外，可以添加CRC校验
- 在STM32端验证指令格式
调试工具推荐
Jetson端：
screen或minicom：直接与串口交互cutecom：图形化串口工具- Python脚本：灵活测试各种指令
STM32端：
- 逻辑分析仪：捕捉实际传输波形
- STM32CubeMonitor：实时查看变量
- 串口打印调试信息（通过UART1）
性能优化建议
指令压缩：
- 对于高频指令，可以使用简写形式
- 例如："MOTOR:L,SPD:80,DUR:1000"
批量发送：
- 合并多个相关指令一次性发送
- 减少通信开销
异步处理：
- Jetson端使用独立线程处理串口通信
- 避免阻塞AI推理主循环
6. 实际应用案例
让我们看一个具体的智能小车控制案例，展示如何将上述技术应用于实际项目。
场景描述：
- Jetson Nano运行YOLOv5模型，实时分析摄像头画面
- 检测到行人时，控制小车转向
- 检测到停止标志时，小车刹车
系统工作流程
初始化阶段：
- Jetson启动串口连接
- STM32初始化PWM和GPIO
- 双方进行握手确认
运行阶段：
- Jetson每100ms发送一次控制指令
- STM32实时响应并反馈状态
- 异常情况下进入安全模式
紧急处理：
- 通信中断超过1秒时，STM32自动停车
- 收到"EMERGENCY_STOP"指令立即执行
关键代码片段
Jetson端的控制逻辑：
def process_detection(results): has_person = any(obj['class'] == 'person' for obj in results) has_stop = any(obj['class'] == 'stop_sign' for obj in results) if has_stop: return "MOTOR:STOP\r\n" elif has_person: person = next(obj for obj in results if obj['class'] == 'person') if person['x'] < 320: # 假设图像宽度为640 return "MOTOR:LEFT,ANGLE:30\r\n" else: return "MOTOR:RIGHT,ANGLE:30\r\n" else: return "MOTOR:FORWARD,SPEED:50\r\n"STM32端的PWM控制实现：
void set_motor_speed(uint8_t speed) { if(speed == 0) { HAL_GPIO_WritePin(BRAKE_GPIO_Port, BRAKE_Pin, GPIO_PIN_SET); } else { HAL_GPIO_WritePin(BRAKE_GPIO_Port, BRAKE_Pin, GPIO_PIN_RESET); uint16_t pulse = (uint16_t)(speed / 100.0 * __HAL_TIM_GET_AUTORELOAD(&htim3)); __HAL_TIM_SET_COMPARE(&htim3, TIM_CHANNEL_1, pulse); } }7. 进阶主题与扩展思路
当基本系统运行稳定后，可以考虑以下进阶优化：
协议升级方案
二进制协议：
- 定义紧凑的二进制格式
- 显著提高传输效率
- 示例结构：
#pragma pack(push, 1) typedef struct { uint8_t command_type; uint16_t param1; uint16_t param2; uint8_t checksum; } MotorCommand; #pragma pack(pop)
协议版本控制：
- 在指令中包含版本号
- 兼容新旧设备
多设备组网
总线拓扑：
- 多个STM32共享同一串口总线
- 每个设备有唯一地址
- 指令中包含目标地址
中继节点：
- STM32可以转发指令给其他设备
- 构建更复杂的控制系统
性能监控与日志
状态反馈：
- STM32定期发送状态报告
- 包括温度、电压、错误码等
日志记录：
- Jetson保存关键指令日志
- 便于事后分析
class CommandLogger: def __init__(self, filename='commands.log'): self.file = open(filename, 'a') def log(self, command, direction='TX'): timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f') self.file.write(f"[{timestamp}] {direction}: {command.strip()}\n") self.file.flush() def close(self): self.file.close()在项目开发过程中，我遇到过一个典型的串口通信问题：当Jetson Nano同时进行大量AI计算时，串口发送会出现延迟。解决方案是使用单独的线程处理串口通信，并通过队列传递指令，这样可以避免主线程被阻塞。另一个实用技巧是在STM32端实现环形缓冲区存储接收到的数据，即使处理速度暂时跟不上，也不会丢失指令。