本文是一些机器人算法（特别是自动导航算法）的Python代码合集。

其主要特点有以下三点：选择了在实践中广泛应用的算法；依赖最少；容易阅读，容易理解每个算法的基本思想。希望阅读本文后能对你有所帮助。

前排友情提示，文章较长，建议收藏后再看，项目地址在文末，记得点个“好看”哦。

目录

一、环境需求

二、怎样使用

三、本地化

    3.1 扩展卡尔曼滤波本地化

    3.2 无损卡尔曼滤波本地化

    3.3 粒子滤波本地化

    3.4 直方图滤波本地化

四、映射

    4.1 高斯网格映射

    4.2 光线投射网格映射

    4.3 k均值物体聚类

    4.4 圆形拟合物体形状识别

五、SLAM

    5.1 迭代最近点匹配

    5.2 EKF SLAM

    5.3 FastSLAM 1.0

    5.4 FastSLAM 2.0

    5.5 基于图的SLAM

六、路径规划

    6.1 动态窗口方式

    6.2 基于网格的搜索

• 迪杰斯特拉算法

• A*算法

• 势场算法

    6.3 模型预测路径生成

• 路径优化示例

• 查找表生成示例

    6.4 状态晶格规划

• 均匀极性采样（Uniform polar sampling）

• 偏差极性采样（Biased polar sampling）

• 路线采样（Lane sampling）

    6.5 随机路径图（PRM）规划

    6.6 Voronoi路径图规划

    6.7 快速搜索随机树（RRT）

• 基本RRT

• RRT*

• 基于Dubins路径的RRT

• 基于Dubins路径的RRT*

• 基于reeds-shepp路径的RRT*

• Informed RRT*

• 批量Informed RRT*

• 闭合回路RRT*

• LQR-RRT*

    6.8 三次样条规划

    6.9 B样条规划

    6.10 Eta^3样条路径规划

    6.11 贝济埃路径规划

    6.12 五次多项式规划

    6.13 Dubins路径规划

    6.14 Reeds Shepp路径规划

    6.15 基于LQR的路径规划

    6.16 Frenet Frame中的最优路径

七、路径跟踪

    7.1 姿势控制跟踪

    7.2 纯追迹跟踪

    7.3 史坦利控制

    7.4 后轮反馈控制

    7.5 线性二次regulator（LQR）转向控制

    7.6 线性二次regulator（LQR）转向和速度控制

    7.7 模型预测速度和转向控制

八、项目支持

一、环境需求

• Python 3.6.x

• numpy

• scipy

• matplotlib

• pandas

• cvxpy 0.4.x

二、怎样使用

1. 安装必要的库；

2. 克隆本代码仓库；

3. 执行每个目录下的python脚本；

4. 如果你喜欢，则收藏本代码库：）

三、本地化

3.1 扩展卡尔曼滤波本地化

该算法利用扩展卡尔曼滤波器（Extended Kalman Filter, EKF）实现传感器混合本地化。

蓝线为真实路径，黑线为导航推测路径（dead reckoning trajectory），绿点为位置观测（如GPS），红线为EKF估算的路径。

红色椭圆为EKF估算的协方差。

3.2 无损卡尔曼滤波本地化

该算法利用无损卡尔曼滤波器（Unscented Kalman Filter, UKF）实现传感器混合本地化。

线和点的含义与EKF模拟的例子相同。

3.3 粒子滤波本地化

该算法利用粒子滤波器（Particle Filter, PF）实现传感器混合本地化。

蓝线为真实路径，黑线为导航推测路径（dead reckoning trajectory），绿点为位置观测（如GPS），红线为PF估算的路径。

该算法假设机器人能够测量与地标（RFID）之间的距离。

PF本地化会用到该测量结果。

3.4 直方图滤波本地化

该算法是利用直方图滤波器（Histogram filter）实现二维本地化的例子。

红十字是实际位置，黑点是RFID的位置。

蓝色格子是直方图滤波器的概率位置。

在该模拟中，x，y是未知数，yaw已知。

滤波器整合了速度输入和从RFID获得距离观测数据进行本地化。

不需要初始位置。

四、映射

4.1 高斯网格映射

本算法是二维高斯网格映射（Gaussian grid mapping）的例子。

