Python算法精粹:掌握50种必备算法的秘诀

本文是一些机器人算法(特别是自动导航算法)的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 扩展卡尔曼滤波本地化


一文洞悉Python必备50种算法的图1


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


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


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




3.2 无损卡尔曼滤波本地化


一文洞悉Python必备50种算法的图2


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


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




3.3 粒子滤波本地化


一文洞悉Python必备50种算法的图3


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


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


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


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




3.4 直方图滤波本地化


一文洞悉Python必备50种算法的图4


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


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


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


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


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

不需要初始位置。




四、映射


4.1 高斯网格映射


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


一文洞悉Python必备50种算法的图5


4.2 光线投射网格映射


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


一文洞悉Python必备50种算法的图6


4.3 k均值物体聚类


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


一文洞悉Python必备50种算法的图7


4.4 圆形拟合物体形状识别


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


一文洞悉Python必备50种算法的图8


蓝圈是实际的物体形状。


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


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


五、SLAM


同时本地化和映射(Simultaneous Localization and Mapping,SLAM)的例子。


5.1 迭代最近点匹配


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


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


一文洞悉Python必备50种算法的图9




5.2 EKF SLAM


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


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


绿叉是估计的地标。


一文洞悉Python必备50种算法的图10




5.3 FastSLAM 1.0


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


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


红点是FastSLAM中的粒子。


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


一文洞悉Python必备50种算法的图11




5.4 FastSLAM 2.0


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


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


一文洞悉Python必备50种算法的图12




5.5 基于图的SLAM


这是基于图的SLAM的示例。


蓝线是实际路径。


黑线是导航推测路径。


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


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


一文洞悉Python必备50种算法的图13




六、路径规划


6.1 动态窗口方式


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


一文洞悉Python必备50种算法的图14




6.2 基于网格的搜索


  • 迪杰斯特拉算法


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


一文洞悉Python必备50种算法的图15


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


  • A*算法


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


一文洞悉Python必备50种算法的图16


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


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


  • 势场算法


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


一文洞悉Python必备50种算法的图17


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



6.3 模型预测路径生成


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


算法用于状态晶格规划(state lattice planning)。


  • 路径优化示例

一文洞悉Python必备50种算法的图18

  • 查找表生成示例

一文洞悉Python必备50种算法的图19


6.4 状态晶格规划


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


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




  • 均匀极性采样(Uniform polar sampling)


一文洞悉Python必备50种算法的图20


  • 偏差极性采样(Biased polar sampling)


一文洞悉Python必备50种算法的图21


  • 路线采样(Lane sampling)


一文洞悉Python必备50种算法的图22


6.5 随机路径图(PRM)规划


一文洞悉Python必备50种算法的图23


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


动画中的蓝点为采样点。


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


红线为PRM的最终路径。




6.6 Voronoi路径图规划


一文洞悉Python必备50种算法的图24


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


动画中的蓝点为Voronoi点。


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


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




6.7 快速搜索随机树(RRT)


  • 基本RRT


一文洞悉Python必备50种算法的图25


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


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


  • RRT*


一文洞悉Python必备50种算法的图26


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


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




  • 基于Dubins路径的RRT


一文洞悉Python必备50种算法的图27


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


  • 基于Dubins路径的RRT*


一文洞悉Python必备50种算法的图28


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


  • 基于reeds-shepp路径的RRT*


一文洞悉Python必备50种算法的图29


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


  • Informed RRT*


一文洞悉Python必备50种算法的图30


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


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



  • 批量Informed RRT*


一文洞悉Python必备50种算法的图31


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




  • 闭合回路RRT*


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


一文洞悉Python必备50种算法的图32


这段代码里,转向控制用的是纯追迹算法(pure-pursuit algorithm)。

速度控制采用了PID。




  • LQR-RRT*


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


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


一文洞悉Python必备50种算法的图33




6.8 三次样条规划


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


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


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


一文洞悉Python必备50种算法的图34

一文洞悉Python必备50种算法的图35

一文洞悉Python必备50种算法的图36


6.9 B样条规划


一文洞悉Python必备50种算法的图37


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


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


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




6.10 Eta^3样条路径规划


一文洞悉Python必备50种算法的图38


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




6.11 贝济埃路径规划


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


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


一文洞悉Python必备50种算法的图39


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


一文洞悉Python必备50种算法的图40




6.12 五次多项式规划


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


一文洞悉Python必备50种算法的图41


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




6.13 Dubins路径规划


Dubins路径规划的示例代码。


一文洞悉Python必备50种算法的图42




6.14 Reeds Shepp路径规划


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


一文洞悉Python必备50种算法的图43




6.15 基于LQR的路径规划


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


一文洞悉Python必备50种算法的图44


6.16 Frenet Frame中的最优路径


一文洞悉Python必备50种算法的图45


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


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


红色线为预测的路径。




七、路径跟踪


7.1 姿势控制跟踪


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


一文洞悉Python必备50种算法的图46




7.2 纯追迹跟踪


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


一文洞悉Python必备50种算法的图47


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




7.3 史坦利控制


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


一文洞悉Python必备50种算法的图48




7.4 后轮反馈控制


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


一文洞悉Python必备50种算法的图49




7.5 线性二次regulator(LQR)转向控制


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


一文洞悉Python必备50种算法的图50


相关阅读: ApolloAuto/apollo:开源自动驾驶平台 https://github.com/ApolloAuto/apollo


7.6 线性二次regulator(LQR)转向和速度控制


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


一文洞悉Python必备50种算法的图51


相关阅读: 完全自动驾驶:系统和算法 - IEEE会议出版物 http://ieeexplore.ieee.org/document/5940562/


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


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


一文洞悉Python必备50种算法的图52


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




八、项目支持


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


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


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




免责声明:本文系网络转载或改编,未找到原创作者,版权归原作者所有。如涉及版权,请联系删

QR Code
微信扫一扫,欢迎咨询~

联系我们
武汉格发信息技术有限公司
湖北省武汉市经开区科技园西路6号103孵化器
电话:155-2731-8020 座机:027-59821821
邮件:tanzw@gofarlic.com
Copyright © 2023 Gofarsoft Co.,Ltd. 保留所有权利
遇到许可问题?该如何解决!?
评估许可证实际采购量? 
不清楚软件许可证使用数据? 
收到软件厂商律师函!?  
想要少购买点许可证,节省费用? 
收到软件厂商侵权通告!?  
有正版license,但许可证不够用,需要新购? 
联系方式 155-2731-8020
预留信息,一起解决您的问题
* 姓名:
* 手机:

* 公司名称:

姓名不为空

手机不正确

公司不为空