4.2 光线投射网格映射

本算法是二维光线投射网格映射（Ray casting grid map）的例子。

4.3 k均值物体聚类

本算法是使用k均值算法进行二维物体聚类的例子。

4.4 圆形拟合物体形状识别

本算法是使用圆形拟合进行物体形状识别的例子。

蓝圈是实际的物体形状。

红叉是通过距离传感器观测到的点。

红圈是使用圆形拟合估计的物体形状。

五、SLAM

同时本地化和映射（Simultaneous Localization and Mapping，SLAM）的例子。

5.1 迭代最近点匹配

本算法是使用单值解构进行二维迭代最近点（Iterative Closest Point，ICP）匹配的例子。

它能计算从一些点到另一些点的旋转矩阵和平移矩阵。

5.2 EKF SLAM

这是基于扩展卡尔曼滤波的SLAM示例。

蓝线是真实路径，黑线是导航推测路径，红线是EKF SLAM估计的路径。

绿叉是估计的地标。

5.3 FastSLAM 1.0

这是用FastSLAM 1.0进行基于特征的SLAM的示例。

蓝线是实际路径，黑线是导航推测，红线是FastSLAM的推测路径。

红点是FastSLAM中的粒子。

黑点是地标，蓝叉是FastLSAM估算的地标位置。

5.4 FastSLAM 2.0

这是用FastSLAM 2.0进行基于特征的SLAM的示例。

动画的含义与FastSLAM 1.0的情况相同。

5.5 基于图的SLAM

这是基于图的SLAM的示例。

蓝线是实际路径。

黑线是导航推测路径。

红线是基于图的SLAM估算的路径。

黑星是地标，用于生成图的边。

六、路径规划

6.1 动态窗口方式

这是使用动态窗口方式（Dynamic Window Approach）进行二维导航的示例代码。

6.2 基于网格的搜索

• 迪杰斯特拉算法

这是利用迪杰斯特拉（Dijkstra）算法实现的基于二维网格的最短路径规划。

动画中青色点为搜索过的节点。

• A*算法

下面是使用A星算法进行基于二维网格的最短路径规划。

动画中青色点为搜索过的节点。

启发算法为二维欧几里得距离。

• 势场算法

下面是使用势场算法进行基于二维网格的路径规划。

动画中蓝色的热区图显示了每个格子的势能。

6.3 模型预测路径生成

下面是模型预测路径生成的路径优化示例。

算法用于状态晶格规划（state lattice planning）。

• 路径优化示例

• 查找表生成示例

6.4 状态晶格规划

这个脚本使用了状态晶格规划（state lattice planning）实现路径规划。

这段代码通过模型预测路径生成来解决边界问题。

• 均匀极性采样（Uniform polar sampling）

• 偏差极性采样（Biased polar sampling）

• 路线采样（Lane sampling）

6.5 随机路径图（PRM）规划

这个随机路径图（Probabilistic Road-Map，PRM）规划算法在图搜索上采用了迪杰斯特拉方法。

动画中的蓝点为采样点。

青色叉为迪杰斯特拉方法搜索过的点。

红线为PRM的最终路径。

6.6 Voronoi路径图规划

这个Voronoi路径图（Probabilistic Road-Map，PRM）规划算法在图搜索上采用了迪杰斯特拉方法。

动画中的蓝点为Voronoi点。

青色叉为迪杰斯特拉方法搜索过的点。

红线为Voronoi路径图的最终路径。

6.7 快速搜索随机树（RRT）

• 基本RRT

这是个使用快速搜索随机树（Rapidly-Exploring Random Trees，RRT）的简单路径规划代码。

黑色圆为障碍物，绿线为搜索树，红叉为开始位置和目标位置。

• RRT*

这是使用RRT*的路径规划代码。

黑色圆为障碍物，绿线为搜索树，红叉为开始位置和目标位置。

• 基于Dubins路径的RRT

为汽车形机器人提供的使用RRT和dubins路径规划的路径规划算法。

• 基于Dubins路径的RRT*

为汽车形机器人提供的使用RRT*和dubins路径规划的路径规划算法。

• 基于reeds-shepp路径的RRT*

为汽车形机器人提供的使用RRT*和reeds shepp路径规划的路径规划算法。

• Informed RRT*

这是使用Informed RRT*的路径规划代码。

青色椭圆为Informed RRT*的启发采样域。

• 批量Informed RRT*

这是使用批量Informed RRT*的路径规划代码。

• 闭合回路RRT*

使用闭合回路RRT*（Closed loop RRT*）实现的基于车辆模型的路径规划。

这段代码里，转向控制用的是纯追迹算法（pure-pursuit algorithm）。

速度控制采用了PID。

• LQR-RRT*

这是个使用LQR-RRT*的路径规划模拟。

LQR局部规划采用了双重积分运动模型。

6.8 三次样条规划

这是段三次路径规划的示例代码。

这段代码根据x-y的路点，利用三次样条生成一段曲率连续的路径。

每个点的指向角度也可以用解析的方式计算。

6.9 B样条规划

这是段使用B样条曲线进行规划的例子。

输入路点，它会利用B样条生成光滑的路径。

第一个和最后一个路点位于最后的路径上。

6.10 Eta^3样条路径规划

这是使用Eta ^ 3样条曲线的路径规划。

6.11 贝济埃路径规划

贝济埃路径规划的示例代码。

根据四个控制点生成贝济埃路径。

改变起点和终点的偏移距离，可以生成不同的贝济埃路径：

6.12 五次多项式规划

利用五次多项式进行路径规划。

它能根据五次多项式计算二维路径、速度和加速度。

6.13 Dubins路径规划

Dubins路径规划的示例代码。

6.14 Reeds Shepp路径规划

Reeds Shepp路径规划的示例代码。

6.15 基于LQR的路径规划

为双重积分模型使用基于LQR的路径规划的示例代码。

6.16 Frenet Frame中的最优路径

这段代码在Frenet Frame中生成最优路径。

青色线为目标路径，黑色叉为障碍物。

红色线为预测的路径。

七、路径跟踪

7.1 姿势控制跟踪

这是姿势控制跟踪的模拟。

7.2 纯追迹跟踪

使用纯追迹（pure pursuit）转向控制和PID速度控制的路径跟踪模拟。

红线为目标路线，绿叉为纯追迹控制的目标点，蓝线为跟踪路线。

7.3 史坦利控制

使用史坦利（Stanley）转向控制和PID速度控制的路径跟踪模拟。

7.4 后轮反馈控制

利用后轮反馈转向控制和PID速度控制的路径跟踪模拟。

7.5 线性二次regulator（LQR）转向控制

使用LQR转向控制和PID速度控制的路径跟踪模拟。

7.6 线性二次regulator（LQR）转向和速度控制

使用LQR转向和速度控制的路径跟踪模拟。

7.7 模型预测速度和转向控制

使用迭代线性模型预测转向和速度控制的路径跟踪模拟。

这段代码使用了cxvxpy作为最优建模工具。

八、项目支持

可以通过Patreon（https://www.patreon.com/myenigma）对该项目进行经济支持。

如果你在Patreon上支持该项目，则可以得到关于本项目代码的邮件技术支持。

本文作者包括有Atsushi Sakai (@Atsushi_twi)，Daniel Ingram，Joe Dinius，Karan Chawla，Antonin RAFFIN，Alexis Paques。

项目地址：https://github.com/AtsushiSakai/PythonRobotics

原文：

https://atsushisakai.github.io/PythonRobotics/#what-is-this

作者：

AtsushiSakai，日本机器人工程师，从事自动驾驶技术开发，精通C++、ROS、MATLAB、Python、Vim和Robotics。

译者：弯月，责编：郭芮

如果喜欢我的文章，那么
“在看”和转发是对我最大的支持！

 （戳下面蓝字阅读）
Python基础——for循环和range()内建函数
超实用的30个 Python 案例
整理了 37 个 Python 网站开发库
Python文本终端GUI框架，太酷了
Python入门教程丨1300多行代码，让你轻松掌握基础知识点
150行Python代码模拟太阳系行星运转
如何用 Python 实现一个图数据库（Graph Database）？
强烈推荐10 个 Python IDE 和代码编辑器！
60行python代码实现弹球小游戏(含源码)

Python编程大全
进群学习交流加 : mm1552923
                          觉得不错，请点个在看呀

分享