← Back
LEARNING PATH / VISION

Vision learning path

What the vision group delivers is not a picture with boxes drawn on it, but target position, pose and confidence that the control group can use immediately. Algorithms must hold a stable frame rate on limited compute while coping with glare, occlusion, motion blur and camera shake.

Source and modifications The learning path below is quoted and adapted from a third-party open-source tutorial and is not GKD's original work — "Understanding CV and the RoboMaster Vision Group", 2022/23 vision group of the YueLu RoboMaster team, College of Robotics, Hunan University · Zeng Qingcheng (neozng1@hnu.edu.cn). Original repository: NeoZng/vision_tutorial ↗. The document is under a CC licence (attribution BY required); the repository is under the MIT License. Copyright belongs to the original author. If the original is updated, the upstream repository takes precedence. We have removed and adapted parts of the original — please note: the original targets the RoboMaster University Championship (UC), while GKD competes only in the RoboMaster University League (RMUL, i.e. UL) and not in UC. We therefore removed the chapters on UC-only units (engineering robot, radar, dart, autonomous infantry), removed ROS material unrelated to our technical stack, and dropped the author's personal promotion section. The technical content itself is unchanged; for rules, dimensions and parameters always refer to the current season's official manual.

版权声明

Copyrights 湖南大学 RoboMaster跃鹿战队2022视觉组 曾庆铖 All Rights Reserved.

本文档采用CC协议,要求保留姓名标示(BY)。本仓库采用 MIT License,因此你也可以将本文视作 HTML代码或说明文档。

只需要注明引用来源,就可以任意使用、复制、修改、合并、散布、出版/再版本文的部分或全部内容;本文中引用的第三方内容版权归其著作者所有。

文中的有部分取自网络图片未能识别版权信息,若出现了您的图片被使用却尚未标注出处等的情况,请通过邮箱/qq等方式联系笔者。

-1.Catalogue

推荐使用支持Latex的markdown引擎作为本文的阅读工具,如typora、vscode markdown addons、smark等。由于文中有一些动图,导出的PDF可能无法正常观看,可以到Image_base文件夹下寻找对应动图,或打开markdown对应。


每个RoboMaster心中应该都有一个冠军梦,和一座金灿灿的奖杯

0.前言

在阅读本文之前,你需要有计算机科学的基本知识并至少掌握一门编程语言,同时对 robomaster比赛规则和过程有大致的了解。

若只是希望知道视觉组的基本工作,仅需要阅读第二部分、第五部分的第零节。

若你是计算机视觉的初学者,并不打算/已经过了参加 RoboMaster的年纪,请跳过第二、第六部分。

而你是/打算称为一名RMer尤其是算法/视觉组的队员,通篇阅读即可。

面向的主要读者:参加RoboMaster比赛的感知/视觉/算法组成员和有志于计算机视觉领域研究的同学。

笔者希望在这本教程中向大家介绍视觉组的工作的基本概况和进入视觉组需要学习的知识,面向的对象为热爱机器人的朋友、战队中的其他技术组或准备进入视觉组的同学。本文会尽量广泛、全面地向你介绍视觉组的方方面面,同时可能涉及一些技术细节,提供尽可能直观的认识和必要的公式推导,但又不涉及过多深奥的数学知识,让其他技术组也能够了解机器人视觉模块的运行机制,以便更好地协作开发,防止出现各自为战的境况。同时让新人能更快地接触这些知识,明白视觉软件开发的过程,减少踩坑的次数。本文不提供算法对应的代码,即没有手把手的实践环节。我们力求帮助你尽快扩展视觉开发、机器人开发所需的知识面。因此,看完本教程并不意味着你已经掌握了所有知识,当且仅当你成功把所学应用到实际之中,才可谓成功。

本文提供了大量的优质教程的链接,避免重复劳动,并给予读者更多选择。若已有外部优质资源,这里将给出链接并以综述或总览的形式进行介绍。

最后也是记录一下笔者一年多来的成长,算是给自己的一个交待。限于笔者的水平,文中难免出现错误和理解不当之处(小孩子不懂事写着玩的),请读者指出错误,多多包涵。可以直接提交issues或者pr、通过邮箱或qq联系我。

23/10/12增补:现在回看当时编写的内容,很多知识理解并不是很恰当,有一些想当然,也有一些对新人不够友好的地方。最要命的是没有做到详略得当,介绍了太多关于神经网络的东西。技术的发展飞快,在大规模与训练当道、0代码训练基础模型的时代,这些东西多少有些无力。当然,所谓尽信书不如无书,你已经是一个成熟的学生了,要学会自己辨别哪些内容值得学习。

1.摘要

  • 计算机视觉(computer vision)无疑是当今AI界最火热的研究领域之一,和自然语言处理、强化学习并称三架马车。自然而然,在RoboMaster的赛场,视觉的软硬件开发也占有一席之地,视觉组便和其他技术组一样应运而生了。
  • 视觉组负责的机器人模块主要是传感器和数据处理,即通过对相机、激光雷达等传感器采集到的信息进行处理从而让机器人在一定程度上具有”视觉”或“知觉”功能。
  • 在比赛中,视觉组能够让机器人自动识别地方装甲板,实现“自瞄外挂”;也能让操作手轻松地击打能量机关,使得全队获得增益;视觉组打造的感知系统更是哨兵机器人上的“大脑”,没有视觉组的工作,这些机器人就完全失去了在场上的作用;视觉组同时还全权负责雷达这个兵种,耳听八方眼观六路,可谓是战场指挥官。

在第二部分,主要介绍了视觉在每个兵种中的作用。

第三部分则是视觉组在开发时会使用到的软件,如Linux系统、一些IDE和小工具。随后将会简述视觉组会使用的硬件如相机、各种运算平台等。

在第五部分是本教程的重头戏,介绍了比赛中会用到的算法,第五部分几乎覆盖了计算机视觉的所有基础内容;紧接着在第六部分着重叙述比赛中对第五部分算法的工程应用,即如何让算法落地发挥作用。这两个部分占据了全篇3/4以上的篇幅。

第七部分简要说明了在视觉组需要的开发技能和知识,同时也代表着你能在视觉组学到什么东西。如果你刚入门,看完第二部分以后一定要先去第七部分的最后面看看视觉组的学习路线图和第八部分的常见Q&A!

最后一个部分是笔者在视觉组练习时长两年半的心得体会,和对新人上手视觉工作的一些建议。

!另外,在每个部分中都会穿插地介绍一些视觉组在这些方面会接触到的知识!

计算机视觉学习之旅马上开始,祝你学习愉快,不要忘记实践噢!




2.视觉在各兵种中的作用

看图说话,有图有真相!先看看RM比赛场地”战场“上和视觉组有关的场地道具和交互模块吧。记住这些框框和注释,方便理解接下来出现的一些名词和概念,记不住也没关系,再回来看看。

图源RoboMaster2022超级对抗赛规则手册;有删改

2.1.装甲板识别(步兵、英雄、哨兵)

  • 由于机器人上安装的图传模块到操作手看到的第一视角的传输延迟加上操作手反应速度的滞后,操作手几乎很难手动瞄准高速运动的机器人上的装甲板。因此,视觉组在这三个兵种的研发上主要负责装甲板的识别算法,通过处理图像找到相机视野范围内的装甲板(相机一般安装在云台上,和枪口平行放置并指向同一个方向,类似瞄准镜),进而向下位机(STM32等用于控制的MCU,microcontroller unit,单片机)发送此装甲板的相对枪口的角度数据,电控根据此数据控制电机自动转向目标装甲板,实现装甲板的自动打击。
自动跟随装甲板效果
  • 因为图像处理、上位机和下位机的通信、电机的动作(即转向目标)、发射机构的动作(拨弹机构旋转将弹丸推向摩擦轮+弹丸链路空隙),以及最耗时的弹丸在空中飞行,都会带来延迟,它们的总时长最高可以达到0.5~0.7s。倘若将当前时刻识别到的装甲板相对于云台的位置原原本本地发送给下位机进行控制,在识别到装甲板时刻之后的这段时间,对方机器人很可能已经移动到其他位置了(以步兵机器人为例,最快移动速度甚至可以达到6m/s,平均移动速度也有2~3m/s),我们打出的弹丸只会和目标装甲板擦身而过。因此,视觉算法还需要根据对方机器人在过去一段时间的运动状态或掌握其他的先验知识,预测对手在当前时刻后一段时间的运动趋势(即上述所有会带来迟滞的环节的时间总和),从而提前将云台转向对应位置,实现对目标运动的预测。玩过空战游戏的同学可能很清楚,开火的时候都要给出一个提前量。通过理论来预测物体运动趋势最早可以追溯到二战期间,控制论(cybernetics)之父维纳博士曾在美国空军基地研究自行防空火炮预测敌机运动轨迹的问题,维纳滤波器(最小方差滤波器)也是在那时候被提出的。

  • 随着自瞄算法的不断升级进步,RoboMaster的赛场上也出现了“反自瞄”,其中的代表之一就是“小陀螺”。机器人在小陀螺模式下,云台和底盘处于分离状态,在底盘绕运动中心高速自旋的同时云台保持稳定不动。在这种情况下,视觉识别的难度会大大上升,一是自瞄很难跟上装甲板的高速转动,基本上我方机器人的云台运动会滞后于装甲板的运动;即使使用了前述的预测算法,由于装甲板在很短的时间内就会在车身的一侧“消失“,这往往会导致预测失误,给出的提前量过大使得预测值超出车身范围(瞄到空气);二是即使跟随到一个装甲板后,那个装甲板随着底盘的转动很快消失在视野中,此时就要锁定另一块装甲板,使得云台在切换目标的过程中会来回转动,无法稳定,导致弹丸命中率下降,且从操作手的第一界面看来,整个画面不断晃动带来晕眩感,体验极差(更有甚者直接”晕车“)。其他单位尚且可以通过”火力覆盖“的方法凭借运气击中”小陀螺“,而开火冷却时间长、弹丸价格高昂的英雄机器人可不敢随随便便对着小陀螺开火。

因此,反“小陀螺”算法出现了:通过设计合适的策略,识别对手的机器人处于小陀螺状态,然后让云台对准敌方机器人的中心位置而不再跟随装甲板移动。由于机器人处于“小陀螺”状态时基本上是匀速转动,这样就可在适当的时机开火(可以采用一些预测算法预测装甲板何时运动到云台所对准的位置),提高命中率。也可以通过一段时间的检测,计算出对方机器人地盘转动的速度,从而预测出每一块装甲板出现和消失的时刻,进行打击。

当然,随着反”小陀螺“算法的出现,赛场上又开始出现反‘反“小陀螺“ ’算法如变速小陀螺、超快小陀螺(舵轮步兵)、随机转动等。哈尔滨理工大学(荣成校区)还研发出了偏心小陀螺,它的云台有额外的平动自由度,使得转动的中心不在底盘的中心,戳这里看(精准空降)。快进到反反反反反小陀螺(禁止套娃)因此,今后必定需要研发出鲁棒性(健壮性)和泛化性能更好的算法,才能应对愈发“卷”的比赛啊~

处于小陀螺运动下的步兵机器人
  • 英雄狙击点机制的加入让三维视觉又有了用武之地。英雄机器人在狙击点发射大弹丸将会返还10个金币,并且狙击点处发射的大弹丸若击中对方的基地会获得2.5倍的攻击力增益!而传统英雄吊射(狙击)都是采用量测射表的方式进行的:通过云台不同仰角的弹丸发射测试,获取不同距离的落点信息,得到一张射表;然后在操作界面上利用自定义UI画出不同距离时需要对准的位置,从而实现瞄准。
类似狙击枪/火炮的瞄准镜

这种方法一方面对发射机构和云台的机械精度要求很高,另一方面还要保证摩擦轮的状态始终保持一致(如果摩擦轮的硬度/磨损程度发生改变都会使得弹速变化),同时,操作手还得是“写轮眼”,要能够精确对准对应距离的瞄准线。实际上算是一种半开环的控制过程,操作手只能根据每一发弹丸的落点对云台姿态进行修正。诸多约束条件的限制使得英雄狙击难上加难,狙击成功率基本靠“抽奖”

目前似乎没有队伍使用远距离视觉吊射(狙击基地)。存疑。

哈工大 I HITer 战队的英雄机器人头顶一个 ArUco 定位标签,用于辅助定位

2.2.能量机关(步兵)

2023赛季中能量机关的图样已经更改,旋转机制保持不变,但激活之后的增益有改动,击打的准确度越高,增益越强力。详见规则手册。

  • 在比赛场地的中央有一个风车形状的场地道具,就是能量机关。它也被称作”风车“、”神符系统“或”大符“(大能量机关)”小符“(小能量机关)。能量机关的激活点是我方高地处的一个高台,距能量机关7m。我们需要用小弹丸按顺序连续击中五片随机亮起的扇叶的末端装甲板才可以激活能量机关。激活小能量机关能够为队伍带来50%的攻击力增益,大能量机关能为队伍提供100%的攻击力增益和50%的防御增益。由于图传模块使用的是广角镜头,因此从操作手的第一视角看来五片扇叶的间距非常小,再加上遥控的延迟,难以通过鼠标移动来进行打击,并且小能量机关处于匀速旋转的状态,大能量机关更是以 0.785sin(1.884t)+1.305 的角速度旋转(在2022赛季其旋转角速度已经修改为,上式的四个参数都是在一定范围内随机初始化的),操作手很难预测其运动轨迹。两片扇叶之间的击打间隔必须小于2s,否则能量机关将会重置回到初始状态,之前已经击打成功亮起的扇叶会直接熄灭。这便需要视觉组设计算法来识别未被击打过的末端装甲板并对其实现自动瞄准,找准时机控制子弹的发射从而实现自动击打能量机关。

  • 能量机关任务对视觉来说可以分为两个部分:扇叶识别和运动状态拟合。首先需要识别出待击打的扇叶,然后利用包括能量机关的旋转速度、能量机关的几何尺寸等已知信息预测出一段时间后的目标位置,进而发送给下位机实现击打。对于大能量机关,我们还需要辨识出被随机初始化的四个用于确定旋转角速度的参数:幅值、角频率、初相和角速度常数。扇叶识别的任务和装甲板识别非常相似,难度更是有过之而无不及,因为在激活点看向能量机关,背景处常有形式复杂多样的光线干扰。运动状态拟合则需要综合信号处理、傅里叶分析、非线性优化等知识。

  • 除了激活我方的能量机关,我们还可以在对手激活能量机关的过程中干扰对手。在能量机关的激活的规则中规定:倘若击打了错误的扇叶(如已经击打过的扇叶或尚未亮起的扇叶),能量机关会将进行重置,因此,我们可以进入敌方半场,通过识别对手已经激活的扇叶并自动瞄准它,发射子弹击中错误的装甲板进而触发重置以干扰对手的激活过程。

正在激活小能量机关的步兵机器人;图源RoboMaster2021内部技术交流

2.3.哨兵

  • 哨兵机器人是场上的全自动机器人,没有操作手。它的移动、目标搜索和打击都依赖自身的感知与自主决策,官方也常称之为 AI 机器人。编写一套稳定可靠的感知与决策程序,是发挥哨兵威力的关键,也是视觉组在哨兵上最主要的工作。

  • 在 2026 赛季的高校联盟赛(RMUL)3V3 对抗赛中,对战双方需要自主研发步兵机器人、英雄机器人和哨兵机器人,在指定场地内进行战术对抗;比赛以胜利点结算,使对方胜利点先归零的一方获胜。哨兵是其中唯一全自动运行的单位,它的存活与输出会直接影响场上的控制权。哨兵的具体参数(血量、发弹量、底盘功率、装甲模块布局、复活与增益规则等)以当赛季《比赛规则手册》和《机器人制作规范手册》为准,本页不再罗列具体数值。

  • 哨兵对系统可靠性的要求远高于其他兵种。整场比赛无人干预,定位、限位、通信、自动决策和异常保护必须能够长时间稳定工作,而不是只在短时间演示里有效。比赛中常见的失效形式包括:目标丢失后云台长时间空转、在多个目标之间来回切换无法收敛、通信中断后状态机卡死、以及上电自检不完整导致开局即异常。这些问题需要在联调阶段专门设计测试用例,而不是等上场后才发现。

  • 若自下而上剖析哨兵机器人的技术栈,可以分为如下几个层面: 1. 运动控制层,一般由电控实现,即能够让底盘或云台按照上层指令(传统机器人为操作手通过遥控器发送或算法平台发送),达到特定位置或跟随指定轨迹。注意,此处的闭环对象是电机等执行机构。例如后面介绍的例子中,当机器人与目标轨迹存在偏差时,轨迹跟踪层会给运动控制层发来机器人底盘运动的期望速度和加速度等信息,若当前使用麦克纳姆轮底盘,我们根据底盘运动的期望信息,通过运动学和动力学分析解算出每个轮子应当输出的扭矩和速度等,并让电机对这些控制目标进行闭环。 2. 轨迹跟踪层,由算法实现。其上的路径规划算法会给出一条预定轨迹,区别于运动控制层,这里的闭环对象一般是机器人底盘等,例如底盘当前距离目标轨迹的偏差为150mm。该值会发往运动控制层,交由电机pid计算出具体需要的控制量以通过电子调速器完成最终闭环。轨迹跟踪层拥有当前的机器人位置和其他状态信息,这些信息是通过同步定位与建图技术(SLAM)获取的。 3. 路径规划层,由算法实现。给定由建图与定位层 提供的一个导航地图上的位置,路径规划层使用 A*、D* 等算法规划出一条可行的路径,该路径往往由一系列路径点组成。通过对路径点进行样条曲线插值并避免碰撞,生成的轨迹会交由轨迹跟踪层处理。 4. 定位与建图层,由算法实现,一般可以看作最高层的算法。通过机载的激光雷达、IMU、RGB相机、超声波雷达、毫米波雷达等的组合,感知周围的环境并和预先建立的全局地图比较,以确定机器人在场地所处的位置。这里的全局地图特指类似导航软件和地图软件中使用的导航地图,这种地图一般只需要标注机器人的可通行区域而不需要详细的建筑或物品标注。最常见的导航地图是占栅格地图。 5. 感知与决策层,和定位与建图同级,虽然SLAM层也有感知环节,但感知与决策层的感知一般指的是对特定的具有明确含义的目标及其状态的感知,如对手的步兵机器人的方位、对方机器人的运动状态等。例如安装在哨兵顶部的360°鱼眼相机采集回的图片,在其上运行目标检测算法,发现12点和4点方向各有一辆对方的步兵机器人,而3点钟方向有一台我方的英雄机器人;同时通过裁判系统读取的数据得知12点方向的敌人血量较少,但3点钟方向有一台我方机器人,可以配合其对4点方向的对方机器人完成围杀,故做出决策:向4点方向移动+发射机构转向4点方向+准许该云台在瞄准目标之后开火的指令集合。向4点移动的指令会由路径规划层接收,根据定位与建图层给出的导航地图规划一条不会发生碰撞的路径给下层进行进一步操作,发射机构转向4点方向的任务则直接交给运动控制层执行。常见的自主决策方法包括决策树、状态机和基于强化学习的深度策略。

  • 对视觉组来说,哨兵上真正需要交付的是稳定的目标信息:在底盘移动、云台转动、场地光线变化的情况下持续输出可靠的目标位置与状态,并在没有目标时给出明确的“无目标”语义,让决策层能够安全地处理。这比单纯追求识别率更接近哨兵场景的真实需求。

3.视觉组接触的软件

进行视觉开发会用到各种各样的软件、开发环境、辅助工具等,所以很有必要了解一些相关的快捷键、命令、使用技巧。选择一款适合自己的IDE能够提高开发效率,方便版本管理。

3.1.Ubuntu

  • 为什么使用Ubuntu

  • Ubuntu是一个Debian系分支的第一大系统,是当前用户量最大的linux发行版。因此,遇到任何问题一般都能够在用户社区askubuntu中得到解答。它的安装也非常的方便,并且在更新到20.04后,ubuntu的桌面美观性也有提升。

  • Linux下开发C++程序相比Windows有无与伦比的优势,可以方便的配置各种第三方库和依赖。常言道python好用是因为有大量开箱即用的第三方库,可以轻松通过pip安装,而Linux下通过yum/apt/pacman等包安装软件管理的软件包,实际上就是C/C++隐藏的库安装/管理利器!apt是Debian系发行版用于管理第三方库的一个软件,负责管理系统中安装的各类软件包,开发包,依赖库。可以通过apt轻松地安装各种软件(可执行文件)、开发库(头文件headers,.so动态链接库等。如果你曾经在Visual Studio中为项目配置繁杂的头文件、链接库路径等依赖,你一定会爱上Linux下用cmake管理c++环境的开发方式。

  • Linux的内核和上层系统都比Windows更加精简,故在运行时占用的各类资源都要小于Windows。在不打开任何应用的情况下,笔者的电脑在运行Windows10时占用的内存为4.2G,cpu占用率在10-20%左右,而运行Ubuntu20.04LTS的时候,只使用了2.2G的内存,cpu占用率只有10%不到。这样,在运行我们的视觉算法程序时,可以更充分地利用系统资源,最大程度压榨电脑的性能。(甚至可以在测试结束后实际运行时关闭图形界面,只保留终端!这样,系统内核作为唯一需要运行的基础程序,大概能将cpu占用率缩小到1-5%)

  • Linux对于深度学习的支持比Widnows更加友好,经常有sh脚本能够一键配置开发环境。此外Linux对一些设备驱动的支持也更完善,我们可以选择挂载自己需要的驱动和IO,并且精简属于自己的内核,增加实时性补丁等。

截止本文编写到此处,Ubuntu已经发布22.04 LTS

现在,最方便的linux发行版是windows wsl(windows subsystem linux),这是一个由微软维护的和windows高度集成与优化的linux发行版,wsl2本质上和虚拟机无异,但性能和便捷性相比一般的virtualbox和vmwares等都由很大提升。wsl可以在命令行或Microsoft商店直接安装。目前已经支持直接打开linux下的GUI界面。最佳实践在wsl内运行code server,在你的windows下通过ssh连接到wsl。

不建议新人使用这种方式,双系统或一般的虚拟机更适合你。wsl2在连接usb和网络设备上目前仍然存在一些复杂的步骤。

  • 提到Linux就不得不提到命令行的使用,在Linux上进行开发常会使用到命令行,有些软件甚至只有命令行界面的版本。在一些时候,直接在命令行中用键盘操作可能要比数不清的鼠标点击快得多。你需要学习:

  • cd、ls 、pwd、mv、cp、touch、diff、rm、cat、mkdir、rmdir、echo、tar等文件系统的基本操作,grep、find 查找文件和目录

  • 帮助手册man和-help参数。
  • sudo、su、chmod等权限相关的操作。
  • ping、ifconfig、wget等网络相关的操作。
  • 一定要亲手熟悉命令行的基本命令,切忌只看不动手!学习以上命令,戳这里Linux Commands,简版的教程推荐这个:Linux Commands | Harry's Blog

  • 至少掌握一个无GUI的文本编辑器的基本使用,如vi,vim,nano等。这能够帮助你在系统出现问题的时候快速修改一些配置文件,或是在使用ssh连接的时候简单地编写一些程序。当然,笔者不推荐你将这些文本编辑器作为主力IDE使用(即使是安装了各种各样的插件!)虽然一个熟练使用vim的程序员和一个熟练使用eclipse的程序员拥有相同的开发效率,但是vim的学习成本可不知道比eclipse高多少!

  • Linux的设计哲学是“一切皆文件”。它将所有的IO设备如网络接口、usb接口、显示屏、相机、键盘鼠标、应用都视为文件,和这些“文件”的交互就是以规定的方式进行读写。因此,有必要了解Linux下的基本目录和文件组织方式,其目录结构请参考:Linux文件目录结构一览表。对于文件读写权限的管理也非常重要的,这决定了用户/程序对某些文件是否有访问权限。要是对文件系统有一些了解,那便更好不过了。

  • 在使用系统的时候,建议大家有良好的文件分类习惯,把代码库、软件、开发环境分开存放,避免出现home目录乱糟糟的情况。

这篇文章就是在Ubutnu下使用markdown编辑器完成的


3.2.IDE

想要编写代码,光靠文本编辑器+gcc+gdb可不行,我们要充分利用技术进步带来的便利,谁不喜欢做懒人呢。这里推荐几款Linux下编写C++程序使用的IDE:

  • VSCode:微软的小儿子,啥系统都能用。丰富的插件生态只有你想不到没有你找不到,配置完之后使用起来非常方便,比如C++就有一个C++ extensions pack。官方文档也很详细,毕竟是微软主推的下一代编辑器。关键是好看啊!在使用了snippets和Visual Sutdio Intellicode这两个插件之后,智能提示也足够智能。想要写其他的语言也能够一条龙配齐,总之,上手容易且可定制化程度极高。

安装了wsl后可以在windows上利用vscode server可以获得与ubuntu无缝衔接的体验。

我们建议c++开发者使用clangd或c/c++以及cmake插件以实现代码语法高亮和智能提示等功能。

  • Clion:JetBrain家的IDE,界面很美观,智能提示也很智能。以前用过PyCharm或者其他jb系的IDE的同学可以继续使用。统一用cmake管理项目,对cmake扩展的支持非常到位。也提供了大量的可选插件。用.edu后缀的学校邮箱可以免费申请教育资格,就可以免费用了。

JB家为了对标VSC,推出了自家的轻量级编辑器Fleet,喜欢jb设计风格的同学可以尝试。同时,Clion也更新了UI设计,选用新版ui更加简洁方便。

  • Qt:Qt也是一款跨平台的C/C++ IDE,在Qt上编写的GUI程序能够在所有平台上运行。用Qt可以方便地编写一些图形化的程序,比如串口调试助手、调参助手等。他的整体界面也算是比较清爽,适合新人使用。
  • 这里需要特别提及的是CMakeLists的编写。Linux没有Visual Studio这样保姆级的IDE,并不存在一款能够自动为你生成makefile的软件。所以至少要学习qmake和cmake中的一种工具。这里推荐cmake,虽然比qmake的语法稍微复杂一些,但是cmake的功能非常强大,拥有非常优良的跨平台支持。学习cmake还能帮助你进一步了解程序的编译、链接过程。关于程序是怎么从源代码到机器代码最后在电脑上运行起来和cmake的基本使用,请参考《程序的生前死后-Cmake-noob-comein》-NeoZng这篇文章。

对于那些不太复杂的项目,你还可以使用语法规则更简单的xmake,这是一款由国人研发的基于Lua的跨平台构建工具。

萝卜青菜各有所爱。虽然IDE把工具链都集成在了一起,极大地方便了我们的使用,但笔者还是推荐你学习一下GNU工具链的使用和基本原理,至少熟悉编译、汇编、链接的过程。这样可以更深入的了解软件的运行,以便在开发过程中出现问题的时候,快速定位问题所在并找到解决方法。



3.3.Git

团队协作开发需要一款优秀的代码管理工具,那Git就是不二之选,大家肯定都听过GitHub这个最大的提供gitlab服务的同性交友平台,它便是一个基于Git的代码托管平台。这里有个小故事,Git是Linux的元老Linus因为Linux社区被禁止使用BitKeepter这款版本控制软件后,一怒之下在一周之内用C写出来的程序哦。

我们实验室开始的时候都是用u盘拷贝程序,有时候在某个人的电脑上写一点有时候又在minipc上写一点,虽然在文件夹上标准了时间和版本号,然而这并没有什么用,这导致一次合并代码的时候有十多个版本的代码,根本不知道哪个能用哪个不能用,那时候又还不知道diff这个工具,弄得眼睛都快无了。

要学习Git,推荐这几个网站:廖雪峰的git教程 git简易指南-no deep shits! GitHub Guides

在学习Git的时候,一定要动手跟着一起实践,切忌光看不动!俗话说熟能生巧。

git的标志性图标,分岔的icon表示强大的分支功能

只要学会创建分支、合并冲突、建立远端仓库、合并分支和版本回退等基本操作即可,过于复杂的功能需要时查阅man和help,没必要强行记忆。

3.5.其他常用软件和小工具

  • Microsoft Edge DEV for Linux :Edge浏览器Linux版,可以方便同步windows下的收藏夹、设置、插件等。集锦的功能非常好用。
  • VLC:一款多媒体播放器,方便录制调试视频后进行观看。若安装系统的时候选择最小安装,则不会预装媒体软件,因此需要自己安装。
  • qv4l2:linux下相机驱动的图形界面,在Ubuntu软件商店可以找到,方便调节普通USB相机的参数。
  • Meld:一款diff软件的图形界面,方便对比文件的不同,在Git使用merge或pull的时候可能会用上,在Ubuntu软件商店可以找到。虽然vscode也预置了此功能(只需要安装git便会激活),但是只能在被git管理的repo文件夹里面启用。
  • Fsearch:和Windows下的everything类似,提供超快速的文件检索功能。
  • Typora:好看好用的markdown编辑器,本文就是使用typora编写的。使用markdown编写代码的说明文档是一个很好的习惯,这可以降低其他人阅读你编写的代码的难度,也有利于代码分享和代码的传承。同时,你的也可以使用markdown来记录自己的学习历程、一次艰难的问题解决之路。使用markdown可以提高你的记录效率。vscode内也有相关插件提供对markdown支持。现在似乎要收费了,可以在官网下载beta history版本,版本号为0.x的公测版仍然免费。
  • TigerVNC:一款局域网内可用的远程桌面软件,VNCViewer也可以作为替代。强烈推荐使用远程桌面调车!电控都有无线调试器,我们怎么能跪在地上呢(气抖冷)。在把运算平台安装到机器人上之后,我也曾经拿着一块小屏幕和键鼠,蹲在地上和机器人进行亲密交流,这不仅加深了我和机器人的感情,也加重了我的颈椎病和腰椎键盘突出。(最恐怖的是车车的云台或者底盘疯了的时候,线全部缠到机器人上!!机器人甚至有可能对你造成伤害!!都是电控的锅,你云台怎么又疯了)使用了vnc后,只要将minipc和你的笔记本连接到同一个局域网,你就可以优雅地拿着笔记本调车了。如果校园网的带宽不够,建议买一个路由器,或者和搭建裁判系统的路由器公用也可以。

其他远程桌面如Xrdp(分辨率和画质最好)、NoMachine(最流畅,画质次之,可以实现局域网内IP自动搜索,强烈推荐!)也是很好的选择。

配置Xrdp的步骤稍微有一点多,参考:Ubuntu利用xrdp实现远程桌面连接;在此过程中可以熟悉linux系统的环境变量配置以及简单的计算机网络知识。

上交的同学更是把这件事做到了极致,他们直接通过网页来修改机器人的各种参数并得到反馈信息,能做到不需要任何远程桌面就能实时调参,此想法以为妙绝!华南师范大学使用foxglove进行网页可视化,也是一种选择。不过这些都需要了解包括动态网页的构建在内的一些基本前端知识。

  • SSH:在外面不需要图形界面的时候,能够直接连接终端就是一件很方便的事情。并且在代码真正部署上车时通常我们为了最大程度降低额外开销会选择关闭图形界面,终端连接就成了不二之选。使用过SSH连接服务器的同学应该对此不陌生了,有兴趣的同学可以了解一下非对称加密的原理。linux下可以直接使用终端作为ssh的客户端,windows下Microsoft全新推出的windows terminal也十分美观。其他的如PuTTy等也可以尝试。

VScode也有SSH connection插件,可以将连接端的文件夹映射到本地,同时还可以直接在code里重用端口开启多个终端!同时,通过宿主机安装的Code Server,所有的插件也支持在远端运行。关于SSH的原理及其在windows/linxu下的简单配置,戳这里。

  • Docker:标志性的小蓝鲸logo。Docker可以把开发环境和依赖打包在一起并于Docker engine上运行,运行时完全和系统的环境隔离。就像一个轻量级的虚拟机。Docker能够将应用程序与基础架构(操作系统、开发环境、依赖等)分开,使用docker engine提供的抽象大大提高兼容性,不需要关心底层和外部复杂的关联从而专注于软件开发。相信很多同学都遇到开发的时候配环境时醉生梦死的情况,使用Docker镜像则可以免除这一切烦恼!华南师范大学PIONEER战队的开源代码就提供了Docker的部署方案。
Docker的架构

有了docker,你甚至可以在windows下运行代码(实际上对宿主机的系统没有要求),虽然不推荐这样做。




4.视觉组接触的硬件

虽然别人总觉得视觉组就是整天对着屏幕臭敲代码的程序员,实际上我们也会接触很多的底层硬件与传感器,在使用硬件的同时很可能还需要综合运用其他学科的知识。

4.1.相机

相机是机器人的眼睛。和人眼的成像原理一样,相机通过镜头汇聚光束使他们聚集在一块半导体感光元件上(相当于视网膜)从而产生可供读取的数据。随后图像随着数据线传如miniPC等运算平台(视网膜刺激视神经传到神经冲动到大脑)。时下的感光单元主要分为两种:CMOS和CCD。电类的工科生或是摄影爱好者对此应该不会感到陌生。

  • CMOS(complementary metal-oxide semiconductor)传感器是由金属氧化物半导体排成的阵列,和发光半导体相反,其上的pn结受光照时会产生电荷存储在电容中,通过和内存一样的结构采用行选和列选开关,为每一行/每个像素点配备放大器和AD(Analog-to-Digital converter,模拟-数字信号转换器),选中位置的信号会通过OP和AD,从而将电荷转换为数字信号,最后生成图像。优点是成本低,图像帧率高,但是固有不可消除的采集噪声(每个采集单元的参数不可能完全一致)会影响成像的质量。

  • CCD(charged couple device)传感器是由最简单的MOS电容器阵列构成的,与CMOS不同,CCD的每一行像素只配有一个信号处理器,利用时钟脉冲驱动产生的差压,这一行电容器采集到的电荷会以串行的方式依次通过每一行末端的信号处理器从而产生图像。因此CCD能达到更高的像素密度(CMOS上的op和ad都要占用空间),其优点是成像的一致性好,噪声很小,由于不能同时处理所有信号,其帧率一般不高。

  • 想要获得关于感光单元的更多信息,请参阅cmos和ccd。

  • 系统地学习成像原理,参见国防科技大学的MOOC:《第十五讲:CCD图像传感器》

这是我们实验室的一枚MV工业相机,可以看到正中间有一块小小的CMOS传感器晶片

相机的快门还有全局曝光和卷帘曝光之分。全局曝光时一次性采集所有感光单元的信息或为每个感光单元配备一个寄存器暂存电荷,而卷帘曝光则是分时逐行采集信息(根据其名“卷帘”就能看出是一排一排地采集数据)。若经费充足建议购买全局曝光的相机,其所有单元在同一时刻采样,曝光时间和我们设置的曝光时间才是真正一致的。而卷帘曝光的相机第一行首先完成曝光,接着是第二行、第三行...到最后一行完成曝光时,中间大概有数百ns甚至ms级的时间,这就导致整个画面其实不是在同一时间采样的,在面对高速运动的物体是会出现“果冻效应”,即图像的不同部分出现倾斜、断层等现象,极大地影响了成像质量和识别的效果。CCD相机一般使用全局快门,而CMOS相机使用全局快门的制造成本很高,需要为每个感光单元配置一个寄存器。

这两幅风扇的图像是在相同时刻分别用全局快门和卷帘快门的相机拍摄的

4.1.1. 镜头

就像人的晶状体,不同的是,人的晶状体是可以改变焦距但像距固定,而一般使用的工业相机和USB相机配套的镜头是定焦但可以调节像距的。不过在小孔成像模型中,我们把透镜看作被压缩成一个点的大小,因此在这种情况下我们认为焦距和像距等同。

从透镜到小孔的近似过程,截取自知乎-龚健男的回答-凸透镜和小孔成像的原理与联系
  • 我们都知道镜头是一块凸透镜,在他的两边各有一个焦点,焦点即所有平行进入透镜的光线的汇聚点。不同焦距的镜头其视距和视野范围不同,一般来说,视距大(看的远)的镜头,其视野范围小(可视角小);而视距短(看的近些)的镜头,视野范围大,典型的例子是广角镜头。拿生活中的例子来说,现在的手机的摄影系统都是由多个镜头组成的,每个镜头的焦距一般不同,从而适应不同焦段和视野的摄影需求。高级的镜头通常可以调节光圈大小,从而改变镜头的进光量。

  • 在比赛中,我们一般给步兵机器人配置广角镜头(适用近战,可视角广)或是6mm的镜头(中庸的选择,兼顾长短)。打击能量机关的步兵机器人会选择8mm、12mm的长焦镜头来获得更好的远距离成像效果。哨兵机器人可能会配置两个相机,分别搭载广角镜头和中短焦镜头,广角用于“广撒网”,对敌方目标进行大致的定位,之后交由另外一个相机进行精确的定位。有些打得准(弹道精度高)的队伍甚至为所有机器人都搭载了两枚镜头或选用变焦镜头,让机器人成为各个距离都能实施自动打击的多面手。海康机器人的官网提供了一个镜头选型工具:海康机器人-镜头选型工具,输入参数后可以自动计算需要的视场角和靶面尺寸、焦距等,不过它推荐的肯定是自家的产品,只需要拿着相应的参数自行联系经销商或去tb上找平替款即可。

若想要更有针对性地选择镜头的焦距,可以根据相机成像模型进行视野要求和焦距之间相关关系的计算。

  • 由于凸透镜本身的性质和镜头制造的工艺问题,使得光线在通过镜头时无法保持物体在空间中原本的位置关系而发生畸变,好在这些畸变都能够通过数学建模并由反向解算而得到还原。这需要我们通过相机标定来去除这种畸变以便还原图像中物体的真实位置。畸变主要分为切向畸变和桶型畸变,我们可以利用标定板和畸变的数学关系来进行相机标定,OpenCV中也有相关的函数可供调用。关于成像模型、畸变和标定,可以参阅相机标定和OpenCV官方文档中的CameraCaliberation。在 5.1、6.3 中我们还会提到这一点,并且给出了提供标定流程的参考文章。
USB工业相机上几种不同焦距的M12镜头,依次是超广角,4mm,6mm,8mm,12mm

M12镜头的12指的是相机上的接口直径为12mm

  • 光圈就是镜头前面可以开闭的小扇叶(M12镜头一般不可以调节光圈,但板级工业相机一般有一个“auto aperture”的选项,可以根据环境亮度调节“虚拟光圈”,毕竟上面没有机械光圈)。不同光圈大小代表不同的镜头开度,其影响的是镜头的进光量。一般用f值刻画光圈的开合程度:
f/x,x代表的是开合程度,镜头上此数值越小说明通光孔的直径越大

光圈还和成像的景深有关系,越大的光圈得到的景深越小,即成清晰像的范围越小:

光圈越大,景深越小,反之景深更大

因此,在调大光圈提高进光量的同时,能够成清晰像的距离范围就缩小了,我们需要在两者之间进行权衡。不过没有关系,为了提高画面的亮度和可视性,我们还可以调节一些其他参数:在下一个小节会介绍曝光时间和增益,以及gamma这三个参数。

此外,不可追求景深而将光圈缩得太小。一方面是进光量大大下降(平方反比级),另一方面是光在通过小孔的时候会有强烈的衍射现象,导致像的边缘模糊,边界不够锐利。


4.1.2. 相机参数调节

相机在使用过程中,除了硬件参数我们还可以调节许多采集参数,之所以很多队伍(可以说是大部分队伍)会选用工业相机,就是因为其参数调节是高度定制化的,厂商提供了SDK以方便我们进行二次发开。这里列出一些相机的主要的可调参数:

  • 曝光:每一帧图像的感光时间,其值愈大则画面的整体亮度越大,曝光时间过长过短都可能会出现宽容度不够的情况(一片雪白或是漆黑无比),选择正确的曝光是算法能否奏效的关键。如传统的灯条特征选取算法就要求恰当的低曝光以保证灯条不会出现过曝而显现白色但同时又要能够看清装甲板中间的数字以便进行模板匹配、svm分类或其他数字识别;运用卷积神经网络的目标检测算法则需要相机采集到的画面能尽量接近数据集中图片的情况,一般来说需要高一些的曝光。

  • 帧率:相机每秒钟能够获取的图像数。一般来说,如果你的图像处理算法的速度够快,那么帧率越高越好,这能够保证你处理结果的实时性。一些相机提供了硬件触发 功能,这样能够让相机以固定的时间间隔触发采样,保证两帧之间的时间相同,以便于和机器人的控制进行时间线同步。视觉算法处理、数据传输、电控算法处理、再到控制执行机构动作,最后子弹在空中飞行——这几个过程中都有时间延迟,累加之后算是非常可观。高速的算法和确定的延迟时间是打造预测算法的基础。

  • 白平衡:设定白色是怎么样的“白”,本意是不管在任何光源下都让原本呈现白色的物体通过增加偏置而还原为白色。涉及到色温和颜色空间的概念,调节此参数即调整RGB三原色的混合比例。

  • 图像保存格式:图像的编码方法,如JPEG, RGB, YUYV, YUY2等,不同的编码格式保存的信息量可能有差别。我们会在 5.0 节进一步了解相关信息。

  • 分辨率:一张图像的像素数,常见的有1920x1080,1280x720,640x480,一般来说,分辨率越高则图像保留的细节就越多,但同时相机处理的数据量变大,会降低采集帧率和每秒传输的图像数量。在之后的图像处理中,同样意味着更大的开销和处理速度下降。

  • 增益:调节感光单元在进行电荷信号放大时的增益(gain,一般是用dB表示的,这需要你注意数量级),对于图像的亮度和各颜色信息的保存都有影响。在低曝光的时候可以有效提高成像质量,但同时也可能提高噪声(不规则噪声信号也会被放大)。

  • 对比度:图像中明暗区域最亮的白和最暗的黑之间不同亮度层级的测量,差异范围越大代表对比越大,在高动态范围和高宽容度的时候,提高对比度可以凸显图像中亮度不同部分的区别,相当于用一把刻度更精细的尺子去测量物体能够得到更精细度量信息。某些情况下,提高对比度所指的则是直接增加亮暗之间的差别,让亮部更亮,暗部更暗。

  • 饱和度:是HSV色彩空间中的概念,代表了一种颜色的纯度(Saturation)。

这里推荐使用qv4l2这款软件,可以方便的给相机调参并实时显示效果。

这是软件qv4l2的截图,v4l2是linux自带的相机驱动程序,可以看到有许多可供我们调整的设置

工业相机的配置则需要使用厂商提供的sdk,OpenCV也提供了一些修改相机参数的函数


4.1.3. 硬件参数

这些参数都是在选型的时候需要注意的,衡量了一个相机在可调参数上有多大的调节空间。

  • 靶面尺寸:该参数即感光元件的面积大小,值越大表明面积越大,面积越大进光量就越大,信噪比自然会相应提高,对于暗光环境会有更好的成像效果,还有其他种种优势,这也是所谓的底大一级压死人。使用靶面尺寸这个看起来无厘头的metric其实是有历史渊源的。

在CCD出现之前,摄像机是一直使用光导摄像管的成像器件感光成像的,其直径的大小,直接决定了其成像面积的大小。因此,后来大家就用光导摄像管的直径尺寸来表示不同感光面积的产品型号。直到CCD出现之后,也就自然而然沿用了光导摄像管的尺寸表示方法,进而扩展到所有类型的图像传感器的尺寸表示方法上。

例如,型号为“1/1.8”的CCD或CMOS,就表示其成像面积与一根直径为1/1.8英寸的光导摄像管的成像靶面面积近似。光导摄像管的直径与CCD/CMOS成像靶面面积之间没有固定的换算公式,从实际情况来说,CCD/CMOS成像靶面的对角线长度大约相当于光导摄像管直径长度的2/3。

常见的靶面尺寸其和实际面积的对应关系如下:

  • 宽容度/动态范围:大家常常会问一个问题:为什么我的相机拍出来的画面很暗,但是灯条中间仍然发白?这是相机的硬件特性动态范围或宽容度(这两个词汇在摄影领域常常被混淆,但是对于数字相机而言这两个词汇代表的意思几乎一致)所决定的。动态范围即感光元件能够记录的光强从最低到最高亮度的范围。最低即感光元件刚好有输出(或是当没有任何光线进入时产生的响应,即暗电流),会根据光线变化而产生电荷变化的阈值;最高亮度范围则是相机能够捕捉的最大累积光强,就像放大器一样,超过了其输出上限就会产生截断,此时即使继续提高曝光时间或外界光强增大,转换原件的输出也不会变得更大。因此上面的问题就出在相机的宽容度不足上。以高级的摄影器材为例,它们通常拥有超高的宽容度,下方是RM官方制作的数据集ROCO中的一张图片,可以发现画面整体明亮,但是红色装甲板表现地非常“红”,蓝色也很“蓝”,没有出现“过曝”的情况。
图源ROCO数据集,建议在新标签页中打开放大观看

动态范围在硬件实现上体现为单个CMOS/CCD单元能够容纳电荷的数量,光电耦合单元的容量越大就能保证曝光时间增大时不会发生“溢出”。摄影技术中常用的一种从软件层面提高动态范围的方法称作HDR(high dynamic range),这是一种一次性拍摄多张不同曝光的图片并利用算法对多帧图像进行对齐以及融合的技术。例如在逆光拍摄的时候光线来源区域往往会过曝,而背光处的物体常常“黑成一团”,这时候开启HDR,相机会采用两种曝光参数分别采样数次,高曝光图像能够捕获更多背光区域的细节,低曝光图像能够防止光线来源区域的感光单元溢出。通过自适应阈值技术自动为不同区域选择合适的图像进行填充即可得到一张完美的图像。

上方图片采用了6种曝光下采样的数据进行融合,左下角是融合的最终结果;图源http://bbs.a9vg.com/thread-5113593-1-1.html

因此在选购相机的时候,特别是对于传统算法,应该选择动态范围大的相机。动态范围一般以dB为单位,表明最大输入和最小输入之间的倍率关系。

  • 像元尺寸:顾名思义就是一个像素的实际面积有多大。在靶面尺寸相同的情况下,自然是拥有更大像元尺寸的传感器的感光能力更强了(合并后的像素可以采集更多的光线)。不过相应的,这自然会造成分辨率的下降。而相反,较小体积更小的像元尺寸可以做到更大的分辨率,不过由于单个像素的面积减小,CMOS/CCD单元保存电荷的能力也会下降导致动态范围无法提高,安放AD的位置也会缩小。

一般工业相机的SDK都支持合并数个像元(通常是水平合并 以及竖直合并 ,或者叫Binning),让用户可以合并相邻像素获得更高的画面亮度(binning也分为模拟binning[硬件支持?]和数字binning[利用算法进行插值],模拟binning由于DSP需要处理的数据变少了/CMOS转换的行列帧数减少,可以提高采集帧率)。还有另外一种降低分辨率(一般是为了提高采样速度)的方法是下采样,这种方法和像素合并不同,直接看下图就很好理解:

只选择深色的像素进行成像,白色像素不工作

特别注意,若选用相机的ROI模式或者直接修改分辨率,则会在cmos上裁剪一块区域用于成像,剩下的区域不会工作。这会提高相机的工作帧率,但是显然会减小成像的视场角,因为有效成像面积下降了。下采样则在分辨率下降的时候保持相同的视场角。

  • 信噪比:SNR(signal noise ratio)是真正的由光线引起的动作信号和噪声(电磁干扰、杂光、暗电流和散射光等)的比值,一般也是用dB来衡量。注意区分此参数和动态范围此参数越大,在增大增益的时候,噪点引起的干扰会越小。
信噪比和动态范围的对比,图源https://www.znjtech.cn
  • 像素格式:或者叫图像保存的格式/采样格式。常见的有这些:

  • Mono 8/10/12,黑白

  • Bayer RG 8/10/10Packed/12/12Packed
  • YUV422Packed,YUV422_YUYV_Packed
  • RGB 8,BGR 8 ;BGR 8是最友好的,采集后可以直接用cv::Mat处理,不需要额外的转换开销

对于工业相机采集得到的原始数据,OpenCV可以直接处理第一种和第四种,而Bayer格式和YUV系列的则需要经过手动转换或者使用相机自带的SDK进行格式转换,才能为OpenCV所处理(OpenCV似乎也提供了转换的API)。实际上大部分相机在拍摄时的格式都是Bayer RG,这是因为像素的实际排列并不是规则的RGB三种像素(一个像素单元实际上只能感受一种波长的光):

最常见的RGGB排列方式

既然绿色像素比红蓝像素多,其转换的时候必然要进行权重的矫正。对于没有对应颜色像素存在的地方,转换的时候肯定要进行插值,选择的插值方式(线性、双线性、泰勒)对于转换速度/转换质量的影响是比较大的。

如果能够直接让相机采集某种你需要的格式的图片,那么直接对相机进行设置,而不是采集到图像之后再于程序中对图像进行编解码转换等。相机内部一般有asic或fpga芯片,可以快速完成这些工作;而对于你的cpu来说,这是一个不小的负担。

当初华为手机在宣传其摄影功能的时候就提到它使用的是RYYB的排列以最大程度提高感光能力,不过缺少绿色像素,其成像即使经过算法矫正也通常偏黄。

  • 链接方式:即相机连接到运算平台使用的硬件定义和通信协议。对于工业相机而言,我们会用到的、最常见的有USB3和工业网口GigE,工业场合历史遗留或标准遗留的包括CamerLink和IEEE1394,由于通用性和可扩展性不佳正在被逐渐淘汰。CoaXPress和CameralLink这种超高速的接口,由于通用性受限、价格昂贵,我们一般用不到。现代工业相机一般还支持自定义的可编程数字IO接口,即下图中的圆形6-pin数字IO:
上方为USB3.0,左下方为数字IO

6-pin接口一般包括光耦输入和输出以及光耦地(这两路信号都通过光耦二极管进行电气隔离形成保护),电源和电源地,还有一路可配置GPIO(可以配置成输入或输出)。额外电源一般在USB接口供电能力不足或高负载工作的时候才需要连接,GPIO以及光耦输入输出三条线可以用于硬件触发、采集触发等功能,对于与其他设备同步采集时间戳是一个非常有效的解决方案。

USB连接线在相机一端一定要选用有固定装置的接头(相机上都有由于固定的螺孔),防止松动导致连接异常,损坏接口甚至由于不明原因的异常导致相机内部DSP损坏。USB和GigE虽然都支持热拔插,但还是尽量停止取流并断开相机后再行拔插,笔者使用海康威视工业相机的时候就常因为连接不稳数次后相机亮红灯,而一些机械结构的设计又让重新拔插分外困难。连接到运算平台的一侧,在确认机械结构不再改动后可以打上热熔胶或使用3D打印件进行固定防止松动。



4.2.运算平台

常见的运算平台有这几种:定制的minipc、Intel NUC、jetson系列、DJI manifold2(有cpu版本和gpu版本,分别相当于同配置的i7-8265u的minipc和jetson tx2,不过manifold的体积很小)、工控主板/工控机、OpenMV、k210、k510等。其实选型的空间并不大,不过需要大家根据预算平衡一下性能和价格。

4.2.1. 运算平台的性能指标

那么如何评价一个运算平台的性能呢,这里要提一下CPU和GPU运行的概念。CPU的时钟频率高,但是内部的运算单元(ALU、FPU等)数量有限,是为通用计算和程序控制所设计的,其实并不擅长进行大规模的并行运算,比较擅长单线程的流水线处理。GPU则相反,GPU有大量的低速运算单元,但是能够一次性处理巨量并行数据,因此尤其适合图形计算(相当于每一个运算单元计算一个像素的相关数据)。视频解码、单线进行的程序适合在CPU上运行;图形处理、显示渲染则适合用GPU进行。这也是为什么大家都说玩3A大作这些大型游戏需要一张好的显卡(GPU)。

GPU和CPU的对比;图源https://zhuanlan.zhihu.com/p/156171120

当然,也有专门用于各种运算的TPU、NPU、APU等等等等,他们都属于ASIC(application specific integrated circuit),直接将一些特殊运算(比如神经网络中的激活函数、矩阵乘法、非线性运算)或者软件功能固化在硬件中,以提高处理速度。

常听说的DSP(digital signal processor )其实也可以看作一种ASIC,如相机中用于采集cmos信号或专能于图像处理的芯片。


  • 一般用于评价性能的算力指标有GOPS(Giga Operations Per Second),MOPS(Million Operation Per Second),TOPS(Tera Operations Per Second)。若有F如TFLOPS,则是衡量浮点运算能力。英伟达官方也有一套衡量N家显卡机器学习算力的标准,参见CUDA GPU | NVIDIA Developer。若显卡的compute ability大于5,一般来说就比较适合进行机器学习的训练。(但是貌似官方没有给出定量的计算方法,可能是根据自家cuda内核数、tensor单元数和主频等参数通过一些加权方法得到的分数)

注意,不同类型的数据运算速度也不同,因为CPU、GPU等有专门的ALU或针对某些运算优化的指令集。常见的类型有INT8/16/32(8/16/32位整型)和FP16/32(16和32位浮点),也有较为少见的FP8以及一些”奇怪“的混合精度计算。

  • 在RM比赛中的识别算法中,传统的灯条匹配算法是更依赖CPU的,但是因为涉及到矩阵运算,OpenCV支持的一些图形库能够利用电脑GPU的能力进行加速(intel自家的ippcv以及OpenCV Contrib的cuda库)。而新兴的基于卷积神经网络的目标检测算法则是非常依赖电脑的GPU性能。不过,Intel推出的OpenVINO部署平台(仅仅支持Intel的cpu)和腾讯优图开源的NCNN(适合arm架构)推理框架都能够通过优化CPU的运算来提高神经网络的性能。英伟达的TensorRT推理框架则是只支持自家的GPU(jetson平台上也可以部署),能进一步提高神经网络的推理速度。

4.2.2. 设备介绍以及benchmark

为了提供性能的大致参考指标,这里介绍一下下文所说的Nanodet(我们使用的是Nanodet-m,320x320的输入,backbone使用的是shuffleNet V2)和传统算法。Nanodet是一款超轻量化的目标检测模型,是基于卷积神经网络的目标检测算法的代表之一,可以以此作为其他目标检测算法的基准(benchmark);下文提到的传统算法则是基于特征提取和灯条匹配完成的。我们在第五部分、第六部分会分别详尽地介绍这两个算法。至于传统算法,以i7-8265为例,按照一般的处理流程可以运行150+fps,若加入ROI则可以跑到300fps以上。而i7-1165G7则能实现1ms左右波动的单帧处理时间。

以下分别介绍一下各个运算平台:

  • minipc:搭载i7-8265u, i7-8565u的minipc较为常见,在tb上能够找到各种大小的minipc,至于那些搭载intel J系列和N系列的Genimi Lake架构的工控机、软路由、电脑棒、计算卡之类的玩意,是根本带不动神经网络的,如果是J1900甚至解码视频流都会出现卡顿。我们实验室曾经为了缩小体积购买了一台40x40x40大小的N4100 cpu的minipc,结果就是神经网络的目标检测大约10-12fps,传统的灯条匹配在不进行数字识别、环境光线简单的情况下也只有50-60fps。所以,从性能的角度考虑,至少是标准架构的i5-7代之后的cpu才能胜任RoboMaster赛场的视觉任务。

更新:现在有AMD锐龙的minipc下场了,参考配置为R7-4800H和R5-5600U,性价比都还可以,不过外壳都做的比较大,参考价格2000RMB。关键的是,Zen2+在完善了AVX512指令集后似乎也兼容OpenVINO,笔者的电脑(amd r5-5600H)可以正常使用OpenVINO 21.04LTS。

  • Intel NUC:8代和9代的nuc和上面提到的miniPC并没有什么区别,价格还要贵一些。而nuc10在nuc11登场后就成为了一个性价比比较高的选择,最近NUC11的价格也非常的不错。nuc11建议购买i5以上的配置,intel在第十一代cpu中挤了很多牙膏,用锐炬Xe显卡替代了万年不变的UHD630,这让nuc上运行神经网络成为可能。我们实验室同样配置了一台nuc11猎豹峡谷,cpu为i5-1135G7,运行传统算法能够达到250-350fps(全图检测),加入ROI检测后甚至处理一张图片只需要1ms左右的时间,速度不可谓不快。并且配合Intel OpenVINO推理框架,即使使用OpenVINO推理神经网络也能达到很好的效果。我们实测运行nanodet网络,使用Vulkan+NCNN/MNN/LibTorch对Xe显卡加速后能够达到50-60fps的速度,若运行OpenVINO推理框架则拥有约150fps的速度。选择i7系列的nuc11,并且配置更高频率的内存条(核显没有显存使用的是内存因此需要更高频的内存来提供访存带宽),应该能够有进一步的提升。11代最高支持3200MHz的内存。

参考算力:i7-1185G7的核显拥有1.9FLOPS的fp32算力。FP32推理YOLOX-n拥有50fps的速度。

推荐购买薄款,厚款其实就是增加了一个2.5inch的机械硬盘位,当然也可以购买厚款然后把外壳拆了让机械组的同学帮忙设计一个亚克力外壳或3d打印一个外壳。

  • 工控主板:现有派勤intel 11代的工控主板,相当于NUC的翻版,只不过砍掉了雷电接口和无线网卡,但是价格同样直接腰斩!(单单主板2000以内)散热能力稍逊于NUC11,其他方面持平。而且拥有板载串口、USBo1/2,自带2个网口,非常方便扩展。只需要自己用3d打印或板材拼接制作外壳即可,是替代NUC的不二之选。若后续推出AMD的工控主板,性价比应该更高。其他品牌也有推出NUC的平替,体积一般和NUC的厚款相似,可以拆卸外壳制作亚克力,tb上搜索有一大把,不再赘述。

  • Jetson系列:使用jetson系列主要是看重其GPU性能。目前能够使用神经网络检测达到实时性要求的设备,应该只有jetson tx 2(有些吃力)、jetson Xavier NX和jetson Xavier AGX。若使用Tensorrt框架部署后,Xavier AGX的int8算力接近1080ti,浮点算力和1070持平,是当下最强力的边缘运算平台。Xavier NX的算力是AGX的2/3~3/5左右,tx2就要更劣一筹。其大小相对minipc和NUC来说有比较大的优势,哈工深的同学自己定制了一块Xavier NX载板,进一步缩小了运算平台的体积。不过由于jetson搭载的是ARM64架构的cpu,可能在某些软件的兼容性上有一些问题,如conda等,好在大多数可以在网上找到解决方案或替代方案。GPU类型设备适合推理标准卷积架构。不过Jetson系列的CPU稍显孱弱(相比x86大核)。

不推荐使用 jetson nano、jetson tx1。

更新:Jetson nx以下的开发板现在应该是非常吃力!!Nano和TX2强烈不推荐,而且现在溢价非常严重。

更新:新版的Jetson NX orin和Jetson Xavier orin在2022的第三和第四季度马上要推出了,算力爆炸(当然价格也可能爆炸)。

更新:Jetson Xavier AGXorin已经上市,int8算力是Jetson Xavier AGX的八倍不止...... 目前经销商能拿到的价格是16999。

  • DJI Manifold:主要优势就是体积和重量,若不差钱且机械对体积重量要求高则可以选择这款。但是使用改造载板的Jetson Xavier NX应该在价格体积重量性能上完胜Manifold G系列了(没有催更DJI的意思,反正就算更新了也买不起)。

极度不推荐使用 Manifold1

更新:不再推荐使用DJI Manifold,刷机过于复杂,而且串口外设容易出现一些奇怪的bug。manifold2则已经停产,二级市场价格直接翻倍。求求快出Manifold3吧。

华硕旗下的研扬工控仍然在推出和manifold2C大小相仿,装载laptop级cpu的卡片式工控电脑。价格和妙算也类似。如果有极致需求的队伍可以考虑购买。

  • OpenMV:一般指基于stm32f7和h7系列,刷入openmv固件的嵌入式设备。32位架构的处理器算力在复杂的视觉处理场景明显不足,openmv相对其他的运算平台优点就是体积非常小非常轻,并且编写其程序使用micropython,开发的速度很快。openmv也可以为机器人增加辅助视觉功能,提供一些不那么复杂的算法处理工作。

  • k210/k510:勘智科技推出的人工智能处理芯片。特点是体积小,非常小!现在市面上能买到用micro-python开发的maixpy出品的k210芯片。AI算算力在同价位和大小属于独一档。特别提一下最新推出的k510芯片,算力到达了2.5TFOPS,超过了jetson tx 2的2.1 TFPLOS!如果队伍内的电控组或硬件组有能力的话,可以让组内的同学根据datasheet制作pcb板,进一步缩小运算平台的体积。nihui大神已经成功把ncnn移植到RISC-V上(即k210和k510的处理器架构),如果后续能移植linux mint等微型linux系统到k510上(机械狂喜ohhhhhhhhh),将会大大便利在这些soc上的开发。并且这些芯片同时提供了低速IO外设,从这里也可以看出它们是专为嵌入式边缘应用打造的。k210适合作为机器人的辅助视觉单元,而k510的超强算力(同体积重量来说)甚至能够胜任整个机器人的控制!不过由于其cpu相比其他平台还是稍显孱弱,除非能在视频解码上进行优化,否则帧率可能会受限。

更新:K510已经上市,参考价格为999RMB,目前只有勘智官方自己做的开发板,体积比较大。

  • FPGA:全称为Field Programmable Gate Array,是可编程逻辑器件(PLD)的进化产物。要通过CPU这种固化的数字逻辑电路实现不同功能,就产生了编程语言和代码,利用复杂的软件实现对单一硬件的多功能扩展(ALU只支持那些在设计之初被固化在硬件中的操作,如加法器,乘法器等)。而FPGA则是让你拥有直接对硬件编程的能力,你需要什么样的运算,就能使用硬件描述语言(HDL,hardware description language)直接构建对应的运算逻辑器件。使用FPGA的时候,你就是芯片的设计师!当然,现代FPGA为了方便使用,常常在外部挂载一些通用性非常强的外设如串口、CAN、SPI等。FPGA相比CPU、GPU等通用运算芯片,最凸出的优势就是硬件级的大规模并行和几乎零额外开销的算法实现。HDL的编程范式也和软件编程语言有很大的不同。

笔者没用过FPGA,之前的课设都是找同学帮忙做的。如果你对FPGA比较了解,可以为本小节贡献内容,提出一些选型建议!(笔者的FPGA课设仿真激励文件就是这位大佬帮忙写的,感谢万能的群友哈哈哈~)

  • 各种Linux/Android开发板:例如全志和瑞芯推出的各类arm64架构的开发板,火爆的RK系列,以及荔枝派、香橙派各种派等树莓派的平替。arm64对于ncnn、mnn等推理框架的支持较好,适合推理深度可分离卷积的模型。不过目前似乎没有队伍尝试使用此类运算平台。
各个运算平台大小的对比,从左起依次是k210、openmv、某intel N4100的minipc、Jetson nano、jetson Xavier NX、i7-8265u的minipc、intel NUC11猎豹峡谷、intel NUC7 Bean canyon(豆子峡谷)
  • 有些经费比较宽裕的队伍可能会专门搭建一台用于队伍网络服务的服务器+深度学习训练机。若预算不足,则可以选择在一些提供云服务器训练的商家购买GPU核时进行训练,价格也不算太贵。现在的显卡价格实在过高,不适合自己额外购置显卡,如果有钱当我没说。(挖矿的都给爷爬)

这里推荐使用搭载较好显卡的游戏本或者intel NUC 11幻影峡谷:i7-1185G7+RTX2060的配置进行前向推理还是没有任何问题的,平时若有需要还可以用于训练小模型。

22/Aug更新:30系显卡已经跌破发售价,刚需的小伙伴可以入手了,从此炼丹再也不用求人了(没事的时候还可以试试RTX ON)。

再更新:以太坊解算制度修改,比特币算力翻番,现在大家都用专用asic矿机挖矿,显卡终于自由。Pytorch也在最近释出ROCm支持的正式版,锐龙显卡

之前在RM视觉交流群 就有小伙伴提出大家可以利用各自手里有的运算平台测试各种网络和传统算法,制作一个modle-zoo,如果有想法的同学或者已经测得数据的小伙伴可以戳我,让我们一起完善这个表格哦!

  • 2023增补:现在小主机突然火爆,利好机器人领域!小米、华硕、机械革命、零刻等纷纷推出自己的小型主机(Intel退出中止NUC业务,交由华硕维护)。


4.3.IMU

IMU对于RoboMaster比赛十分重要,是电控实现“小陀螺”、控制平衡车的基础,也是视觉算法预测是否准确的根基。我们需要云台和机器人的姿态数据,以实现对目标在绝对系下的准确位姿测量。更多关于这部分的内容在第六章击打预测和弹道模型将会介绍。

全称为Inertia measurement unit,惯性测量单元。一般包括加速度计和陀螺仪,气压计、磁力计等也可能被一起集成。这里主要介绍MEMS IMU的原理和对其输出数据的处理。

MEMS:micro-electronic-machine system,微机电系统。它是集微传感器、微执行器、微机械结构、微电源微能源、信号处理和控制电路、高性能电子集成器件、接口、通信等于一体的微型器件或系统。下面随笔者一起看看介绍MEMS IMU的结构,你就能理解上述定义了。

  • 工作原理

先看看微机电陀螺仪的实现。我们知道加速度的来源是力,因此需要通过某种物理效应将力转化为方便测量的量,对于现代传感器来说,大部分选用电作为信号载体,因为相关的信号处理技术已经非常成熟(而新的发展方向是利用光,其功率更低载波频率更高)。考虑下面的一个质量块弹簧阻尼系统:

质量块M在受到力F的作用的时候,将发生运动;由于阻尼器的存在,质量块最终会在新的位置停下来:这个位置满足。学习过自动控制原理/电路的你应该知道,可以用微分方程/传递函数描述这个标准的二阶系统。只要K是已知的,我们就能根据位移量x确定力的大小。相信此时你已经有了不少方案,最简单的一种也许是在质量块上搭载滑动变阻器抽头根据电阻分压确定位移。不过,为了让我们的传感器足够小,小到能够放进一块芯片当中,并且对外部的加速度有足够的敏感性,我们采用如下的方案:

MEMS加速度计的原理,实际上有3组这样的器件,分别置于三个两两垂直的方向;图源HowToMechatronics.com

绿色的小条是电容的极,一个质量块被固定在两个微小的弹簧上。当受到加速度的时候,质量块由于惯性将会受到反向的“惯性力”,此力的大小就等于加速度乘上它的质量;在短暂的过渡过程结束后,质量块将停止在某个位置上,此时其位移就满足我们前面得到的公式。我们选用检测电容的方式确定质量块的位移,质量块两边延伸出的长条是导体材质的,将和绿色小条形成电容。根据电容和极板距离的关系:求得距离的变化量,再根据得到力的大小,最后由求得加速度大小。

之所以要“两片夹一片”的电容构造,是因为使用了传感器中常用的提高灵敏度、减小非线性误差的方法:差分感知。学习过传感器相关课程的同学肯定不陌生。实际上MEMS加速度计中包含了数十组这样的差分电容片,通过求平均的方法降低随机误差。

再看看MEMS陀螺仪,高中的时候就学过物体在旋转时会受到向心加速度,可以据此原理进行检测。但是在设计上太难实现,而且很难区分旋转产生的向心力和物体受到的其它力。因此,我们通过间接的方式对旋转进行测量:科里奥利力。如下图所示,考虑一个光滑的竖直圆盘和一个自由下落的小球,由于参考系的选取问题,若一个物体在转动参考系上发生径向运动,转动参考系上的观察者会发现物体实际上走过一道弧形。因此在旋转系下看,物体受到了某种虚拟的惯性力作用。我们就可以利用这个虚拟的力对旋转进行检测。由公式推导可以得到这个惯性力的大小为,为旋转系的角速度,m是物体质量,V则是物体在绝对系的速度。

上方为惯性系视角,下方为旋转系视角(红点不动)

那我们的设计思路就是,让一个质量已知的质量块以某个速度运动,并以适当的方式检测出惯性力的大小,从而推得旋转速度。像上图小球那样一去不返的运动显然是不可取的,我们通过对质量块施以周期性的力使其在限定范围内往复运动,它的速度是可以获知的。然后,在其内部再添加一组运动方向和前者垂直的质量块(内部的结构和加速度计相同,即让惯性力反映为位移大小),一旦发生旋转,内部的质量块就会受到科氏力从而发生位移,通过电容的变化将检测出位移大小,从而推算惯性力的大小,再由得到物体旋转的角速度。

MEMS陀螺仪的原理,这里同样只展示了一个轴;图源亚诺德半导体

可以发现,上图中内部用于检测科氏力的加速度计装置和外层框通过弹簧进行了隔离,这就能在很大程度上防止装置在受到纵向的加速度时产生误动作。

这是MPU-6050的晶片显微镜视图,依稀可以看见两个传感器的排布

关于两个传感器工作原理的详细介绍,可以参考MEMS 陀螺仪工作原理和MEMS 加速度计工作原理,都是Analog公司的科普作品,非常详细。

若集成了气压计,IMU还可以检测海拔高度从而与加速度计在z轴方向上的数据融合,提高精度。引入磁力计则可以通过检测地磁的方向以确定IMU的位姿,与陀螺仪融合。

RoboMaster 开发板C型(C板)上集成了一颗博世生产的六轴IMU(即三轴加速度,三周角速度)BMI 088传感器,并内置了恒温加热电路和减震海绵垫。同时,C板旁还有一块iSentek生成的三轴磁力计芯片IST8310。

  • 数据处理

温度常常是影响各种传感器精度的一大因素,因为材料的物理化学性质基本都会随着温度发生变化。C板上集成的加热电阻可以很好地控制温度保证温差的影响被降到最低。可以使用PWM+PID对电阻发热进行控制。虽然陀螺仪通过科氏力检测角速度已经最大程度降低了加速度对检测结果的影响,但难免还是会有检测误差。陀螺仪最常见的误差类型就是零漂,即IMU在禁止状态下仍然会产生一个很小的角速度输出。因为我们一般通过对陀螺仪的输出进行积分以获取位姿,因此零漂将持续引入误差,最终累计到不可接受的程度。加速度计也有量程的限制,并且对阶跃输入、撞击很敏感,在外力频率极高或幅值大的情况下会输出奇怪的数据,同样也需要利用某些手段进行矫正。

实际上加速度计也存在零飘,以及两种传感器都有随机游走的噪声。已经有一些开源的工具包可以对imu进行动态和静态标定。一种有效的分析噪声和误差的方法被称作Allan方差标定,感兴趣的同学可以自行搜索。不过在电控场景下,标定陀螺仪的零飘和imu的放缩系数一般来说已经足够。而对SLAM场景,一般需要与其他传感器进行更高精度的联合标定。

对于六轴姿态融合解算,一些经典的算法包括Complementary(互补滤波)、Mahony、Madgwick。它们都利用加速度计的数据对陀螺仪的数据进行修正(因为总有一个重力加速度,因此加速度计能较好地估计静止状态下的姿态)。要学习位姿解算,你首先要拥有坐标变换和旋转描述的知识,我们在5.6.1提供了最基础的必要工具,并在5.6.5增加了四元数和欧拉角相关的扩展以供选用。

这里我们暂时不考虑磁力计和气压计,这是因为机器人附近的电磁干扰太大:电机、超级电容等都会产生较强的磁场,导致引入磁力计后姿态解算效果反而更差。对于地面上工作的机器人,高度几乎不怎么改变,气压计谓之鸡肋。

当然,有很多开源库已经实现了大量方法。AHRS库包含了常用了各种位姿解算算法,可以轻松通过pip安装Attitude and Heading Reference Systems,它的文档非常详细,对于算法原理的介绍与模型调用、接口定义等的描述一应俱全(遗憾的一点是只有python的实现,不过理解原理之后你就可以用高性能的语言对其进行转写了)。Fusion是Mahony解算的官方实现,加入了磁力计(你也可以选择去除)融合,提供了C/C++的接口,不过也可以使用python调用。RoboMaster C型开发板的姿态计算例程,使用的就是Mahony算法(但听说好像有些地方存在小bug?)。

状态估计/控制算法实践这个专栏较为详细地介绍了姿态解算中用到的知识和Mahony解算,以及作者自己实现的EKF四元数融合算法,在有了卡尔曼滤波器、PID算法和四元数的基础之后,推荐阅读。



4.4.激光雷达

雷达(Radar,RAdio Detection And Ranging)是一种通过发射电磁波并根据回波强度和时间判断物体距离的一种传感器。那么激光雷达的工作媒介就是激光(故称LIDar,LIght,实际上激光也是电磁波)。在自动驾驶大热的今天,激光雷达、毫米波雷达、IMU、相机和GNSS系统可以算是Auto-driving界的五虎上将(有时候还包括超声雷达)。根据工作原理的不同,可以将雷达分为TOF(time of flight,飞行时间法)和FMCW(frequency modulated continuous wave,调频连续波)两种;根据工作范围也分为点式、线阵和面阵三种。还有其他的分类方法如根据扫描方式和发射/接受方式分为机械旋转、MEMS、转镜、棱镜、Flash等...... 我们主要介绍工作原理的分类,并分析在RM比赛场景下激光雷达可能的用途。

Livox MID-70激光雷达;参赛队伍可以以优惠价格购买供开发使用
  • 工作原理

TOF的原理非常简单,雷达发出光脉冲并检测反射受到脉冲所用的时间,乘以光速便得到距离。为了保证收到的反射是自身发出的,一般激光雷达会发出特定编码的脉冲序列(相当于某种“通信协议”),接收时要检测到该序列才视为对准。(这也是不同厂家、不同批次的雷达防止“串号”的原理)

FMCW的原理稍微复杂一些,雷达通过发送和接收连续某种频段激光束,让回光和本地光发生干涉,并利用混频探测技术来测量发送和接收的频率差异,由发送和接收波的频率差得到距离。此外,FMCW雷达还可以测速:利用多普勒原理,激光束击中目标物后被反射,而反射会影响光的频率——如果目标物向车辆走来,频率会升高;如果目标物和车辆同方向行走,则频率会降低。当反射光返回到探测器,与发射时的频率相比,就能测量两种频率之间的差值,从而计算出物体的距离信息。更详细的介绍参见:详细分析FMCW雷达测距/测速原理,需要有信号与系统/数字信号处理的基础,或者了解调频和调幅的原理。

  • 参数

这里以Livox MID-70(棱镜扫描式)为例激光雷达的一些关键参数包括量程、近处盲区、FOV(field of view)、测量误差、光束差角等。量程一般以回波强度至少为100klx记,对于不同反射率的物体有着不同的量程。由于当激光脉冲出射的时,碰到镜头将发生反射,这时候回光接收器将检测到此反射波从而出现误动作。为了避免误动作,一般加入一个不响应的死区时间。如果障碍物距离太近,接收器仍处在死区内就会使得近距离物体的脉冲回波无法被探测到。激光雷达的盲区范围基本都小于0.4m,因此对于大部分应用不用太过担心。FOV是面阵雷达的常用参数,和相机一样表示可视范围。测量误差主要是随机误差,一般小于5cm。光束差角则是衡量雷达的(角)分辨率的参数,即发出的光线的致密程度。Livox官方对其雷达的具体原理和所用技术进行了科普性的说明,讲得很不错,戳这里:解密Livox激光雷达原理系列(高低要吹吹牛对吧)

  • 数据处理

从激光雷达的工作原理我们很容易得知,采集得到的结果一定是一堆点,每个点都记载着和传感器的距离和角度(回忆FOV的概念,接收器会接收不同角度射入的回波),而在实际中为了方便处理一般转换成以雷达为原点的笛卡尔坐标。数量众多的点就形成了点云,因此数据处理方法主要式点云处理算法。我们在5.6.6会大致介绍一下点云处理,前提是有二维图像处理的知识(或你已经有相关背景)。Livox官方也提供了开发SDK和一些开源的算法实现,包括点云去畸、配准、雷达自动标定、相机联合标定、车道检测、点云目标检测等,戳这里:Livox-SDK。

配合IMU,激光雷达在城市道路上生成的点云图;不同的颜色代表采集该点云时与激光雷达的距离
  • 赛场应用

和IMU配合工作,激光雷达可以搭载在机器人、汽车上移动并实时获取点云信息,根据位姿的变换将测量结果投影至三维地图中,完成三维重建、同步定位与建图(SLAM)等。和相机配合工作,激光雷达与相机进行联标后,可以确定RGB图像上每个像素在点云中的对应点,从而获取其深度信息以进行精准测量。而雷达系统(此处指的是兵种)中使用激光雷达也是非常不错的选择,毕竟LiDAr可以提供相机视觉系统无可比拟的测量精度。另外,对于哨兵机器人的自动运行来说,激光雷达几乎是必不可少的。

在SLAM部分的介绍中,我们会进一步了解和激光雷达相关的算法,以及如何标定激光雷达与其他传感器等。

大赛组委会还为激光雷达专门设立了创新开源奖,如果你能想出更多有趣的应用,在取得出色效果的同时,说不定还能拿下额外的奖金!



4.5.特殊相机

  • 双目相机

双目相机应该是最早出现的也是最成熟的立体视觉方案。利用两个相机的视差和他们固定的相对位置信息,通过3d数学的几何解算方法计算对应点之间的位置偏差,得到物体的深度信息。关于双目相机的测距原理,请参考《机器人视觉测量与控制》这本书。这篇教程较为详细地讲述了双目相机测距的基本原理和用法。

双目相机

OpenCV也提供了双目标定的例程和大致的原理说明。同步双目相机(两个相机在一块pcb上,共用一个相机号,在Linux系统下只有一个文件描述符)的标定要比两个单独的相机更容易标定,若使用两个单独相机也需要尽量让参数保持一致以便得到更简单的解算模型。

双目相机通过视差由相似三角形得到的测距精度显然是要比单目高的,不过解算的开销要远大于单目。但是提升的精度对于我们补偿算法的影响其实并不大,而且高精度也带来对误差敏感性的提升。单目相机在目前的RMUC赛场是应该是足够满足我们的需求了。不过,了解双目标定和解算的原理对于理解3维重建、学习3d数学与线性代数是有很大的帮助的。我们会在 6.3 中详细介绍距离解算和三维重建的内容。

西北工业大学WMJ战队在2021赛季采用了双目视觉,可以看到他们的哨兵云台拥有一个像《机器人总动员》中WALL·E一样的长条形大脑袋。不知道他们会不会开源他们的双目视觉解算方案呢~

  • 双光相机

可以采用NIR(近红外)+RGB的双目相机组合,这在目标检测领域这是一个新兴的方向。红外成像可以弥补低可见光照条件下的噪声影响和细节信息的丢失,若采用这种组合则可以进行优势互补。机器人的电机、超级电容、电源管理模块发热量都较大,在近红外相机上会有明显的响应;而荧光充能模块的发热量更是惊人。

最简单的利用方法是直接把NIR采集的图像追加到rgb之后,将神经网络的输入变为四个通道,让后续的层自动抽取特征。我们也可以手动添加先验,使用类似transformer的方法给予RGB抽取出的feature map上NIR通道有高响应的区域以更高的权重(乘以一个系数)。

对于传统的handcrafted方法和装甲板检测,红外通道应该帮不上忙。而且这种相机很难用于测距,因为两个相机采集到的特征不同,我们无法在RGB相机上寻找NIR相机的特征点(关于这部分可参见本教程的双目相机和3D视觉部分)。

遥感卫星和航拍无人机上会搭载一些高光谱相机,拥有多个不同波段的镜头,可以同时获取各种光谱信息进行分析,在农业、防灾和测绘领域有广泛应用。

  • 深度相机

生产深度相机的规模化厂家目前主要有4个:ZED、intel Realsense、奥比中光、Kinect(还有和OpenCV合作,最近主推的Oak)。大部分厂商采用的解决方案都是双目➕TOF/结构光的方案,相机内部的DSP芯片会融合多个信息来源的图像进而结算得到深度图。使用深度相机的时候,厂家一般都是对其进行了彻底的封装,我们对于其原理只需要有大致的了解即可。

在第五章三维视觉基础的介绍中,我们会更详细的讲解不同方案深度相机的深度估计原理。

intel深度相机拍摄得到的深度图像,像素的不同颜色表示距相机的距离不同

在上图中由紫色到红色表示距离的远近,黑色则是因为遮挡或相差导致双目相机无法解算生成的深度空洞,深度相机一般也有一个空洞率的参数,越低越好。

相机提供了相应的SDK供我们进行二次开发,调用对应的函数接口就可以得到一幅深度图,每个像素的value即为相机视野范围内对应点和相机的距离。不过商用产品的闭源特性让我们很难进行深度DIY(深度相机还是不够有深度呀~)。由于需要内置DSP对图像信息进行实时解算,限于体积和功耗,目前市面上的深度相机帧率多少都有些不足,最高的为intel D435,能够在800x640的分辨率下达到90fps的速度(深度图90Hz,而RGB光学相机的采样频率是60Hz故同步时也仅有60Hz)。

Intel D435深感相机,采用usb3.0-typec连接
笔者所在学校的实验室居然还有一个kinect Xbox360体感相机,确定不是买来玩游戏的吗...

深度相机由于开发方便,赛场场地小,经常被用作视觉SLAM中的一环,有些深度相机同时集成了IMU(惯性测量单元),使得机器人的位姿估计可以轻松进行。

  • 事件相机

事件相机是一种新兴的仿生传感器。与一般的曝光成像相机原理不同,事件相机的像素只有3种输出:-1,0,1且每个像素都是独立工作的。0表示这个像素的亮度和之前相比没有发生变化,-1表示这个像素之前是被激发的而当前变成了未激发,1则和-1相反。由于其只检测像素的亮度的变化,每个像素需要的曝光时间都非常短,因此有动态范围极高、事件无运动模糊等特点。其简要工作原理如下:

(a)中的具体电路设计则是:

这里C2的作用是和C1一起调节放大器的增益

左侧的光感受器能够将光电流大小按对数比例转化成电压,中间的差分电路用于计算两次采样的电压差,输出到右侧的比较器电路,当压差超过阈值则完成一次触发,触发方向取决于压差的方向,是相比之前升高还是下降。单个单元的工作的流程如下,当一次事件触发之后,差分电路的reset开关会闭合,使得C2和放大器被短路,导致C1接地(上图没有画出,前面的图(a)中的reset则连接了地),其两端的电压保存为上一次触发的电压;随后reset断开,新的一个周期光电流产生电压,由于C1没有放电通路,C1两端的电压应该保持不变,此时C1左侧电压变为,为了保证电压不变C1C2之间的结点电压则要变为,从而使得C1的压差仍然维持在。这样,两次采样的电压差被反向放大器放大之后输入比较器,以判断是否超过正/负阈值。C2的作用是和C1一起形成差分电容放大电路,用于控制在C2右侧输出的电压即输出电压和C1C2之间节点电压即输入电压的增益,若你学过模电便知道这是一个经典的反馈放大器,可以自己列式计算增益。因为不同单元的放大器的增益不可能完全一致,因此通过设置的值,可以放大倍率一致,防止出现不同像素触发门限不同的情况(when in trouble, use feedback!)

相比之前介绍过的RGB相机(或其他expose-based camera),事件相机单个像素的结构要复杂许多,因此分辨率相比其他相机有较大的劣势,目前主流的事件相机分辨率在3MP以下,超过该值的品类价格则非常恐怖。

感谢Dr.yyy对事件相机电路的分析。

当触发事件时,事件相机后续的逻辑模块会记录这个像素的坐标、激发方向(-1or1)和激发时间。每次激发,事件相机都会立刻将数据反馈给上位机(当然也有累积一段时间的事件然后发送整个事件包的,和激光雷达非常类似)。由于亮度没有发生变化的那些像素不会产生输出,因此最佳的观察事件相机生成的“图像”的角度是把时间加上,从的三维视角去分析其产生的数据。

事件相机常用于高速物体运动估计、动作捕捉,暗光环境的动态物体观测,以及一些新兴场景的SLAM。



4.6.低速IO外设

在处理完图像并得到装甲板、能量机关扇叶相对相机中心的角度或找到救援目标和矿石之后,我们要设法和电控通信,让下位机能够接收到数据,告诉机器人目标在哪个方向、敌人处于什么状态,进而控制电机、气缸等来执行我们的任务。同时我们也需要从电控获取一些信息,如敌我的颜色、当前工作模式(自瞄、激活能量机关还是手动)、裁判系统读取到的各种状态等。以下介绍两种通信方法和对应的协议。CS专业、了解过计算机网络或学习过嵌入式开发的同学对此应该不陌生。

4.5.1. 串行通信

常用下位机通信方法之一串口(Serial Port)。为了方便简单起见,我们使用的都是异步串行通信,通过一个usb转串口芯片将解算好的数据发送给电控,电控经过一定处理后再控制电机动作。

  • 串口通信的基本知识:在最简单的串口使用中,通信需要两条线路,一条用于发送,一条用于接收(其实还需要一根共地连接线)。顾名思义,串口通信就是将数据像“串”一样一位一位的顺着一条线发送出去。任何通信方法都需要制定通信协议,串口通信也不例外。异步串口通信中需要通信的双方提前约定好波特率、数据位、停止位、奇偶校验位。因为使用的是最简单的两线通信,需要关闭硬件流控制。关于串口通信的更多基本信息,请参考What is Serial Communication and How it works?

  • 串口在Linux上作为一个设备,同样会以文件的形式挂载在 /dev 分区,使用时需要赋予串口权限:

sudo chmod 777 /dev/ttyUSB0 
#当电脑只连接了一个串口时,将挂载为ttyUSB0,对于一切皆文件的LINUX,其实和赋予文件的读写权限差不多

若不想每次使用都输入一次,希望永久赋予串口权限,参考这个教程:永久打开串口权限

  • 串口通信协议:通信发送开始时,先发送一位0表示开始发送,紧接着是8位的数据(相当于一次发送一个byte,低位在前),然后是一位奇偶校验位(如果开启此功能),最后是一位停止位1(可以设置不同的位数1.5、2)。以上便是串口通信的一个数据包。显然单单通过这样简单的协议(一个数据包传送8位数据,最多可以表示256种状态),我们无法完成复杂的通信功能。因此,在此基础上我们定制一套自己的通信协议以完成数据包更大、数据类型更复杂的通信,形成一个简单的协议栈。

  • 一个简单的协议需要包含帧头(标明数据包的开始)、数据内容(需要传送的数据)和帧尾(表明一个数据包结束)。

  • 在数据包中可以增加用于数据校验的校验码,通信中常使用的有CRC(cyclic redundant check)、奇偶校验、和校验、哈希校验等。
  • 高层的通信协议(在这里为我们自己制定的通信协议)是以底层协议为基础(这里为串行通信协议),如典型的、当下最复杂的网络——计算机网络,就采用了5层通信协议栈(也有7层的说法)。

  • 刚刚入门没有看懂或是想要更具体的实例,请参考这篇文章:在串口通信的基础上定制一个简单的通信协议-NeoZng

这是一个USB转TTL串口模块,由seasky-lw设计

4.5.2. CAN\&总线通信

和直接的点对点通信不同,还有一种常用的工业通信方式的分类为总线通信,即多个设备挂载在一条线上。大家最熟悉的总线莫过于USB总线(universal serial bus)。但USB的协议过于复杂,包括了物理到消息包的多层协议栈,这里我们简要介绍最常用的CAN和485。总线分为两种,无主机和有主机,区别在于有主机总线上的通信只能由主机设备发起。

被广泛使用的无主机方法是CAN(Controller Area Network)通信,CAN是一种由博世开发的通信规范和协议,是一种应用广泛的现场总线,在工业测控和工业自动化等领域有很大的应用。其最大的特点就是稳定性(使用了差分通信的方法)和灵活性(设备只要挂载在总线上就可以使用,不需要额外的连接)。我们在比赛中大量使用了DJI生产的电机,这些电机都非常的智能,通过集成了mcu的电子调速器,我们可以跳过下位机器,直接通过CAN与电调上的微控制器进行通信,将控制信息直接发送给电调从而控制电机的转动。当然也可以把信息通过can发送给电控再由下位机对执行单元进行控制。

  • 想要让运算平台支持CAN通信,有两种方法:

  • Jetson(nano除外)、树莓派等主板上自带 40 pin接口,只需要一个CAN收发器即可实现CAN通信。

  • 购买一个USB转CAN转接器,也可以实现CAN通信。

希望学习具体的例子,在Linux上利用CAN进行通信,请看这里:在Jetson Xavier NX上实现CAN通信并控制3508电机-NeoZng

  • CAN通信的基本知识:和串口一样,必然也需要一个通信协议。不同的是,其信号是通过其总线上电平的差值来表示的,这样能够有效抑制共模信号(因为噪声对两条线的影响通常是一样的,相减之后噪声的影响便被消除了),因此一般采用两条通信线。不过也最好连接地线,共地可以最大程度降低干扰。还有4-pin的CAN线,额外的一条线用于独立供电,适用于对信号质量特别高的场合,此电源专门为CAN收发器和信号电平供电。

  • 由于采用了总线结构,数据的传输只能是分时复用地,其通信机制更为复杂一些,故不在此赘述,想要详细了解其硬件构成和通信协议,请参阅一篇易懂的CAN通讯协议指南。

  • 注意CAN的带宽有限,常用的有250k、500k和高速的1M,不要在一条总线上挂载过多的大流量设备,否则会出现丢帧、乱码等情况。

  • 每个结点的ID必须独立,CAN协议中的仲裁顺序和send-ack机制就决定了总线上的不同设备需要设定不同的id。新人很容易犯这样的错误:两个结点(常常是想要同步转动的电机)设置相同id,这样就能发一次信息让两者同时接收。

一个CAN收发器 而RS485是一种常用的串行总线,总线上有一个主机和若干从机,当主机发起通信后,从机会检测总线上的电平,若通信id和该从机匹配,便会开始和主机的数据传输。RS485属于半双工总线,由于通信使用差分信号,同一时刻只有一个发送者和一个接收者。在机器人领域中有些电机会使用RS485作为通信手段。

还有通过网口连接的工业标准总线EtherCAT,对TCP/IP协议改良之后满足了IEEE和ISO的一些工业总线标准,常用于机械臂和流水产线的通信。近年来也逐渐应用到机器人上。相比CAN和485,EtherCAT的功能多,扩展能力强,带宽远超传统的低速总线。

对于其他的总线通信方法如SPI、I2C等,其实都大同小异,只要了解了通信的基本原理,都能够很快上手对应的API。感兴趣的同学可以自行搜索学习具体的软硬件实现。

说到和电控方面的通信,这里不得不提一下广东工业大学的视觉控制一体解决方案:rm-control,他们机器人只使用了一个运算平台,将上位机下位机合二为一,通过CAN控制电机。这消除了minipc到单片机通信产生的延迟,并且有良好的兼容性和复用性,维护起来也很方便。并且他们还研发了一款应该是市面上体积最小的usb转can模块。很佩服第一个有这样的想法同时又将其付诸实践的队伍(廖佬nb!)。




5.比赛中的CV算法

讲了这么多,视觉组的重头戏——算法终于来了。

在大部分时候我们都不需要设计底层的算法,而是直接调用封装好的API,设计更具体的应用于特定问题的算法。当然,有必要了解一下造轮子(底层算法的实现)的过程,这能够让我们深入理解算法内部的构造,从而更好地使用这些算法,出错的时候也能更快定位问题。如果只是调用API而不了解原理,那么只是简单的缝合+搭积木,对于提升自我的思考能力和逻辑思维没有任何帮助。应当要有“使用科技的黑箱会使我惶惶不安” 的觉悟。

我们最常用的OpenCV和一些神经网络模型都是开源的,它们都有优秀的注释和说明文档,尤其是OpenCV的Documentation和Tutorial十分详细,全是使用doxygen生成的标准文档系统。通过阅读这些材料,很快就能上手。在GitHub社区你可以提出Issues,和其他开发者一起讨论问题。

5.0.CV的常识性概念

计算机视觉是让机器拥有视觉同时让机器能够理解所看到的东西并对其进行一定的分析和处理的研究领域。目前主要分为图像识别、图像分割、图像生成、目标检测、目标追踪、视频处理 等,因为有着共通的根基和大量知识交叠,其实很难将他们分得太开。此部分就主要介绍最基本的概念:

  • 图像的构成

  • 像素:像素是构成图像的基本单元,像素通过行列组合成矩阵就形成了图像。在计算机中一般都以矩阵的形式存储图像。若仔细观察你的手机或电脑屏幕,应该能看到微小的由红绿蓝三色组成的发光单元。当像素密度足够高,人眼就会认为一张图片是连续的了。在图像处理的过程中,我们常把图像视作一个二元函数(如果是灰度图的话),在两个坐标轴上,亮度随坐标而变化。当图片是彩图时,下面会介绍颜色空间的概念。

一张皮卡丘的图片,由一个个像素点构成。
  • 像素深度:自然届中的颜色固然是连续的,但是在计算机中存储的数据是离散的。存储一个像素所使用的位(bit)数叫做像素深度,可以看作图像在某一点取值的值域。常听说的8位宽颜色就是用8位数据表示一种颜色,存储一个像素使用的位数越多,其能保存的信息就越丰富,主要表现在能显示的色彩的数量和对比度上。
各中位深度图像的对比,显然24位深度的图像最能还原真实的场景,逼近连续的情况。不过你的显示器未必支持高位色深(一般来说10位已经很大),这里只是展示效果

图片源自csdn-丁香树下丁香花开,侵删

  • 通道数和颜色空间:当一张图像只有一个通道的时候,他只能表示一个维度的信息,比如这张灰度图,在唯一的一个亮度(灰度)通道中保存。
一张以灰度图形式展现的英雄机器人

当一张图片想要以彩图的形式保存,它至少需要三个通道,即Red Green Blue(RGB),每个通道都是一个矩阵,矩阵中的每一个元素保存着0~255的值,对应不同的颜色分量。仔细观察下图就会发现,原图中呈现蓝色的部分在蓝色通道中比其他通道要更亮,比如机器人后轮下方的一片蓝光,在红色通道中就几乎没有分量存在。

图片被拆分为三个通道

除了RGB空间,还有其他不同的颜色空间如HSV、YUYV、LAB等,他们都是把图片投影到不同的空间中,图片在这些空间中的每一个坐标轴的投影,就是它在这个方向上的分量(和线性代数中概念的具象)。

RGB空间的坐标轴-来自百度百科
HSV空间的坐标轴(柱坐标系)
  • 图像的压缩编码方式:为了达到减少空间占用的目的,我们会通过某种算法将图像进行压缩。存储图片时格式一般有bmp,jpg,png,tif,gif等。不同格式图片的解码速度和占用空间大小不同,有时候甚至是算法时间占用中的关键一环。

  • 视频

视频就是连续的图像集合,从另一个角度来看,可以把时间当作除x、y外的另一个坐标轴,看作是一个三维的函数。如果选取一个确定时间,则视频退化为图片。最简单的视频保存格式和图像的压缩编码相同,而高级一些的压缩算法会根据两帧或多帧内容的相关性,找到关键帧和相似相同部分,进一步压缩空间。

  • 典型的任务

  • 图像识别:给定一张图片,通过算法确定这张图像的分类,又叫图像分类。比如提供一张含有猫的图片给计算机,计算机应当认出:这张图里有一只猫咪。图像识别的输出是整张图片的标记,是图片(若把一张宽高分别为w、h的图片看成一个长度为w*h的向量,则图像识别是找到一个从w*h的空间到图像分类标记的映射(函数)。

图像分类的例子

与图像识别相近的任务还有单帧场景理解、描述生成等,这些任务的输入仍然是一幅图片,而输出是若干个描述图像内容的句子或单词,也可以是图像中物体之间的关系。

  • 图像生成:这部分的内容比较复杂,现在一般是通过神经网络训练一个生成模型,它可以根据你给予的标签(如猫咪),根据学习的数据生成一张对应的图片,因此又被成为“画家AI”。VAE、flow model、自回归模型和GAN是生成模型领域的始祖。若感兴趣可以参考生成模型之PixelRNN、VAE与GAN三种算法浅解。

生成模型和判别模型的差别就在于一个是学习联合分布然后由贝叶斯公式求后验分布,即学习,给定一个输入,由求出给定时最有可能的;一个是直接拟合条件分布,输入一个就给出最有可能的标签。

GPT作为第一个超大的语言模型带给人们无限的遐想,其后继者GPT2、GPT3以及Google为了对标OpenAI退出的BETR等更是在训练参数量上翻了几番,Transformer在多模态学习中展现威力后,MoCo、CLIP、DALL-E,以及基于基于diffusion model的模型等直接把语言-图像模型推向新的高度,对超大预训练模型的进化和细节感兴趣的同学可以参考跟李沐学AI:论文精读系列的相关视频。

下面的这几张图片是由DALL-E2生成的,给定的输入是Bitter Cocoa Brioche And butter,模型就真的输出了栩栩如生的奶油可可卷和黄油,光影效果处理的非常好甚至对背景进行了虚化!更多由它生成的图片可以在Dalle-2 | Image Database (dalle2.app)浏览,你也可以申请模型使用权(虽然几乎没法排上队)创建一些有趣的作品。

生成模型一般会一次性生成大量输出,上图展示的是概率最高的几张

在RM比赛中,我们可以利用生成模型创造出一个会让对手的自瞄算法认为是装甲板的图片,用它来作为机器人的涂装,以此干扰敌方的识别(一般对特定的神经网络有效,对传统算法无效)。这样的涂装看起来和装甲板毫不相干,但是检测算法却会认为它是一个装甲板!这种方法叫做对抗样本生成,下图给出了示例。

通过对抗学习,给一张熊猫的图片增加了一些噪声

虽然在和一张噪声图片叠加之后的熊猫看起来和原来别无二致,但是目标检测算法却将它认作一只黑猩猩!当然,我们不能直接在装甲板上添加这样的噪声,这显然无法实现,但我们可以在周围的涂装上使用对抗样本,让神经网络认为贴在涂装上的喷涂样式是一个装甲板。是不是觉得这和迷彩服、隐身战斗机有相似之处呢?

  • 目标检测:这是Robomater赛场上最常用的算法。目标检测和图像识别在一些方面有些类似,图像识别主要是对图像进行分类,让计算机判断这张图“有什么”或者”是什么“,而目标检测不仅要判断图片中是否有对应的物体,还要输出关于这些物体”在哪儿“的信息。和目标检测相似的“目标定位”的任务则是对图像中的一个特定物体进行定位,而目标检测算法中,图像内含有的对象种类和数量都是未知的。目标检测的输出是目标物体的位置和类别。
目标检测算法不仅对图像中的对象给出了分类,还用一个Box把他们框出

上图展示的是多目标检测算法的检测结果,基于神经网络的目标检测算法能将一套框架运用到所有目标对象的检测问题上,在训练过程中习得待检测对象的特征。而基于灯条匹配、扇叶识别的算法则是专门针对装甲板识别和能量机关识别的,相当于我们手动设计需要检测对象的特征。他们各有优劣,我们会在 5.2、6.1、6.2 中对他们进行更详细的介绍。

时下效果好、速度快的目标检测算法几乎都是基于神经网络构建的,常见的有R-CNN系列、YOLO、SSD等,我们也会在 5.2 中分别解读这几个算法。可以参阅这个系列的文章来进一步了解目标检测:目标检测入门。

  • 图像分割:根据图像的特征把图片分为几个有确定性质的区域,并寻找我们感兴趣的区域。图像分割算法可以目标检测的基础上进一步解析图像,其输出可以是对每个像素的像素级别的描述,如下图中灰粉色的区域就代表“运动员”。常见的算法有阈值分割、边缘分割、聚类、基于神经网络的语义分割等。想要了解更多可以参考图像分割传统方法整理。同样,最新的效果最好速度最快的算法也是基于神经网络的。
图中的运动员和他的自行车被算法从背景中提取了出来

图源知乎-芝芝-https://zhuanlan.zhihu.com/p/143261645 ,侵删

上文所述的几种任务的直观区别,a为图像分类,b为目标检测,c为语义分割,d为实例分割

​

  • 目标跟踪:视觉目标(单目标)跟踪任务就是在给定某视频序列初始帧的目标大小与位置的情况下,预测后续帧中该目标的大小与位置。有同学可能会疑问,明明目标检测算法能对每一帧图像进行处理确定出目标的位置,为什么还需要目标跟踪?这是因为目标检测算法需要对整张图片进行处理,其消耗的运算资源很大,而目标跟踪不仅运算量以数量级的优势比前者小,还有简单准确,适用面广,抗噪性好的特点。因此在检测出目标之后,可以使用目标跟踪算法来进行后续的处理,同样能识别到装甲板等物体。在 5.3 中我们会更具体地介绍这个算法。

  • 视频理解:在以上的几个任务中,大多以单张图片作为任务对象,视频处理则是将一段时间内的所有帧都作为任务输入。一个RM赛场的例子就是,我们可以保存在一秒钟内相机拍摄的所有图片,将它们堆叠成一个张量,送入卷积神经网络的输入层(卷核的维度也要进行相应的改变)。此时,我们便不单单可以处理定格在图像中的信息了,还能够对包含时间的数据如机器人处于小陀螺运动这个状态进行检测,以启用反小陀螺算法。利用了历史数据的图像处理算法都可以看作是视频处理算法。

  • 这里只介绍了一些和机器人与比赛相关度较高的视觉图像/视频处理和分析算法,实际场景中还有大量不同的计算机视觉任务,感兴趣的同学可以到paperwithcode和arXiv上看看时下热门的研究问题有哪些。



5.1.OpenCV常用算法

OpenCV 是一个软件工具包,用于处理实时图像和视频,并提供分析和机器学习功能。使用这些标准化的软件包可以极大提高我们的开发效率,并且这些工具包对一些算法运行速度有特别的优化,能够使得这些算法在拥有GPU或支持多线程的电脑上得到加速。掌握OpenCV中的基本数据类型和常用函数是视觉组迈出开发的第一步,同时也能学习大量的相关知识。

这是OpenCV的官方网站,可以在这里的社区和其他开发者交流或查阅说明文档和例程。

首先你需要安装OpenCV,可以参考Ubuntu下OpenCV+contirb模块完全安装指南-NeoZng。

  • 基本数据类型

  • Mat:矩阵类型,能够保存图像。

  • Point:一个像素点,或者任何类型的“点”。
  • Scalar:一个四维点类,是许多函数的参数,常用于描述颜色。
  • Size:同样是一对数据构成的组,一般表示一块区域或图像的宽高,有些时候可以和point互换。
  • Rect:rectangle,矩形类,拥有Point和Size成员,用于表示一块矩形的区域。
  • RotatedRect:同上,不过有额外的成员angle用于表示角度。注意这个类的角度系统有些独特,务必阅读:OpenCV中旋转矩形的角度。

具体的说明请参阅OpenCV的说明文档,或在IDE内选择switch to declaration,便能转到注释处。

5.1.1. imgproc 模块(image process)

是我们使用OpenCV时最重要的模块之一。主要是一些像素级的操作,通过图像滤波、形态学操作、阈值操作、通道处理、图像变换、轮廓查找等功能来凸显图像特征或滤除噪声。还可以通过一些简单的绘图函数在图片上作画、输出文本。下面列出一些常用函数:

  • 画图
circle()   //画出一个颜色、大小、粗细可调的圆,一般用于标记角点等特殊位置
line()     //在两点之间画出一条直线,用于框出目标或作为参考。
//用于标记装甲板、能量机关的角点,框出候选的目标
  • 颜色空间转换
cvtColor()		//将图片从一个颜色空间转换到另一个颜色空间
split()		//把图片的不同通道进行拆分,放入不同的Mat
subtract()		//将两张矩阵的每个元素相减
//这在自瞄中将用于RGB到GRAY和HSV等空间的转换和颜色通道的分离。
  • 阈值
threshold()		//阈值操作,对一个特定的分量与阈值进行比较,大于阈值则全部设为某个值,小于阈值设为另一个值
inRange()			//进阶版本,可以确定一个分量是否在一个区间内
//我们使用这两个函数来筛选特征,对拆分后的颜色空间进行操作以屏蔽不感兴趣的部分
  • 滤波与平滑
blur()				 	//加权模糊图像
GaussianBlur()		 	//高斯加权模糊
medianBlur()		    //中值滤波
bilateralFilter()	 	//双边滤波
//用于对图像进行降噪处理,或是抹去小光斑等
  • 形态学操作
 erode()				   //腐蚀操作,二值图的边缘或收缩
 dilate()			 	   //膨胀操作,二值图的边缘会扩张
 getStructuringElement()   //获得结构元素(核)
 morphologyEx()     	   //更多的形态学操作,包括Opening,Closing,Morphological Gradient,Top Hat,Black Hat等
//用于增强图像的某些特征
  • 其他图像算子
 Sobel() 	  //微分运算,检测边缘,微分会使得图像中像素强度(某个分量)变化最大的部分为极值
 Laplacian()  //二阶微分,检测边缘,二阶微分会使得图像中像素强度变化最剧烈的部分为零
//寻找图像中的边缘

滤波、平滑、形态学操作等都属于使用图像算子对图片进行卷积操作,学习过数字图像处理或信号与系统的同学应该对此熟悉。在OpenCV中,你可以使用 getStructuringElement() 来构建独特的卷积核,随后使用 filer2D() 来对图像进行卷积运算。

  • 寻找/画出轮廓 + 矩形/椭圆拟合
 floodFill()    //漫水法,常用于寻找轮廓的预处理操作,和“画图”软件中“油漆桶”工具有相同的效果
 findContour()	//寻找二值图中的轮廓,并保存为一组点,算法类似于漫水法,遍历所有像素并查找相邻像素
 drawContour()  //根据一组点画出轮廓
//承接上面的各种预处理,用于找出图像中的轮廓并进行下一步操作
   
 minAreaRect()	 //通过轮廓点,拟合出最小面积的RotatedRect
 boundingRect()	 //通过轮廓点,找到其外接矩形Rect(水平)
 fitEcllipse()	 //通过轮廓点,用最小二乘法拟合出一个外接椭圆,函数会返回椭圆的内接旋转矩形RotatedRect
 minEnclosingCircle() 	//通过轮廓点,找到最小面积的包含圆(注意不是外接圆)
//将轮廓点转换为更容易处理的形状对象
  • 仿射、投影变换
 remap()					//根据给定的映射(函数)改变图像中每个像素点的位置        
 warpPerspective()			//进行透视变换  
 getPerspectiveTransform() 	//获得透视变换所需的矩阵(4个点)  
 warpAffine()				//进行仿射变换  
 getAffineTransform() 		//获得仿射变换所需的矩阵(3个点,为什么比透视少一个点?)
 //一般用于把图像根据变换关系转化成正视图以便进行模板匹配、SVM匹配等操作

要区分仿射变换和投影变换,请记住仿射变换是线性变换,而投影变换不单单是线性变换。仿射变换会保持对象的相似关系和平行关系,而投影变换可能会改变这种关系,添加了非线性的因素(仿射变换的维度比投影少1,投影是一个商空间)。

5.1.2. imgcodecs 模块(image reading and writing)

imread()		//根据路径读取一张图片
imwrite()   	//向对应路径写入一张图像
imreadmulti()	//一次读取多张图片	
//读取、保存测试用的图片或者自己制作的卷积核、用作模板匹配的图片模板等

5.1.3. videoio模块 (video input and output)

class VideoCapture()	    //构建一个视频捕获类,捕获的视频可以以每一帧图像的形式保存到Mat中    
//VideoCapture cap(0);
//Mat frame; 
//cap>>frame; //这样就把一帧图片保存到frame内部了     
//这个类能够通过get(),set()方法获取和设置一些相机参数    
    
class VideoWriter()		    //构建一个视频保存类,能够方便地保存视频,并且提供各种格式    		    
//在实验室时无法模拟赛场的光线环境,常常在比赛时录制相机第一视角的视频,以供之后测试使用                 
//也可以把检测完的每一帧图片连成视频,保存下来,之后根据这个视频来查找问题、改进算法

5.1.4. highgui模块(high level graphis user interface)

imshow()/*在指定名称的窗口中显示一张图片,注意和waitKey()配合使用否则可能导致异常,用于查看一些算法处理后的结果,waitKey()的参数为图片显示的时间*/   
    
//以下这个组合可以极大地方便参数调试,在程序运行的过程中通过回调函数,可以实时修改参数值
//这种根据用户或客户端的请求进行动作的编程范式被称为“响应式编程”,在GUI、OS和应答服务中很常用
nameWindow() 		//新建一个空窗口  
createTrackBar()	//创建一个拖条,传入相关的参数可以实现参数调节  
getTrackBarPos()	//返回拖条所在的位置     
    
//这个组合能够通过键盘和鼠标向程序传递参数,改变程序的状态,调试的时候非常好用  
setMouseCallback() //设置鼠标的回调函数  
waitKeyEx() 	   //从键盘读取输入

除了第一个 imshow(),在使用highgui模块时需要你了解一些响应式编程的方法(有些类似于中断编程),不同于以往的的控制流命令式(面向过程)编程,响应式的程序在运行的时候会监听并响应异步数据流(Event Stream),可以时时和用户交互。我们使用的操作系统图形界面几乎都采用了响应式编程。

5.1.5. 其他模块

  • ML:machine learning模块,内有封装完全的多层感知机、基于Dtree决策树的boost集成算法、EM(expectation Maximization)、逻辑回归、朴素贝叶斯分类器、模拟退火优化、支持向量机等经典的机器学习算法和一个能够提供各种功能的内置的数据集包。对于哨兵等自动机器人的开发有很大的帮助,可能也能帮助我们构建其他需要这些算法的模块如:装甲板分类(Bayes或SVM)。
  • DNN:deep neural network模块,随着深度网络的流行OpenCV自然也提供了对包括onnx、tensorflow、pytorhc、caffe等模型推理的支持(不支持训练)。基本的api封装的非常简洁,以分类网络为例,分别运行readnet(),blobFromImage(),setInput(),forward()就可以得到结果。对于检测网络和分割网络,还要进行后处理,不过cv::DNN也提供包括NMS()等简单的后处理。如果要在程序中嵌入一些轻量的网络,使用DNN模块足堪使用。若有复杂的pipeline和前后处理,建议还使用特定的推理框架(OpenVINO、TensorRT、NCNN、MNN等)。

  • calib3d:Camera Calibration and 3D Reconstruction模块,包含了相机标定的算法和一些三维重建方法。我们在得到装甲板的位置后,需要解算装甲板到相机的距离,这就会用到这个模块内的 solvePnP()函数。在前面提到过,为了正确的反映物体在原本的位置关系,我们需要对相机进行标定,也是利用这个模块中的函数,幸好OpenCV官方已经为我们编写了一个标定例程,我们可以在OpenCV编译目录下的 /opencv/samples/cpp/tutorial_code/calib3d 处找到它,在修改 in_VID5.xml 和VID5.xml 内的参数后就可以开始标定了。

如果你想要更具体的实例和操作,请参考在OpenCV中用例程标定相机-NeoZng这篇文章。

  • video analysis:包括各种版本的Kalman Filter、光流估计等。OpenCV提供了封装好的标准KF,我们可以通过修改状态转移矩阵、控制矩阵和测量矩阵从而将其升级为Extended版本(扩展卡尔曼滤波,4.5之后似乎自带EKF和UKF以及AUKF)。可以方便地调用此模块完成基础的运动轨迹预测功能,同时减少噪声。

  • extra_module

  • ccalib:Custom Calibration Pattern for 3D reconstruction,双目相机的标定和双目测距可以利用这里的函数进行。

  • tracking:目标追踪模块,内有几个经典的跟踪器如KCF、HAAR、HOG、GOTURN等。可以用于装甲板追踪等目标跟踪任务。
  • videostab:视频稳定模块,提供了一些提升视频稳定性的工具,如防抖、插帧、消除模糊等。但由于处理速度稍慢,缺乏实时性,难以用于自动瞄准算法。也可以用于分析制作测试视频。
  • cudaXXX:以cuda开头的这些模块都是可以利用英伟达的通用并行计算平台(CUDA)进行加速的(如果你的显卡是英伟达生产的)。

若要使用extra_module中的模块,需要和opencv-contrib交叉编译,前文的安装教程中提及了这一点。

这篇博客:RM 教程3-OpenCV 传统视觉介绍了OpenCV中基本数据类型、基本api的简单使用并辅以代码示例,推荐阅读。对应的视频教程推荐东南大学的OpenCV入门教程。学完上述内容你就基本具备了完成传统装甲板识别算法的能力。



attention:5.2、5.3、5.4对于新人来说可能有一定难度。

若是新人或刚入门的 RMer,可以由此直接跳转道第六部分继续阅读,第六部分看完后再回来这里继续 ~~

5.2.目标检测

时下RM赛场上的自瞄算法分为两个流派:传统特征提取和神经网络。前一个部分已经介绍了和比赛相关的OpenCV函数,因此为了保证行文的连贯性和整体性又不重复叙述,这个部分主要介绍新兴的基于神经网络的目标检测算法。在第六部分当中则会介绍传统算法的全部流程。

神经网络是时下最火热的ML的子领域,也就是我们常听说的“深度学习”、“深度神经网络”。而基于卷及神经网络的目标检测算法又是这个领域的超级明星。基于机器人学习的目标检测几乎在各方面都替代了传统算法:传统算法需要手工设计检测目标特征,并且很难分离前景和背景。而前者则是通过学习算法来习得检测目标的特征,并且在速度上随着各种trick的加入已经对算本身的改进,已经达到了实时性。虽然缺乏一些可解释性,但也不能阻止其在CV方面大放异彩、在各种顶会顶刊上乱杀的势头。这个部分会介绍卷积神经网络的基本原理和几个热门的目标检测算法的相关知识。检测装甲板、机器人、能量机关这些都只是冰山一角,这些目标检测算法的检测对象几乎是没有限制的,因此基本只需要更改训练集和很少的超参数就可以将算法部署到其他问题上。

5.2.0. 目标检测基础

  • 目标检测算法经过近十年的发展已经成长为一颗参天大树,基于CNN的目标检测现在大致可以分为两类:1-stage和2-stage。也可以分为anchor-based和anchor-free,再加上使用transformer的后起之秀这三类。
  • 若要入门深度学习(神经网络),首先要学习机器学习的基础知识(神经网络算是机器学习的一个分支)这里推荐斯坦福大学吴恩达教授的系列课程机器学习经典名课,深受大家欢迎,一般在看深度学习课程之前将此课程学完会有更好的理解,当然最好是能先看完。而机器学习又是以多元微积分、概率论和线性代数为基础的,这三门基础课一般会在大二上学期之前结束。额外学习一些凸优化的内容将有助于机器学习的理解。高数相关的内容推荐国防科技大学朱建民教授的一系列mooc,线性代数推荐MIT Gilbert Strang教授的18.06。
  • 吴恩达教授的深度学习系列课程相信是不少同学入门DL的选择,深入浅出贴近实践并且在coursera上有相关的练习。务必至少看完这个系列的前四部分视频再学习目标检测的相关内容(第五部分序列模型是Transformer的基础,但是我们暂时不需要,有时间的同学可以一并学完)。若能学完这两个系列的课程,在AI方面就算是入门啦。

  • 深度学习方面广为流传的教程还包括李沐的《动手学深度学习》和台大李宏毅的深度学习课程。《动手学深度学习》在任意搜索引擎搜索d2l即可,拥有配套的jupyter notebook,非常方便,bilibili上有作者本人的教学视频,他出品的面向真实应用场景的实用机器学习也非常受欢迎。台湾大学李宏毅教授的深度学习课程也是广为流传,上课的时候很多梗也很多二次元,PPT精美可视化很多,寓教于乐且紧跟学术前沿。

我们首先介绍目标检测的基础——神经网络。

当开始阅读 5.2 之后,笔者默认读者已经拥有了微积分、线性代数、机器学习的基础。


5.2.1. 神经网络

  • 和传统的统计学习方法如SVM、PCA、LDA等都是需要经过训练的。以最简单的单层感知机为例,其实我们可以将所有的神经网络看作是一个多维特征空间到另一个我们需要的得到的属性的映射。对神经网络的训练,就是用一对对标记好的数据(已知输入的特征向量和输出的属性,即数据标签)对这个“函数黑箱”的建模过程。其中的每个神经元的w、b即是我们需要得到函数的许多构成参数之一。然后我们根据BP(back propagation)算法等对网络进行最优化(让网络学会这个我们需要的函数,即学习过程),使得这个函数能够逼近我们需要的理想的映射。

分类问题就是在回归问题的基础上增加后处理,如加入SVM和softmax函数等。对于基于卷积神经网络的猫狗图像分类器,假设其输入是320x320的图像,那么就是有320x320=102400个特征(每个像素作为一个特征,单通道位图),其输出是一个三维向量[cat dog none],每个元素智能取0和1并且最多只能有一个元素为1。那么,某个元素的值为1时说明此图像是对应的那种物体,none为1则为说明图像中没有猫也没有狗。那么,训练出来的函数就是一个102400元函数,其输出为3维的离散序列,此函数通过映射将一张图片映射到上述的三维离散空间中。

在此网站:A visual proof that neural nets can compute any function,有对神经网络是如何拟合出任意函数的形象的、直观的解释。手机上还有一款就叫做神经网络的App,你可以手动设置隐藏层的数量、各个神经元的w、b和激活函数等,然后得到一张清晰的拟合出的函数的图片(当然只有3维空间以下的结果能够被可视化地输出)。

app神经网络的截图,内部还提供了著名的网络如shuffleNet、语言模型、TextCNN、自编码器等
  • 对于深度学习或机器学习模型而言,我们不仅要求它对训练数据集有很好的拟合(训练误差小),同时也希望它可以对未知数据集(评价集、测试集、实际应用)有很好的拟合结果(泛化能力强,即训练生成的函数可以很好的根据输入预测输出且基本没有错误),所产生的测试误差被称为泛化误差。度量泛化能力的好坏,最直观的表现就是模型的过拟合(overfitting)和欠拟合(underfitting)。过拟合和欠拟合是用于描述模型在训练过程中的两种状态。这两个问题的都会导致神经网络的泛化能力下降,欠拟合一般是通过增大训练数据量和模型复杂度解决,而过拟合则是当前科研面临的主要问题,解决方法也多种多样。
训练模型得到的几种状态的图解,以回归分析为例
  • 因此,在训练神经网络时,我们需要用一些trick来改善它的性能或加速训练过程,在此过程中就涉及到需要人为确定的许多“超参数”。这些超参数设定的数学依据较少甚至没有,很大部分都是启发式的(拍脑袋想出来的),而不是有一套严密优雅的数学方法来对神经网络的表达能力和学习有效性进行分析。这也是神经网络被传统的统计学习学派诟病为“炼丹”的原因:需要在实践中不断调整超参数的值和网络结构,根据训练结果才能改进网络的结构和超参数的值(就如控制领域的pid算法一样,虽然三板斧调参用起来猛,你却很难说出个所以然来)。

不过它们确实有效,所以我们还是来看看他们的作用吧.


5.2.2. 防止过拟合

  • 正则化(Regularization)

个人认为这种翻译的有些生硬也不够直观,按照笔者的理解,应该被称做“规范化”或“约束化”。通过对损失函数加入额外的先验知识,防止w在训练中变化过于剧烈,进而在很大的程度上改善过拟合的问题。在一些教程中,常常用一个两参数的单元来可视化这种方法:

摘自周志华教授的《机器学习》,L1正则化和L2正则化的形象图解

通过对w添加额外的限制,倘若w在训练过程中变化过于迅速(这是过拟合的一个主要原因,对训练集数据过于“亲近”,以至于学习到了部分分布当中的一些偏差与噪声,而不是整体分布的信息),额外的惩罚项会约束w的增减,降低其变化速度。详细的解释,在Andrew Ng的深度学习课程第二部分p5.

  • 数据增强(Data Augmentation)

以图像分类为例,通过将训练集中的图像进行翻转、反转、裁切、旋转、变形和添加色彩偏差等方法来扩充数据集,是最廉价最简单的防止overfit的方法。虽然其效果不如收集更多的图片来的得好,可架不住成本低、几乎无开销的特点。在小规模训集上,数据增强是最有效的方法之一。

高级一些的数据增强如MixUp、Mosaic等复合了以上几种中的若干种,有兴趣的同学可以自行查阅资料。别小看数据增强的方法,一个优秀的DA甚至能提高几个点(mAP)。

通过翻转和旋转、裁切等操作,扩充了猫咪数据集
  • Dropout正则化(随机丢弃)

初看dropout方法的同学一般都会认为它深度学习中显得更加玄乎。其原理大概是在某个隐藏层中随机使某个神经元失活(使其输出为零),这样拟合出来的函数参数量下降,不容易拟合复杂的函数,就更不容易出现过拟合的现象。然而dropout会在反向传播中出现难以解决的问题,由于其丢弃的结点是随机的,无法在每一轮反向传播中确定要更新的参数,因而无法明确的定义损失函数,除非每一轮训练只使用一个样本(想想为什么?)。

从数学角度理解,似乎dropout能够产生L2正则化的效果,随机丢弃某一层中的一个结点,其结果就是该结点不会在本次反向传播对损失函数的计算产生影响。若该结点在之前的训练中变化过于剧烈,那么使其失活就有机会让其他结点的参数得到训练从而获得变化,而非将所有的赌注放在那个结点上(put all of its bets on)。因为每个结点失活的概率是相同的,那么他们应该会得到均等的训练机会。详细的解释参见深度学习课程第二部分p7。

dropout图示,使神经元失活即置零其输出
  • Eearly Stopping(提前停止、早停法)

非常容易极致简约的方法,如果训练的轮数过多导致模型过拟合了训练数据,那就减少训练轮数。不过我们不用真的用不同的轮数训练多次,而是如一次训练300个epouch(将所有样本都循环300次),然后根据其在测试集上的表现,选取测试误差最小的那一轮,保存其权重作为最后使用的模型。

训练集上过度优化会导致过拟合

5.2.3. 让你的网络学习得更好

  • 归一化和标准化(Normalization and Standalization)

这两个操作常常被混淆,在一些文献中也表示相同的意思,他们都是特征缩放的方法。因此本文将他们视作同一种类的不同操作。常见的归一化操作有minmax缩放、零均值化、方差归一化等。也可以把归一化理解成“单位化”、“统一化”。minmax缩放将所有数据通过线性映射把值投影到一个区间内。零均值归一化操作会将数据特征的均值化为零,方差归一化则是将特征的方差变为1。后二者常结合在一起使用。

使用了零均值化和方差归一化后的数据,截取自深度学习系列课程

通过这些尺度缩放的操作,可以缩小存储数据需要的空间,同时可以让损失函数变得更加“圆润”。下图给出了未归一化和归一化后的损失函数的对比。显然通下降法,归一化后的数据能够更快地使网络收敛,因为在圆周上的任意一点梯度方向都是垂直于“等高线“的,梯度是函数变化最快的方向(多元微积分知识)。

使用前后的损失函数对比,下方为损失函数的等梯度线(标量)
  • 防止陷入局部最优解

当损失函数可能不是一个凸函数,或者它有多个极值点(大多数情况下都是这样),那么,使用梯度下降方法可能会使神经网络拟合出来的函数陷入局部最优解(向着不是最小/大的极值点不断迭代)

  • 多种初始化值(从头训练)

在初始化网络参数权重时,一次选取多组参数,相当于一次训练多个有相同结构的网络。这样,我们能够从参数空间中的不同位置开始运行梯度下降。即使部分解陷入了局部最优,我们还有其他的解。可以把这种方法看成”广撒网“式的训练。最后我们会测试这多个网络的性能,选取其中最好的一个。虽然这种方法的效果最好,但是其缺点也十分明显:极大的训练开销。

从参数空间的不同位置(蓝色点)开始梯度下降,即使有部分陷入了局部最优解,也可能有一些会收敛到全局最优,至少也是比较好的极值
  • 局部扰动

使用局部扰动有一定的概率跳出局部最优解:在网络的Loss function不再减小或缓慢振荡的时候(也许此时陷入了局部最优解),随机地给网络参数进行微小的扰动。若在有限步迭代(人为设置地超参数)内能够往其他方向移动并且进一步缩小了Loss function,那么说明我们找到了一个比原来的解更好的解,此时再继续运行梯度下降。若迭代次数超出了我们设置地扰动次数上限,则返回原来的位置,认为函数已经收敛到比较好的极值点了,得到了较优的结果。如此,就有概率跳出局部最优解。

局部扰动和模拟退火算法在有某些相似之处。也有研究人员提出,可以利用模拟退火算法来在小规模网络上进行训练,收敛速度十分迅速。

对局部最优解加入扰动后,成功找到了更好的解
  • mini-batch(小批量训练)

不需要遍历整个数据集就进行一次剃度下降,选取部分样本作为一个批次来计算其loss。从另一个角度来理解,就是加快了迈步的频率。朴素的梯度下降是遍历整个数据集后再计算loss,对参数进行一次更新。minibatch则是把数据集分为32/64/128/256...这样的“小训练集“,在一个小训练集遍历后就进行更新。其优点很明显,就是可以提高训练速度。但它也同样存在一个小缺点:只使用部分训练集进行训练可能产生局部的过拟合或是欠拟合(小部分的数据集无法反映整体数据分布),还好在实践中对模型的影响似乎并不是很大。在数据集稍大的情况下,我们一般都会选择使用minibatch进行训练,否则收敛速度过慢且空间开销过大。

mini-batch的极端就是每个batch只有一个样本,即SGD(stochastic gradient descend)。经过调优后的SGD在训练的时候也有很好的效果,这一般需要有经验的炼丹师对学习率、lr schedule和初始化参数等进行调节,新手还是乖乖用Adam吧(这些名词在下面都会进行介绍)。

  • Batch-Norm(批量归一化)

  • batch-norm字如其名,对每一小批数据进行标准化。它会将一批数据的每一个隐藏层的输出(可以看作编码后的特征)都进行归一化。我们在前面已经了解到了Normalization的好处,于是,有研究人员就想到为什么不能对神经网络中的每一层输出都进行类似的操作呢?于是,我们采用的Batch-Norm会将数据集的数据划分为一个个batch,并对每个batch中的数据在每一层的输出都进行归一化。(思考:那为什么不对所有数据即整个训练集的数据在每一个输出层都进行normalization?)

  • 在训练结束后进行推理时,一般都是单个数据的输入(特别是实时视频序列的处理),这时候要注意的问题就是每一个输出层的归一化使用的均值与方差的确定,显然对单个数据没有均值和方差可言。因此在训练过程中,我们会用指数加权平均的方式记录每一批norm的均值和方差,并以此作为推理时的参数(根据大数定律,这个值最终会近似实际的分布)。

  • Batch-Norm为什么奏效呢?在前面的归一化中我们已经了解过,这些操作能够让参数搜索空间中的loss function改变成“碗状”从而获得更好的梯度下降效果。但是既然数据分布已经完成了这件事,在中间的隐藏层为何还要继续使用normalization?这里就要涉及到一个叫做covariate shift的问题。想要理解covariate shift,请看下面这张图:

从第三层开始,可以将其看作整个网络的”子“网络。每一层的参数都试图从前一层的输出中习得这些数据到训练标签y的一个映射

这里只是以第三层为例,其实可以没去掉一层(计算一层)就把它看作是一组数据到y hat的映射。那么自然每一层的输出向量都可以看作是一个特征向量,则我们就对这一批“特征向量”进行归一化。Covariate Shift就是一个向量经过前层的计算,数据原来的分布已经改变了(每一层网络都是非线性映射,从极端的角度来看每一个隐藏层都是一个网络,单层感知机)。那么显然,如果数据的分布一直在改变,网络是很难习得一个映射的(数据分布随着参数更新始终发生变化,就像解方程我们需要一些确定的信心,而在这里这个信息却会一直发生变换表现得像未知数)。使用batch-norm则可以把每一层的输出都标准化为方差为1均值为0的分布,虽然它的具体形式还是未知,但至少我们可以线只它的均值和方差达到限制前层参数的更新对输出数据分布影响的目的。

  • 同时由于batch-norm只对每一个batch的数据作用,显然一个batch不能体现整体数据的分布,因此会引入统计偏差和噪声,有时候这反而是有利的,其效果类似dropout,能够产生一定的正则化效果,避免网络拟合出来的映射过分依赖于某个神经元。

关于Batch-Norm的更多信息,可以参阅论文原文。文章:re-thinking batch in batchNorm 也是一篇很好的参考文章,介绍了更多关于统计和数学原理的知识

  • 学习率衰减

学习率作为最重要的超参数之一,于我们来说有很大的操作空间。训练的开始阶段加速学习迈大脚步,采用较大的学习率;在接近最优解时、Loss function下降减慢时减小学习率,防止超调和振荡。

不同学习率对网络收敛的影响,对比蓝色曲线和绿色曲线,我们可以取二者的精华

有些训练策略还会在初始阶段进行warmup,先选择较小的学习率防止随机初始化的权重使得模型大幅振荡。Warmup也有两种:一种是常量warmup,在热身时选择一个固定的学习率。它的不足之处在于从一个很小的学习率一下变为比较大的学习率可能会导致训练误差突然增大。另一种是渐进warmup,即从最初的小学习率开始,每个step增大一点点,直到达到最初设置的比较大的学习率,再在之后的学习中逐渐减小学习率。还有warmup-restart、cycling、cosine,每隔若干epoch就会重新启动一次warmup、让学习率从大到小以正弦规律往复变化等等learning rate schedule方法。需要更深入的了解可以自行查阅相关资料。


5.2.4. 梯度下降的进化

5.2.4.1. 进击的mini-batch
  • GDM(Gradient Descent with Momentum)

动量梯度下降法的收敛速度总是要快于普通的梯度下降法,这得益于它的”动量累积“思想(Andrew Ng说物理好的人容易理解这个算法)。假设在一个网络的训练中你遇到的参数搜索空间如下图所示,其等梯度线呈椭圆排布,那么在训练的过程中就会遇到比较大的振荡(梯度总是沿着等梯度线的切线的径向),即蓝线所示的学习路线。一旦你增大学习率,就会得到紫色的学习轨迹,产生过大的振荡甚至出现不能收敛的问题。我们希望在这个学习问题中能够让竖直方向的学习速度下降,水平方向上的学习速度增加(蓝字所示)。

一个二维的搜索空间,红点表示Loss的最小值

观察蓝线会发现其竖直方向上的振荡均值始终为零,走了很多没必要的路,是否可以通过某种方式让竖直方向上的学习率降低?(注意此处说的竖直方向和水平方向的都是以二维情形为例,显然我们的搜索空间没有这么小,在高维空间里就对应多个方向)GDM就是通过观察学习的特点得到的一种优化方法。

在每一个batch的迭代中,会记录下当前的dw和db,并且以指数加权平均的方式对dw、db进行累积,把得到的累积值用来更新参数。这样,GDM就会记录那些振荡项并把方向不同的dw、db分量抵消,而那些方向相同的分量则会随着迭代的进行不断增大,就好似GDM能够聪明地找到最快下降的方向,顺着“山坡”(最速降线)不断累积“动量”,迅速到达最小值点。

竖直方向上的分量随着迭代减小,水平方向不断增大

如果拿具象的例子类比,就是将一个小球置于碗中,并且给它一个切向的初速度(相当于一步迭代)。对于普通的GD,每次小球的切向速度为零时就让小球停下(把径向速度也归零,即重新计算dw),然后再次释放小球;而对于GMD,就只给小球一个切向速度,之后不去干扰小球(每次计算完成之后,dw都会加权累积),那么切向速度会随着往复运动而逐渐减小(dw的反方向分量会抵消缩小),径向由于有重力加速度(方向相同,累积后不断增大),动量会持续累积,以更快的速度到达碗底。

  • RMprop (Root Mean propagation)

RMprop和GDM异曲同工之处,同样会在迭代中保存每一次的dw值并对其进行均方加权平均。在更新参数的时候会使用当前一次迭代的dw除以前述得到的值即前几次迭代中dw的平方的加权均值(想想为什么用平方),这样,当dw变化很大的时候,该累计值也会很大,经过相除这个值就会相对变小;而dw小的时候,累计值也相对小,那么相除后用于用于更新的参数将会相对变大。这样的操作让每次更新都更加保守,不会像GDM一样速度越来越快而是始终稳步前进(此速度可以通过加权平均的系数修改)。这种特性也允许在训练时使用更大的学习率。

绿色的箭头就是是用了RMprop后的效果,可以看到学习的步伐非常稳健
  • Adam (Adaptive Moment Estimatation)

Adams融合了上面两种方法,最后得到的参数更新公式如下:

Adam方法“稳中求进”

Vdw是GDM的累计项,Sdw是RMprop的累积项,在计算结束后对Vdw和Sdw都进行了指数平均修正,确保在刚开始的时候指数加权平均不至于过小。是一个很小的数(一般为10e-6)防止出现除零得到NAN的情况,这也是在设计网络中,若出现除法操作常用的训练技巧。

5.2.4.2. 尝试更高的阶数

这部分内容主要是作为了解,因为pytorch、tensorflow、mindspore等流行的框架都只支持自动微分,并不支持更高阶的优化方法。当然,如果你要为自己的某个项目的优化问题编写求解方法,可能会用得上。

  • 牛顿迭代法

  • 拟牛顿法和其他高阶方法

  • 这些常用的非线性优化方法实际上是一门系统的课程/科学,一般被称作运筹与优化理论。实际上,研究和工程上的很多问题都可以转化为优化问题。即如神经网络一般确定一个目标函数,并寻求合理的求解工具,如简单的初高中就学过的线性规划,微积分所学的KKT问题,以及更深入一些的凸优化和半正定优化方法,当然还有神经网络中最常用的梯度下降法(一节优化方法),使得目标函数能够最小/最大化。

了解了神经网络的构成和基本原理后就可以开始进一步学习卷积神经网络了。这里也再次提醒读者应该至少至少先学习吴恩达深度学习课程的第一个和第四个系列视频再来了解相关方面的知识。拥有基本的了解和学习背景对于探索新领域是非常有必要的,否则你只会在n个不同的超链接里面反复横跳(这是啥?查一下,诶,这又是啥?再查一下......wft?怎么又回来了??)。若有信号与系统或数字信号处理的前置知识,对于CNN的学习大有脾益。


5.2.5. 卷积神经网络

在5.2.1中我们提到,可以把图像resize成一个1xn的特征向量当作输入投入一个网络当中进行训练和预测,但是这样做会出现很多的问题:首先是参数量过大,拿一张320x320的rgb图像举例,它是一个拥有320x320x3个维度的向量,若隐藏层也采用这样巨大的神经元规模,其参数量是不可想象的(利用你的排列组合知识计算一下)。另外,将图像resize成一维向量,在一定程度上丢失了两个维度之间的相关性(在映射的过程中有信息被压缩到原向量空间的零空间中了)因此,卷积神经网络就诞生了(LeNet5,1994)。但是由于算力的限制,直到到近年GPU的快速发展才得到广泛的应用。

卷积神经网络的设计思想是参数共享、空间信息的保存与合理的参数量下降。

如何用网络设计一个图片分类器?根据前面学习的知识,可以把图像看成一个width*height维的特征向量并把它输入到一个超大规模的神经网络中。在引言中提到这样做的参数量是随着分辨率的上升以阶乘级上涨的,我们显然无法处理这种规模的参数,并且在 5.2.1 也提到了过大的参数量容易让模型在训练过程中过拟合。

那还是看看造物主的杰作吧:计算机科学家从人类的视觉机制出发设计出了卷积神经网络。从认知心理学的角度来分析,人类的并不是一眼就能判断物体是什么,而是从其纹理、形状与轮廓这种低级特征开始鉴别一个物体,再上升到局部特征,最后通过组合局部特征得到对整体的判断。也就是说,我们没必要在“第一眼 ”就关注所有像素。

根据这个思想,我们改进了神经网络的运算方法、参数传递和激活方式:利用卷积运算替代全连接。即不必对每一层都进行全连接,全连接代表每个像素对之后的每一个神经元都有贡献,然而左上角的像素和右下角的像素可以说几乎没有任何相关性(其他大部分像素也是这样)。因此我们在刚开始只需要关注纹理和边缘,然后是纹理和边缘组成的局部特征,最后才是整体。这样,我们可以依据某种规则只让神经元之间进行部分连接,最后部分再进行全连接以充分利用高层次特征。

那要怎么样确定这种规则呢?接下来分别介绍CNN中最常见的三个构成模块:卷积层、池化层、全连接层。

5.2.5.1. 卷积层

引子部分提到,如果把图像resize成一个一维向量,我们很可能就在此映射过程中把原来处于相邻像素之间的相关性给丢失了。想要保存这种相关性怎么办?这就要利用到“卷积”操作了。

一个3x3的图片,每个像素和周围的像素显然是有位置的相关关系的
把上图resize成1x9的向量,可以看到这里就丢失了原来的位置关系

学过信号与系统的同学应该已经对一维卷积熟悉,时域的卷积操作可以得到系统某个时刻由连续输入叠加产生地相应(一维情形),那对于图像信号该怎么做?显然图像在两个方向上延伸,图像信号就是坐标的函数(二元函数),我们直接把卷积的积分变量由时间改为坐标分量并进行离散化,那么二维卷积操作的结果就是系统在卷积核范围内的输出响应(即计算这个位置是否有相应的特征)。因为只需要采集局部的信息,我们不用把卷积核设置的太大(如果考虑最极端的情况,把卷积核半径设置得和图像一样大,那么就相当于采集了整幅图像的信息,和全连接无异了,只不过保留了位置信息)。来看几个动图:

二维卷积

如果没学过信号与系统或者还是不怎么理解卷积,可以参考知乎的这个问题:如何通俗易懂地解释卷积 。

若你学习过信号与系统,可以用我们所熟悉的一维卷积作为类比,二维情形一个大小有限的卷积核就相当于一个一维的只在某个区间有定义的、在其他地方的值为零的卷积函数(脉冲响应函数)。增大卷积核即增加感受野(在 5.2.3 中介绍),对比一维卷积函数来说就是扩大了使函数不为零的定义域定义域。

大小有限的卷积核与一维情形卷积函数的对比

对一维离散序列的卷积

图源知乎-palet

未学过信号与系统的同学可以忽略这个类比理解。

那么卷积又是如何得到这些纹理等特征的?让我们看看提取图像中垂直边缘的例子就好了。以提取竖直边缘的特征为例,我们需要一个这样的卷积核:

用于提取垂直边缘的卷积核,当某处出现下图所示垂直边缘的时候,卷积操作会在对应位置得到较大的值
一张图片的像素表示,和他的显示效果(下图)

再回想一下卷积的含义,是某个系统在受到连续输入时响应的叠加,这里的“连续输入”指的是随着时间变化的输入信号。很显然图像信号是坐标的函数,那么在拍摄的那一刻所有信息就定格下来,这里的连续输入就变成了不同坐标的像素输入(x,y两个维度)。显然,在有垂直特征出现的时候,该区域的卷积结果就会很大,如果没有相应的特征出现,那么结果显然就会比较小。想要提取其他特征同理,这里给出提取水平边缘、角点的卷积核:

提取水平边缘的卷积核其实就是竖直边缘旋转90度,角点则是两个边缘的交点

因此,有怎么样的卷积核就会提取出怎么样的特征,不过到现在为止,我们还只是手动设计了物体的特征,比如上面提到的几个特征提取器(垂直、水平、角点),如何让机器自己学习特征?和标准的神经网络对比,这里的卷积核参数(即应该要学习到的特征,卷积核上每一个位置对应的参数值)其实也就是神经网络的权重参数w和b(隐藏层变成了卷积层)。这样,我们不必再自己设计特征(自己设计的特征不能涵盖所有情况并且特异性也不够好,学习得到的特征更能反映物体的本质),而是通过反向传播来让网络自己学习特征。

全连接神经网络和卷积神经网络参数(特征)的对比

现在,我们便有能替代全连接并且保持图像中像素的位置关系的同时还能提取特征的数学工具:卷积操作。通过网络的反向传播和梯度下降,我们便可以提取各种各样的特征。

此外我们还可以修改卷积的步长step(上面的例子中都是每隔一个像素都进行一次卷积计算),修改step就会改变每次卷积核移动的像素数,这样能够降低计算量、扩大感受野,对于大物体和高分辨率的图像常常使用3、5、7等大小的步长。

步长为2的卷积示例

对于图像的边缘,常使用“Padding”的技术来充分利用角落和边缘的像素(上面的例子中我们可以发现,每次卷积操作[step=1]得到的feature map都会比原图更小,那么我们可以通过在图像周围填充0或是对图像进行简单的复制,使得卷积核在运算时可以让中心保持对齐,这样在一些有特定需求的场景(比如对精度要求高、希望提升准确度)就可以生成和原图大小相同的feature map从而充分保留、突出图像的特征了。还有一些后面会介绍到的特殊的结构如残差连接、密集连接等(其实在现代神经网络也属于标准设计了),也要求输入输出前后的feature map大小相同。

padding示例
补零padding示例,复制padding则是通过一定的规则例如取边缘平均等方法填充外面一圈的值

是不是已经迫不及待想要训练一个CNN了呢?那么卷积神经网路的反向传播和普通神经网络有什么区别?我们在讲完池化层和全连接层后再来细说,先把前向传播的过程看完吧!


5.2.5.2. 池化层

现在我们已经通过了一层卷积并且使用的是竖直边缘提取核,把有竖直特征的区域都保存了下来,那么这张特征图(feature map)里面数值大的点就代表着此区域很有可能有竖直的边缘存在。聪明的你大概已经想到我们可以通过设置阈值的方式来筛选这些特征,得到竖直边缘的强度大的区域(“够不够竖”)。不过这里有另一种方法,既能得到最“竖”的那一个特征,又能减少参数量——Pooling。

池化其实也是一种降采样的过程,早期对其的翻译也是只表意不表义非常生涩,笔者个人认为叫“汇集层”或“采集层”会更好理解一些,池化操作会提取feature map上最重要和显著的特征。常用的池化操作有最大池化和平均池化(平均池化其实用得非常少了),下图直观地展示了池化操作的过程:

最大池化操作会保留feature map上最显著的特征

平均池化则是更加中庸,对一个区域内的特征进行加权平均,一般加权系数相同

显然我们可以继续用卷积操作来提取feature map上的特征,但是池化不需要进行乘法和加法(最大池化不需要),在速度上远远超过了卷积。并且在用卷积提取了图像的特征之后,利用池化可以让被提取出的特征更有显著性、更凸出,以便之后的卷积操作会更有针对性,并且大大减少计算量。

使用池化相对于卷积来说还能提供额外的一些好处(当然是必须要配合卷积使用,只有卷积能提取特征,池化是让特征更明显,或者说是卷积+池化操作对比单纯卷积的好处):

  • 抑制特征噪声,降低重复信息

当以两个邻近的像素为中心分别进行卷积操作的时候,得到的结果很可能非常接近,这样就带来了信息地冗余并且在之后的对此feature map进行的卷积操作(卷积核必然会同时覆盖到这两个像素)可能会产生不利的影响(过大或过小的结果)。利用最大池化或平均池化,就会把当前特征图上邻近的特征进行融合、取极值以降低冗余度。

  • 提升模型的尺度不变性、旋转不变形、平移不变性

当一张图像发生轻微的旋转、平移和尺度缩放,池化操作对这些变化并不敏感,即使部分特征变换了位置或大小,也不会影响池化提取的结果或影响很小(从池化操作是如何进行的来理解)。

  • 一定程度上防止过拟合

其实和第一个特点有相似之处,因为在进行池化操作的时候,随机噪声会被抑制,而局部样本不太明显的特征显然不如数据分布的相同特征那么显著,池化更加关注全局的特征而不是局部出现的特征,这在另一方面来说是对参数空间的降维,以防模型学习到一些不重要的特征维度上的信息。因此能够防止网络的过拟合。

所以,我们构建CNN的时候一般都是用1~3个卷积层+1个池化层作为一个基本的block(conv-pooling-block,CPB),然后用多个block进行串联得到网络的基本结构,最后加上全连接层完成整个网络的构建。


5.2.5.3. 全连接层

在经过数个基本的block(conv+pooling)后,是不是感觉特征提取的差不多,网络已经知道这些特征对应的标签是什么了呢?这时候我们会选择把最后的feature map,resize成一个一维向量(最后的feature map上不同点的相关性其实已经很差,并且每一个点在此时都代表着某种局部特征,对于图像识别任务来说即使丢失位置信息了也不会对最终的结果产生太大的影)或是使用一个和特征图一样大的卷积核进行卷积或1x1卷积(也就是第二种方法,一些情况下也被称作全卷积Full Convolution Network),这样便会得到一个1xn的张量,本质上就是一个一维向量了(和resize得到的结果相同了)。一般在最后添加1~3个全连接的隐藏层并在输出时连接到一个神经元,对其使用softmax分类(二分类的推广),就可以得到一张图像的类别了。

以LeNet-5的结构为例,在数个Conv-Pooling Block(CPB)后增加了最后的全连接后,得到一个最简单的网络结构

刚刚说到全连接层会损失位置相关性,指的是多个FC串联会造成这种结果。但是一个有趣同时似乎也比较明显的结果的是:在CPB后直接跟上单层FC连接反而不会丢失像素间的位置关系并能够充分利用位置信息输出位置敏感的数据,有兴趣的读者可以思考一下两种连接方式的差异和造成这样结果的原因。在 5.2.6 中将要介绍的目标检测网络的回归分支就常选用全连接层放在最后用于输出检测框。

至此,一个完整的CNN就搭建好了,假设这是一个已经训练好的网络,这时候我们只需要投入一张图片,它就能够输出对应的分类了!可惜它还没训练好

下面让我们看一下CNN和标准的全连接神经网络的训练有什么差别。


5.2.5.4. CNN的反向传播

很多教程在这部分讲得非常模糊甚至根本没有提及这个过程,其实它的原理和普通神经网络是几乎一样的,只不过由于我们更应该去打破这种简单的黑箱子。

CNN在前向传播的时候,计算似乎比全连接网络要复杂,然而其本质都是线性组合+非线性化,即用上一层的输出向量通过分量乘以不同权重相加再投入一个激活函数(非线性过程)得到这一层的输出:

计算过程完全一致,只不过,有没有什么办法可以将CNN的前向传播也写成FC一样的矩阵运算?

为了能够方便的计算出CNN的前向传播结果,充分利用矩阵运算加速库(如numpy等)并发挥GPU的威力,我们只需要将CNN的forward Propogation转化成矩阵形式:

只需要把上方的原图通过一点小trick转换成下方的矩阵,轻松的用矩阵相乘来得到输出了

图源深度之眼-Deepshare.net

这里我们只不过是把原来的一个正方形resize成一个一维向量,观察一下转换后feature map的下标很容易找到resize的规律。这里也回答了前面提出的“为什么单层FC能够保留位置信息”的问题。在此之后,我们也很容易写出反向传播的偏微分方程了。

当然,这里只是对一张单通道图片进行运算,那如果有多张图片,或者一个图片有多个通道该怎么做?小问题,把这些通道或图片全部resize成一个如上图所示的matrix,随后在行的方向上,把它们堆叠起来形成一个长长的矩阵(显然其列数还是保持不变,为卷积核的大小,故满足矩阵相乘的定义),再进行相同的运算即可。

卷积搞定再来想想池化操作,有了上面了例子是不是感觉也简单了不少?平均池化其实就相当于卷积核的参数固定为(1/卷积核大小)的卷积操作,这里不再展开说,直接上图看看最大池化吧:

用同样的方式根据卷积核的大小,把原feature map resize成x hat,再取每一行的最大值

池化层只会提取前层最大的特征,显然没有被提取到的特征点在反向传播中不会得到训练,分配到的误差为零,梯度为零:

图源深度之眼-Deepshare.net

至此,反向传播的基本介绍就到此结束。当然我们没有讲解使用batchnorm、layernorm的时候反向传播应该如何进行,不过原理都是完全一样的!最简单的办法就是先按照规律resize成一维向量,随后就和标准网络的FC层如出一辙!现在,你应该能够使用pytroch、caffe等网络框架搭建属于你自己的CNN图像识别网络了,快动手试试吧,马上复现一个LeNet-5问题应该不大。


5.2.6. 目标检测

利用CNN我们已经可以完成对图像的识别和分类。但是这样是远远不够的,为了能准确定位图像中的物体,我们需要对图像中所有目标进行定位(找出框住目标的bounding box外接矩形框,即[cx,cy,w,h]四个参数,分别表示目标中心在图像中的坐标和bbox的长宽)。此部分会介绍几个经典的目标检测网络实现的原理和方法。

有同学可能会想,那我直接让网络在全连接层后输出一个向量而不是标量(分类),即多输出四个坐标也就是8个值,分别代表图像中目标的四个角点不就行了吗?确实,对于只有一个目标的图像我们可以这么做,可惜倘若图像中有多个目标的话,网络的输出就变成不确定的了(需要一次性输出不确定长度,分别表示每个目标框中心、长宽和其对应的分类),我们是无法训练一个没有确定输出的网络的。

目标定位方法,得到一个WxH到1x9向量的映射

1代表有目标在图像中,随后四个数代表中心坐标和目标框的长宽,最后四个数表示目标分类

因为图像分类是对整张图像进行的,是一个WxH维到一个确定的标量或向量(如上面说的同时进行分类和回归)的映射,标准的CNN方法显然无法完成一张图片中多个对象的分类和定位操作。不过机智的你应该已经想到,我们可以把图片拆成很多的部分,然后将这些子图分别投入CNN从而得到它们对应的分类,不就把里面的对象定位出来了么。

没错,这就是密集采样的定位思想,也是最早的目标检测方法。使用滑动窗口来检测目标位置的方法和模板匹配有些类似。还有一种方法是将图像用不同的长度划分格点,在每个格点中进行图像分类(这其实是一种特殊的滑动窗口方法,即设定滑动步长大小使其与窗口的大小相同)。

采用不同大小的滑动窗口,裁取图像的一部分投入CNN得到分类(图中只使用了一种大小)

显然,这种方法的缺点也非常明显:超高的计算开销。步长设置的过大可能会发生漏检,而为了检测大小不同的目标,我们又需要采用各种大小的窗口来运行CNN,同时还可能需要使用不同长宽比的窗口来应对物体长宽比不同的情况(例如汽车在侧面看是细长的矩形,正面又比较接近正方形,如果仅使用一种类型的窗口,则可能会出现定位不精确的问题,得到的定位框包含了大量的背景)。

使用滑动窗口方法可能出现的问题,没有一个滑窗能够和汽车整体匹配

仔细观察,我们就会发现,在目标检测问题中,我们面临的最大困难就是候选区域太多且目标大小的尺度不一,问题的搜索空间过于庞大,我们难以用暴力搜索穷举全部解。于是two-stage方法的代表作R-CNN(region proposal-CNN)横空出世。不过,再介绍R-CNN之前,先让我们看看目标检测中常用的术语和概念吧。


5.2.6.1. 目标检测中的常用术语
  1. GT(ground truth)

GT可以理解为真实值、有效值、正确的答案、正确的标注信息。对于目标检测来说,标记的坐标点和标记分类就被称作ground truth。

训练标记的bbox区域(蓝色区域)就是ground truth

在目标检测网络的推理过程中,如何为预测框分配标签(训练的label即GT的属于的分类)会极大地影响训练的效率和推理的准确度,这也是当前目标检测研究的热点和难点。

  1. IOU(intersection of Union)

字面意思。两个bbox的交集和并集的比例,详见上方GT的插图,灰色区域为交集面积,除以两个bbox的并集就得到交并比。交并比是衡量检测生成bbox的准确性的一个指标。现在也出现了一些其他用于衡量预测框质量的指标。

  1. 正/负样本(positive/negative sample)

在目标检测中,包含了目标的bbox就是正样本,而没有包含目标即bbox内是背景的将被作为负样本。当然一些网络会对正负样本进行特殊定义。特别注意,我们是根据训练标注来生成正负样本的,并不是说训练标注中地GT就是正样本,其他区域都是负样本。我们会依据某种准则如与GT的IOU或置信度进而由得到正负样本。

我们的检测目标是杯子,假设网络生成了4个检测框,认为这里面有杯子,那么根据特定的IOU阈值如0.5,我们将生成的4个检测框分别作为正负样本对待;在这里红色框是GT,紫色和蓝色框因为和GT的IOU超过50%被记为正样本,绿色框则被判定为负样本(背景)

在阅读了低下的几个经典网络后再回来看看,你应该会对正负样本有新的理解。并且目标检测中有一个很大的、亟待解决的问题就是正负样本、难易样本的失衡。(在一张图片中,没有目标存在的区域占了全图的绝大部分,负样本即背景类占了所有样本的大多数,大量负样本在训练时会淹没正样本,导致神经网络失去鉴别能力对负样本产生过拟合)

正负样本的划分策略(用IOU来划分正负样本是一个合适的准则么?)和不平衡问题同样是目标检测的难点。

  1. ConvNet和特征图

ConvNet一般值的是神经网络的backbone(骨干网络),也就是负责提取图像特征的部分,即我们在上面讲解CNN的时候由卷积层和池化层堆叠起来的block。特征图就是经过n个block之后产生的“图片”,因为原图的特征被提取到这些小”图片“上所以我们称之为特征图。简言之经过卷积或池化处理的图片或特征都可以被称作特征图,上面记录着原图的某些特征。

  1. 感受野(receptive field)

从这个术语的名字应该可以略知一二。receptive field是高层feature map上的一个点的信息是由原图或之前的feature map中多大的范围/面积的像素所贡献的度量。换个简单句:原图上的某个像素或低层次的feature map是否和高层中的feature map上的某个点有关联/间接连接?再或者直接看图:

绿色区域是l2中左上角像素的感受野,而黄色区域是l3中间像素的感受野

有许多扩大感受野的方法如变步长卷积、空洞卷积、膨胀卷积、focus层等我们稍后介绍。其实这个概念应该在CNN部分就进行介绍,但是由于图像分类对此并不敏感所以将介绍移到了此处。

  1. NMS(非极大值抑制)

在生成检测框的时候,可能会出现下图这种情况,这时候我们要根据框的置信度或定位精度等信息筛去一些重叠区域很大的目标框,防止出现重复检测的情况。这里给出了NMS的大致介绍和实现源码:NMS 在目标检测中的应用。

  1. 性能指标与mAP(mean average precision)

mAP是目标检测中最常见的测试检测器性能的指标。在次之前先让我们看看混淆矩阵,这是机器学习中所有分类器都要确定的一个参数:

混淆矩阵和对应的概念含义

由混淆矩阵中的数据,我们可以通过各种运算来得到千奇百怪的指标,常见的如正确率、真阳性率、特异度、假阴性率(漏诊率)、Youden指数等。我们则选择mAP来指示一个目标检测算法的性能,mAP的计算需要查准率(true positive rate,precision: TP/(TP+FP) )和查全率(又叫召回率recall,计算通过TPR:true positive rate,TP/(TP+FP) )两个信息。这两个指标是一对矛盾指标,当查准率高的时候,说明分类器很少把负类分为正类,只遴选那些最优把握、正类置信度高的样本,显然分类为正类的样本TP+FP将会下降(置信度上升,分类器把更少的样本分为正类),同时查全率必然跟着下降,很多正样本将会成为漏网之鱼而被分类为负样本。当降低阈值希望尽可能多地把所有的正类的查出来,那么很多置信度较高的负类同样容易被分为正类,导致查准度下降。

找到曲线上的平衡点,即边际效用最大的点,当继续往曲线的右侧移动,查全率上升的速度将会低于查准率下降的速度

目标检测任务中判断预测框的正负样本一般以IOU是否>0.5为准则(这里的分类是说预测框里有没有物体,而对预测框中物体进行的分类则是直接通过置信度进行)。那么如何绘制处pr曲线?只需根据不同的分类阈值分别设定分类器并对样本进行分类即可得到PR空间上的一个点(一组测试对应一个查全率和一个查准率,也就是PR空间中的两个坐标),改变阈值就能在pr图上画出一些列的点,平滑地连接它们即可得到pr曲线。一般的测试方法是改变阈值每级递增0.05直到0.95,称为IOU=[0.5:0.95]。AP就是PR曲线下方的面积(为什么用曲线下的面积,而不用平衡点处的分类器性能作为指标?),称作average precision平均精度。而mAP是把每个类的AP曲线进行求平均,即可算出多分类的平均精确度。

但是mAP也有一个问题,就是正负样本的数量不同的时候会导致PR曲线严重失衡(从precision和recall计算方法想想为什么),而ROC和AUC就不会出现这个问题,因此这两个指标在分类器性能测试中也很常用。

得到ROC曲线需要查全率和误诊率两个指标。查全率TPR在上面已经介绍过,这里再介绍false positive rate:它被称做假阳性率、误诊率和伪正类率,表示所有被分类器判别为负类的样本有多少实际上是正类(假正类在所有判别为负类的样本中的占比,FPR=FP/(TN+FP) ),同样会反映分类的精确程度(同样你可以思考一下为什么在正负样本比例失衡的时候ROC曲线不会变形?)。

和PR曲线的绘制相同,我们需要设置不同阈值来作出一系列的点随后连接绘制ROC曲线。ROC曲线下方的面积AUC即area under curve同样也有一些含义。使用AUC值作为评价标准是因为很多时候ROC曲线并不能清晰的说明哪个分类器的效果更好(阈值不同且指标不同),而作为一个数值,对应AUC更大的分类器效果更好,其具体含义为:当随机挑选一个正样本和一个负样本,根据当前的分类器计算得到的score(置信度)将这个正样本排在负样本前面的概率。

下图中的(b)可以帮助你更好、更直观地理解ROC曲线的含义,竖线处theta即设置的分类置信度。

看此图理解起来没问题/有新的启发就说明你大概掌握了这些概念

如果还是没懂也没关系,就把它们当作衡量训练得到的预测器的好坏的指标即可,这个指标越大越好!

  1. 公开数据集

列出几个最常用的目标检测公开/比赛数据集:

  • VOC(visual object classes chanllenges)2005-2012

经典的计算机视觉竞赛,任务包括图像分类、目标检测、语义分割和动作检测。20个种类。 VOC07和VOC12最为常用。近年来被更大的数据集像ILSVRC和MS-COCO逐渐取代。

  • ImageNet 2010-2017

训练数据集包含500,000张图片,属于200类物体。由于数据集太大,训练所需计算量很大,因而很少使用。同时,由于类别数也比较多,目标检测的难度也相当大。一般不会用作检测网络的性能,二是作为训练使用。

  • MS-COCO(Microsoft common objects in context)2015-now

看名字就知道是巨硬家的。最大的特点是除了bounding box 注释,还给了segmentation标注。MS-COCO也包含更多的小目标(面积小于图像大小的百分之一)和稠密的目标。这些特征使得MS-COCO更接近于现实生活。MS-COCO已经成了目标检测家族中的实际标准。


了解了基本概念,开始上手目标检测吧。在接下来这三个经典网络的介绍中,我们尽量忽略细节上的推导如卷积的通道数和维度,特征图的大小等,而专注于介绍网络实现的思想和总体方法。因此笔者默认你已经了解卷积神经网络的原理和目标检测的基本概念。如果要深入了解并运用obj detection还是要仔细阅读这些经典网络的论文,精读它们的源码实现!

5.2.6.2. 经典检测网络之R-CNN家族
5.2.6.2.1. R-CNN

Rich feature hierarchies for accurate object detection and semantic segmentation

R-CNN为了减少候选区域的数量,在原图上根据不同的比例、颜色、纹理和形状划分出许多区域(例如使用k-means等聚类算法),随后再根据区域之间的相似性对这些区域进行合并,这一步被称作selective search,会得到大约2000个候选区域即ROI(2000是人为确定的让合并算法终止的值,也可以选择其他值,过大的值会导致很大的运算开销;太小的值又容易将小目标区域被合并而使得漏识别或定位不准);然后对这些区域通过取外接矩形的方法得到一个个可能存在目标的候选框,把这些候选框全部拉伸成正方形(保证cnn的输入相同方便投入相同的网络中);接着在这些区域上运行CNN得到特征图,将特征依次投入到多个的二分类SVM(为每个类训练一个SVM,输出只有是和不是)和角点回归网络中(一幅图像到4个坐标/8个点的映射,看作一个回归问题),得到每个区域的分类与目标框角点的坐标;最后再运行NMS(non-maximum suppression)后处理(Postprocessor)去除一些重叠的目标框,得到分类结果与预测框。

通过selective search得到的区域图示,对这些连续的色块求出外接矩形即可得到预测框

整个算法的pipeline如下:

R-CNN相比最原始的滑动窗口法,采用selective search的方法大大缩小了搜索空间,只在那些有可能有目标的地方运行CNN,使得基于神经网络的目标检测算法变得可用(原来的算法复杂度根本不是多项式时间内可以解出结果的),虽然处理一帧图像的时间为50s量级(怎么还是那么长啊喂),不过相对之前的滑窗方法已经有了极大的提升,R-CNN的成果也让很多目标检测研究人员把目光从传统特征匹配转向了深度学习。

R-CNN也是首个提出用训练好的图像分类网络(R-CNN使用的是pre-trained的ImageNet)作为检测网络的Backbone再在之后的训练中进行微调(fine-tuning)方法的paper。预训练+微调的方法也一直沿用至今。

虽然R-CNN已经大大降低了目标检测的工作量,但是其动辄百秒的检测时间还是令人难以接受。其痛点不仅在与速度,在特征获取和复用上也有很大的问题:

  1. 没有复用特征,直接在原图上裁切子图,会出现大量的区域重叠,而R-CNN又对每一个ROI都运行了一次CNN,从而造成了非常多的重复计算,还是过于暴力了。
  2. 直接对ROI生成的外接矩形进行拉伸使其变成正方形,目标的特征可能会有损失、变形。

既然已经有人开了先河我们也分析了优缺点,那么,有请下一位选手Fast R-CNN登场(名字也够直接的哈,我R-CNN不要面子的吗)!

5.2.6.2.2. Fast R-CNN

论文戳这里:Fast R-CNN

废话不多说,直接上pipeline:

既然还是叫R-CNN,那么我们就对比着R-CNN来看看这个Fast到底有什么能耐吧。它仍然使用了selective search的方法来生成候选区域,不过它不再把每个ROI区域都投入相同的CNN中,而是先直接对图片运行CNN,得到整张图片的feature map,然后把由原图经过selective search得到的ROI根据映射关系(图像在经过ConvNet后会按一定比例缩小)对应到feature map上,也就是上图的ROIs from a proposal method。以上的操作便改进了R-CNN的第一个缺点,避免了大量重复的卷积运算,让所有可能存在目标的预测框共享了很多特征。

紧接着使用对这些ROI使用ROI Pooling的技术,把所有的ROI通过这种特殊的Pooling缩放成相同大小以便投入之后的全连接层,这种技术也尽量避免了强行缩放带来的特征变形、损失(虽然还是在一定程度上有影响,之后版本的Mask R-CNN所采用的ROI Align通过双线性插值进一步改善了这一步特征提取)。

  • 此处mark,我们会发现进行检测的时候只使用了对应ROI的feature,feature map的其他部分不会被检测头使用,因此Fast R-CNN和之后的Faster R-CNN只提取了前景信息而忽略了和目标一同出现的背景,在之后的 yolo 算法的误差分析和优缺点整理中我们会对这个mark进行分析。
ROI Pooling图示,例如这里想要采集2x2的特征,就把对应区域的特征图(这里是5x7)尽量平均地分为四个区域(3+2=5,4+3=7,如果需要得到特征的长宽刚好是feature map的长宽的公约数就能平均分了)

最后就是对FC后得到的特征进行分类和bbox回归了,Fast R-CNN使用线性回归+softmax的方法替代了原来的多个二分类SVM进行目标分类,这样能引入类间的训练竞争;bbox回归还是和原来保持一致(毕竟当时没啥可以优化的),这两部分工作是并行的。

经过这一通操作,不仅提升了识别的准确率,还直接把平均单帧处理时间干到了2s!几十倍的速度提升已经非常可观了 。但是还是达不到实时性的要求,怎么办呢,我们仔细分析一波Fast R-CNN存在的问题:

  1. 仍然使用selective search,这部分工作在整个算法中占了大概七八成的时间,对上千个的区域进行运算、合并的开销还是挺大的。
  2. ROI pooling这部分会损失精度,还有提升空间。

因为我们的精神是更高更快更强(卷就完事了!),所以请出下一位选手Faster R-CNN!

5.2.6.2.3. Faster R-CNN

Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks

这个网络相对于Fast R-CNN的改动就比较大了,最重要的贡献是提出了RPN(region proposal network)替代selective search,用于提出候选区域,大大降低了时间开销并且提高了识别精度。同时由于不再使用selective search,为了保证对不同尺度(大小)的目标的检测效果,加入了anchor作为目标尺度的先验信息。听不懂?先看看pipeline:

和Fast R-CNN一样,先通过ConvNet得到整张图片的特征图,随后将此特征图投入RPN中,而RPN的输出是可能存在目标的候选区域,既然是网络那么还一是一种映射:特征图到候选区域的映射。这里得到的候选区域数量就是可能存在目标概率最大的k个区域,至于如何得到请看之后的讲解。得到候选区域后,还是对这些ROI进行特征提取(ROI pooling),之后的流程就和Fast R-CNN一模一样了,我们重点说说RPN。

为了理解RPN的工作原理,就必须先了解anchor。前文中的滑动窗口检测我们提到用大小不同的滑窗来遍历整个图像,分别投入CNN进行检测。使用大小不同的滑窗就是为了对应不同尺度的目标,而遍历图像则是在搜索空间中寻找可能的目标。Selective Search是通过合并相似的区域来进行区域提议(生成候选的ROI)。显然在这个过程中我们都加入了先验的知识来确定目标,seletive search就认为属于一个目标的区域的纹理和颜色等特征应该是相似的,由此减少搜索空间的大小,即使是滑动窗口法也加入了一些假设如目标的大小。这些先验信息都以超参数的形式加入检测网络(也就是“炼丹”中需要认为修改的参数)。那么RPN就是利用anchor引入先验信息从而缩小搜索空间。首先我们把图像划分为一定数量的格点,然后以每个格点为中心,生成一组比例和长宽各不相同的候选目标框(有同学会想,这还不是在抓瞎吗,这要怎么知道候选框里有没有目标?让我们先保留这个问题)下图是anchor的形象解释。

anchor就是在图像上预置的一堆大小长宽形状不同的候选框,图源知乎-emiya-目标检测算法设计思想
在Faster R-CNN的RPN中,我们使用9种anchor,在每个格点上生成这样一组anchor

Faster R-CNN在每个格点上采用了三组不同尺寸的anchor,每种尺寸又拥有不同的长宽比例,即上图所示的绿色红色蓝色三组。

在进入RPN前,由ConvNet得到的feature map为50x38,于是我们对此feature map的每一个点都设置三组共9个anchor(注意要映射回原图,既然是ROI提议,那么anchor肯定是设置在原图上的,因此要进行一个比例缩放)

上图中右边密密麻麻的就是生成的anchor了,好家伙算一下总共有17100个ROI!到这里为止,采取的方法看起来仍然是“瞎猜”并且候选位置还是太多;而且即使是使用了这么多的anchor,有些物体还是没办法被很好的框住,有可能没完全框住,也有可能是框太大。下面就进入筛选anchor和修正anchor环节。以下给出RPN的结构:

RPN网络的结构,图源知乎-白裳-https://zhuanlan.zhihu.com/p/31426458

进入红色框之前的是ConvNet得到的feature map

首先从对ConvNet得到的特征图再进行一次3x3卷积降低参数量,然后分别投入到两个并行的网络分支。上面的分支通过一个1x1卷积再经过softmax函数,判断每个anchor存在物体的概率(不管是哪一种目标,只要有目标在里面就行,即只做有/无目标的二分类)。那么第一个问题解决了,我们可以通过物体存在的概率来对anchor进行初步筛选。但是你可能还会想,为什么就得到了每个anchor区域存在物体的概率了?关键就在于3x3卷积和1x1卷积构成的这个小网络,我们会通过训练,让这个网络成为一个回归器,之后再通过softmax函数得到每一个anchor包含物体的概率。(映射!还是映射!网络结构中的 reshape暂时不用管他,这个是代码实现的时候需要关心的问题,reshape后可以方便softmax函数的计算,算完后再reshape回去罢了)

接下来看第二条分支是怎么解决anchor对物体的贴合度不够的问题的。这个分支会习得一个修正的anchor使得anchor能够更接近物体边框的映射,也可以说是原anchor和ground truth的offset。以下面这架飞机的检测为例,绿色的框是训练标记,红色框是RPN生成的anchor中最接近标记的那一个,显然此anchor还是只能覆盖飞机的主体步部分,机翼没有被包含进来。

如果要通过变换来使得生成的红色预测框贴近绿色框(GT),最简单的办法是什么?那肯定是先平移到合适的位置,然后再进行一定比例的缩放。如果anchor和gt差距不大,那么这种变换很显然是可以被近似成线性变换的,因此我们假设生成的anchor中最接近gt的那个已经和gt框很相似了,只需要用一个平移+缩放来进一步贴近它。所以我们会通过第二个分支中的1x1卷积来学习这种变换关系,即对bbox中心的平移和横向纵向的拉伸比例(当然,也可以直接学习ancho的四个角点和GT的四个角点的偏移量即四个角点)。那么经过训练,这部分网络就会学习到一个对anchor进行修正的映射关系了。网络会为每个anchor都输出一个数组(属于是量身定制了)。

ok,两步分支结束,接下来我们就要把对anchor的筛选和对anchor的修正给综合起来,为下一步的bbox四角点位置回归和分类提供候选区域。请看RPN结构中的Proposal层,它大概有如下几个步骤:

  1. 根据所有anchor包含物体的概率,对anchor进行筛选,得到概率最大的n个候选框(n还是一个自选数)。
  2. 利用第二个分支得到的[dx,dy,kx,ky]对这些筛选得到的anchor box进行修正。
  3. 对超出图像边界的anchor进行裁切使得它其中一条或两条边变成图像的边界(在生成anchor阶段,处在边缘的格点生成的较大的anchor必然会超出图像边界;同时在进行anchor box的修正时,也有一部分anchor会因为变换而出界)。
  4. 去除特别小、比例异常的anchor(经过变换后可能会出现这样的问题)。
  5. 对剩下的anchor进行非极大值抑制。
  6. 从剩余的anchor中随机选取128个作为正样本,再在被筛选调的anchor中选128个作为负样本,把这些anchor作为候选区域,交给Faster R-CNN的下一个部分。至此,RPN网络即候选区域提议的工作结束。

下一步就是和Fast R-CNN一样,通过ROI Pooling提取候选区域对应的feature map上的特征,把特征连接到FC层,再分别投入bbox角点回归网络和分类网络,得到更精确的目标框和目标类别。可以看到,相比于Fast R-CNN变动最大的就是利用RPN替代Selective Search这一步,利用网络来提取深度特征(用于生成候选框的特征),大大提高了提议区域的精确度;同时也避免了逐像素、逐区域的合并工作,提高了速度,平均单帧处理速度来到了0.2~0.4S!

Faster R-CNN也首次完成了目标检测任务的“深度化”,即整个检测流程都使用神经网络连接完成;在高速的同时成为了当时的state of the art(检测任务竞赛的第一名);这项工作基本奠定了区域提议+ROI-CNN(区域分类网络)的two-stage检测网络结构,从此全面开启了基于CNN的目标检测。两阶段的网络也让监督信息(设计的loss function和训练数学)能够在不同层次更有针对性地对网络参数的学习进行“指导”。

Faster R-CNN的成果在当时是振奋人心的,终于让大伙儿看到了基于网络方法的实时检测的希望!不过一秒3、4帧的速度在实时场景还是难堪大用,并且Faster R-CNN也存在一些问题:

  1. 和Fast R-CNN一样使用ROI Pooling丢失了一些特征。
  2. 属于two-stage,分为区域提议和分类+回归两个阶段而不是端到端(end-to-end)的方法。RPN的速度还是不够快。
  3. 只从ConvNet的最后一个特征图上提取特征,分辨率较小因此对于多尺度、小目标的检测问题没办法很好地解决。
  4. 筛选候选框使用NMS而分类时使用softmax,这样可能会把有部分重叠的目标或是大目标框内的小目标给筛除。

5.2.6.3. 经典检测网络之YOLO

You Only Look Once: Unified, Real-Time Object Detection

此时此刻恰如彼时彼刻,一位大佬在阅读了Faster R-CNN 中PRN设计的思想后直接拍桌怒起:既然RPN能直接根据深度特征提取出可能存在物体的ROI,生成提议区域,那我干脆直接就用网络直接回归出bbox的坐标不就得了,一步到位岂不是美滋滋?(两阶段的方法相当于先通过RPN大致定位bbox,再在后续的bbox回归中获取更精准的定位框,one-stage的方法直接在特征图上进行bbox的回归)这就是YOLO的设计思想。YOLO是you only look once的缩写,“只看一眼”也昭示着这是一个一阶段的端到端网络,不需要中间的区域提议阶段!

  • YOLO

作为one-stage模型的开山之作,yolo直接砍掉了RPN,通过连续的几个卷积层直接回归出bbox坐标和对应的分类。首先一起看看yolo的pipeline:

相比之前介绍的R-CNN家族,是不是感觉一下清爽了不少?从设计结构来说,整个算法的运行就只有3步:

  1. 缩放图片至ConvNet的输入大小
  2. 卷卷卷...
  3. 得到每个框的分类和置信度
  4. 运行NMS去除重复的目标框

有些同学肯定会疑惑,怎么就得到了回归坐标和分类呢?请牢记一点:网络只不过是会在训练过程中会学会一个映射!那么关键就在于我们应该如何提供“监督”来让网络学习这个映射。不过,还是要称赞一下如此简单明了的设计简直满足了所有美好的想像,如此的Unified、elegant!

请看YOLO的设计思想:

1.划分格点 2.得到预测框和置信度/得到每个格点的分类概率 3.综合2上一步的两个数据得到检测结果
  1. 首先还是通过密集采样的思想,把图像分成SxS大小的一堆格点,训练时每个单元格将会负责检测那些中心落在对应格点中的目标,即GT的中心落在哪个格子,那个格子就负责预测(初看yolo的教程的同学都很有可能有这样的问题,训练的时候你有训练标记,能够知道每个目标的中心在哪里;可是在推理预测的时候怎么目标中心在哪呢?答案是不用官管!我们会把每个格点都当作目标中心,一会还会提到这个问题)。

  2. 由前面简介所述,yolo直接回归得到bbox角点和分类,那么对于bbox的预测,假设每个格点会预测N个检测框(不同的规格和尺度,类似于之前介绍的anchor的概念,也是先验的、用于缩小搜索空间的方法),还会得到这个框的置信度(定位是否准确?框内存在物体的概率?总之是这两者的加权值),此置信度在训练时通过生成的边界框与GT的IOU来表示,如果预测得到的边界框不存在物体则此项为零(网络认为这里没有目标那么也没必要计算IOU了)。这一步骤中得到也就是上图中的Bounding boxes+confidence。

  3. 我们还需要为每个格点生成的检测框预测其属于C个类别的概率值(当然也可以训练一个softmax分类器来计算每个类的概率,不过既然网络可以直接学会一个图像到分类概率的映射,我们干脆就直接让网络输出对应预测框属于每个类的概率吧!读者也可以思考对比一下两者的优劣),有些网络会分为C+1个类因为它们把背景也作为一种分类,不过我们前面已经在bbox的预测中增加了置信度这个指标,那么当预测框中没有目标的时候此项为0,这样就不需要一个额外的分类了。这步对应上图2中的生成class probability map。

那么,根据上面的介绍,我们最后总共需要网络输出SxSx(5*N+C)个值,yolo的paper选择的值是s=7,n=2,C=20,即把原图划分为7x7的网格,每个格点预测两种不同大小的预测框,总共检测20种目标,5是框的置信度和中心坐标、框的长宽。因此网络最后会输出7x7x(5*2+20)个值。此处mark,稍后再提如何进行后处理。*

讲完了整体设计思想,你应该能明白为什么我们把每个格点都作为可能的目标中心了:因为每个格点最后都会产生5N+C个值!即使在实际推理的时候这个格点里面没有目标存在也没事,算就完事了!最后在得到预测值的时候,没有目标存在的格点要么bbox的置信度会很低或者为零(表示这个框里啥也没有),要么就是这个bbox属于每个类的概率都很低,在之后的NMS阶段将会被删去*。因此注意训练和推理时的区别,我们只是假设每个格点都有物体,最后是否真的有物体,就看输出得到的置信度和分类分数有多高了,笔者称之为”将计就计“!

看完了网络的设计思想,终于可以看看网络的结构了:

前面就是一堆卷积层+池化层构成的block,不断地缩小feature map的尺寸(细心的你会发现,虽然feature map横向尺寸变小但是通道数是在增加的,think about it)。倒数第三个block结束后,这里的特征已经非常“高级”了,与是把它们接入全连接层充分利用特征之间的关系,随后连接到一个7x7x30的张量,每个值就代表着我们的需要的那些输出。

ok,此处呼应上面忽略的后处理mark,来讲讲后处理:首先从对所有预测框进行筛选,设定一个阈值把那些“假”预测框筛选掉(也就是对我们假定有目标中心的那些格点进行筛选,具体怎么做?用得到的bbox置信度乘以那个bbox对应的C个类别概率!);剩下的就是预测的比较准、类别置信度高的bbox了,再对这些bbox运行NMS,得到最后的预测结果。

一顿操作猛如虎,YOLO将bbox回归和分类任务并行完成且一步到位,将Faster R-CNN中的RPN也直接给“融合”进来,进一步减小了时间开销,在Pascal VOC2007和2012上得到了63.4的mAP和45FPS的速度,成功实现了实时检测!不用想也知道之后one-stage的模型就开始井喷了。

YOLO的作者还对比Faster R-CNN进行了非常严谨的误差分析:

对比可以发现,YOLO出错的地方大多是Localization错误,即bbox的回归框不准确,每个格点只生成了两个b预测框并且两个预测框预测的是同一个种类;而Faster R-CNN在这方面要好很多,毕竟它有两次修正回归框的机会,先提出候选区域再在其上进行bbox回归。

  • 此处呼应前文 Fast R-CNN处的前景背景分析mark

不过有得必有失,Faster R-CNN在背景分类上出现了很多错误,而YOLO将背景错判为目标的概率只有4.75%,这就很有意思了。从它们俩的结构差别我们应该也能推断处一些端倪:Faster R-CNN在进行分类的时候只利用了RPN生成的ROI上的特征,而YOLO从头到尾都是ConvNet的结构,在高层feature map上接近全局的连接获得了更大的感受野,因此也充分利用了全局的特征,不容易将背景分为目标。

由于YOLO没有resion proposal阶段,因此基本没有机会修正正负样本不均衡的问题只能通过一些启发式的技巧最大程度降低负样本过量的影响,而Faster R-CNN在RPN提出候选区域后就能根据生成的候选框与GT的IOU来筛选正负样本使得正负比例约为1:3(作者经过测验发现这是一个比较好的值)。

YOLO还有一个缺点就是对于小物体的检测很容易忽略,7x7的格点对于小目标来说还是太大并且YOLO没有机会修正检测框。


5.2.6.4. 目标检测小结

至此我们已经介绍了4个检测网络,分别是two-stage的代表R-CNN家族和one-stage的开创者yolo,现在可以先小小地来总结一下目标检测网络的设计思想和改进方向了。不过细心的你应该已经发现,我们没有介绍如何训练网络(即loss function的设计思想和具体实现,如何在训练过程中给网络的不同部分提供监督信息?),这其实也是很重要的一部分,我们将在总结完以后大致介绍一下loss funciton和网络pipeline的对应关系。

首先笔者想要再强调一下神经网络作为一个万能的函数拟合器的作用:只要合理的设计网络的任务和其损失函数并且提供对应的训练数据,它就能够完成你想做的事!从Faster-RCNN和YOLO的实现结构我们就可以看出,这些block难道有哪个是有实际物理意义的吗?显然没有,即使使用相同的架构,我也可以通过修改 loss functoin和任务需求来决定最后网络的输出是什么、具有什么意义。因此在设计网络的时候,最需要关注的问题是我们要如何把检测任务拆解成合适的几个子任务和模块、怎么样设计最有效的 loss、使用怎样的策略来减小搜索空间、如何去根据检测任务的特点来有针对性地给网络各个部分赋予实际意义。上面说的这些都是在给网络加入先验知识,告诉网络应该如何更好地去学习目标的特征、完成学习。

R-CNN和YOLO实际上就确定了以CNN为核心算子的目标检测的发展路线:anchor-free和anchor-based,two-stage和one-stage,后续有大量改进都是针对这两组四种方法展开的。

很多人说神经网络只不过是在搭积木、炼丹,此言差矣。诚然大部分时候我们需要花大功夫去调试去改参数,但是怎样设计一个网络还是有很大的学问的。现在也有很多科研人员、很多学者尝试让神经网络的设计更具有逻辑性,通过数学的语言更科学地去建模问题、用更成体系的理论去分析网络结构和模型,减少启发式的设计而更多地去基于模型来调参等等...... 和存在已久的传统医学现代医学之争一样,联结主义也一直为人们诟病缺少解释性,被符号主义和行为主义打压(虽然一直被打压但是神经网络还是以不可阻挡之势统治了当前AI的发展,唉呀妈呀真香~),笔者相信随着科研人员在可解释性这条道路越走越远(比如现在火热的可视化),神经网络最终会拥抱科学,完成华丽的蜕变(我们这些小noob就乖乖炼丹把evolution的事情都交给大佬们吧,不过也许你就是将来的大佬!)。

最后笔者想要再次强调强调强调:神经网络是一个万能的函数拟合器,拟合成什么样关键看你的设计!这里所说的设计也就是loss function的设计,LF能够用于衡量网络学习效果的好坏,又作为学习的优化目标,因此我们依据最小化LF的原则使用梯度下降+反向传播对网络的权重进行训练。

所以接下来让我们把网络需要实现的任务的设计思想具体化,看看其中最重要最关键的一环:loss function是如何设计的,并介绍一些对loss的优化方法。这里将会介绍MSE、交叉熵、FL、FCOS、IOU loss等经典方法。


5.2.6.5. 损失函数的设计与改进
5.2.6.5.1. MAE (mean average error)和RMSE(root mean square error)
  • MAE即平均绝对值误差,计算得到的是各个样本输入得到的结果相对于对应的标签的差的绝对值,这也是计算误差最直观最简单的方法,输出结果和标签差了多少,我们就记多少误差。

从数学计算角度看,MAE的结果就是残差的平均值。例如有3个西瓜样本的数据x1,x2,x3,每个样本都是一个5维的数据,即:[大小,花纹宽度,重量,颜色深度,拍击是否有回响] ,标签分别为30%,60%,90%将他们输入一个神经网络,希望该神经网络输出的信息为西瓜的甜度;那么,假设该网络对应三个输入的输出值分别为25%,55%,96%,计算其MAE就得到损失值为16/3=5.33。从损失函数这个取名我们也可以看出,其含义就是对于每一个和标签不同的输出,将会产生多少的误差(损失)。

对于目标检测网络,我们要输出目标框的位置(x,y,w,h)和分类,因此标签也需要有对应的5个值。若采用MAE来计算损失函数,那么就是在一个batch结束后,分别统计每个待估值(需要得到的预测值)对于该batch各个样本的平均绝对误差,让损失函数对xi求偏导再将得到的结果作为反向传播的初始值,进行BP训练。分类只有是和不是,因此是一个二元的损失,但你仍然可以使用MAE作为分类损失函数,虽然它不是太好用。MAE还有一个缺点,就是在原点处不可导。

  • RMSE和MAE的区别就在于,他会将误差求平方之后再求平均并在最后开方。

我们之前已经提到过,设计不同的、有针对性的损失函数是神经网络改进的关键,那么RMAE相对MAE又有什么改进呢?可以看到,MAE对于不同的样本的损失采取的策略是“一视同仁”,即不论单个样本产生的损失值有多大,最后都是获得相同的权重。对于那些损失小的样本,我们会认为该样本网络已经学习得比较好,不需要在这些样本上花费过多的精力;而对于损失值较大的样本,更应该“照顾有加”。那么有没有什么办法实现这种操作?铺垫了这么多,RMAE肯定就是一种能实现该需求的损失函数。每个样本在最终损失中的贡献会随着该样本损失的增大而增大(取了平方,不再是线性关系,相当于y=x^2和y=x比较)。因此RMAE会对训练得不好得样本施以更大的惩罚。但是同样,对于离群值(在这里是特别大的异常值)将会有一个巨大的损失,这将会误导模型的训练,MAE就没有这个问题。

此处特别注意,并不是 Loss越大,回传的梯度就越大,梯度只和 Loss function的导数(梯度)有关!网上有很多教程都出现了这个讹误,显然我们从梯度下降法的原理出发就会发现,我们把LF看作是权重ω和b的函数,分别对它们求偏导以得到当前的梯度值并根据往梯度方向更新权重,而LF在参数空间中某一点的大小和梯度是完全无关的!在下面介绍focal loss的时候我们还会再提到这一点。

此处mark。

不过不加开方也是可以的,这时候得到的值为MSE(mean square error),公式类似于我们常说的方差(不过方差计算时的yi是这组数据的平均值,MAE的意义显然不是数据的分散程度)。RMSE相对于MAE,由于在最后进行了开平方的操作,就不那么容易产生梯度爆炸的问题了。

聪明的你应该想到,如果我想让大的更大小的更小,或者让大的不要那么大小的不要那么小,只需要改动误差上的次幂数就行:若选择损失的绝对值的三次方,那么误差大的样本的权重将进一步上升;而选择1.5次方,得到的结果将会比平方更缓和。(那么次幂小于1大于0又是什么情况?)

此外,RMAE/MSE还有一个较大的缺点,在使用sigmoid作为激活函数配合使用时(即用于分类),收敛速度会非常慢。sigmoid激活函数在特别大或接近零时的导数几乎为零,这就导致当预测值落在这些区间时产生的误差非常小,而RMSE又刚好进一步减小了本就很小的梯度(列式计算一下L对xi的i偏导即可,如下图),因此在输出层,神经元的学习速度会非常缓慢。不过好在除了这种方法,还有一些更佳的分类损失函数设计策略(也就是下面要介绍的交叉熵,另外,ReLU系的激活函数几乎已经完全取代了sigmoid,这样在输入激活函数的值大于零的时候会有一个稳定的导数值,不过还是不推荐使用MSE作为分类损失函数)。

使用MSE作为分类损失函数时求导得到的结果,激活值为0/1或接近0/1时梯度会非常小
sigmoid的导数

从上面的分析中我们可以得出,这三者在目标检测中一般用于设计距离损失函数,用于计算预测得到的bbox和groundtruth的差别,也就是常说的回归问题。对于分类问题就要使用其他的损失函数了。


5.2.6.5.2. CE/BCE(cross entropy/ binary cross entropy)

既然已经有了距离损失函数,那我们就来看看交叉熵在分类应用的效果如何吧。这篇文章:损失函数:交叉熵详解 从信息熵的前世今生介绍了交叉熵损失函数的由来,强烈推荐。BCE和CE的区别在于一个用于二分类,一个用于多分类,CE就是BCE的多分类推广形式。上面介绍MSE的时候已经提到,我们希望在样本预测值偏离标签大的时候拥有更大的损失值,而RMSE或MSE恰恰在分类任务上对于这点做的不好,因为分类只有错误和正确(0,1),这两者在这些区间的产生的损失值太小不利于网络的训练。

详细的推导的上方的教程里有,不在此赘述,直接看看BCE的形式:

此处时σ(xi)就是将网络最后一层的输出值送入sigmoid函数得到的(将输出归一化为概率,区间为[0,1] )。L由两项构成并且总有一项是零。

求L对xi的偏导,就能算出其结果恰正比于预测值和标签值GT的差。这样就解决了MSE存在的问题。不过前面提到,我们希望让误差大(分类分得不好)的样本拥有更大的权重,CE就无法做到这一点了,欲知如何解决请看Focal Loss。

多分类问题只需要对BCE进行简单的推广,在此直接给出其形式:

其中

即为softmax函数。

对比BCE的损失函数,细心的你会发现(1-yi)不见了,比如要分为 (apple,banana,pear) 这三个类,最后得到的向量为 [0.2 , 0.5 , 0.3] ,而标签为 [0,1,0] 即最后的分类为banana,apple和pear虽然分别有0.2和0.3的置信度但产生的误差即-0.2和-0.3却不会计入总损失!这样岂不是没办法改善假阳性的问题吗?别担心,我们使用的是softmax函数,由于归一化(除以各个分量的综合),它会让输出的向量各个分量的合即概率合为1,因此提升对yi=1的概率自然就会降低其他分类的预测概率了。

实际上,最小二乘问题中若数据的噪声服从高斯分布,那么该问题可以转化为最大似然估计的问题。而交叉熵损失函数也是在最小化真实分布和模型输出的分布的距离,因此本质上也是和最大似然估计等价。殊途同归怎么说。相关内容网络上有许多介绍,这里不再赘述。


5.2.6.5.3. FL(focal loss)

Focal Loss for Dense Object Detection

在介绍交叉熵损失的时候我们知道虽然它改善了输出层梯度消失的问题,却无法赋予不同的样本以不同的损失权重。这个问题在one-stage检测器中尤为严重,如前文的基本概念中介绍的那样,目标检测的正负样本严重失衡,存在大量的背景(负类),而有目标存在的正样本区域则是少之又少。two-stage检测器如Faster R-CNN等可以在区域提议阶段(RPN,因为提议网络实际上已经把那些简单的易分样本给筛选掉了,尤其是背景类)之后对正负样本进行筛选使他们的比例保持在1:3左右(经验值),而one-stage由于没有类似RPN的结构因此无法调整正负样本的比例。

并且,负样本又常为易分样本(背景的提取出来的特征和目标特征通常差异很大),正样本由于数量特别少而无法主导损失和梯度,因此模型会往我们不期望的方向发展。在得到MSE的启发后,自然而然,我们需要为此设计一种损失函数,能够提高正样本和难样本的权重,降低简单样本对损失的贡献从而修正这个问题。

Focal Loss将损失函数设计成如下的样子:

令:

得到统一的表达式:

在这里,描述了预测值和标签的接近程度,越大说明分类越容易(准确度越高)。经过的调制,容易分类的样本的损失贡献将会下降很多,而难分类样本的权重同样会降低(但是降低得不如简单样本那样多),最后的效果就是loss将会倾向难分类样本。

论文原文中不同γ取值的Focal Loss对比CE Loss

此处回应 RMSE中的 mark:可是上面我们强调,损失的大小和回传的梯度是无关的,那即使增大了难样本/正样本的损失值,又有什么用?损失函数的梯度和后者无关,这就说明即使LF的值很大梯度的模还是可能很小啊,因此可能会出现loss大然而梯度很小甚至为零的情况,那么权重就不会更新,难样本就无法影响学习的过程。这里就需要仔细分析计算一下了,我们分别算出CE和FL的梯度,得到结果如下:

怎么比价两者的大小?光看函数似乎很难得出比较的关系,彷佛又想起了高中数学求导比大小的恐惧... 没事我用matlab帮你画好了:

可以发现,在pt小于0.3的时候,FL相比CE将有更大的梯度,而在大于0.3的时候,就更小了

一顿分析猛如虎,我们得到使用FL对于难样本确实可以产生比CE更大的回传梯度,那前面说的”对loss有更大的贡献“到底有没有用?这里我们从两个角度入手考虑这个问题:

  1. 特定于Focal Loss的设计,从前面”论文原文中不同γ取值的Focal Loss对比CE Loss“的图上就可以看出,当loss越大的时候,FL的曲线是要比CE更陡峭的,这就说明了FL在值小的时候会比CE拥有更大的梯度值,而在大的时候前者的变化趋势会比后者更平缓,也就对应了我们刚刚对其梯度的分析。
这里只放了γ=2时的FL和CE,比前面的图更清楚一些

这是一种巧合吗?似乎也不是。当等于1时,我们必然要求loss=0(预测值和标签完全相同,说明很好的拟合了需要的函数),那么,如果我们要求在小的时候拥有较大的梯度,在大的时候梯度必然会相对更小。举一个实例更直观地理解:让汽车从静止开始在固定时间内加速到10m/s,起始的加速度越大,你在后程就必须松更多的油门(减小加速度),对应loss就是一开始有很大的梯度(斜率大),而在的时候loss必须为零,那后程的斜率就会相对减小了。

从这里的分析我们也会发现,其实log本身就对有一定的”focus“效果,CE曲线对应的损失和梯度也是在小时更大,大时更小,只不过效果没有FL那么明显罢了。

要从数学上严谨的证明也不是什么难事,若该函数是单调 的,在loss-pt平面上取(0,loss0)和(1,0)两点的连线,若开始时函数的斜率大于此直线的斜率,后面就一定会小于此直线的斜率(凸?凹函数?)。直线对应的就是CE的梯度。

  1. 抛开FL的函数形式分析,难样本的贡献大是否有助于网络对难样本的分类?

这部分主要是笔者的一些感性的、直觉性的想法,请戳:样本对Loss贡献越大,网络会对它”多多关照“吗?。并没有严谨的数学分析,仅作参考!


5.2.6.5.4. GHM

Gradient Harmonized mechanism for balancing different examples

上面介绍的Focla loss为难易样本施以不同的权重从而让网络倾向难样本分类的训练,一定程度上解决了样本分布不均衡的问题。而GHM(Gradient Harmonized mechanism)则直接从样本产生的梯度分布入手,从这个角度看待样本数量和难易程度的不均匀问题。

笔者私以为FL其实颇有一点“瞎打误撞”的感觉,原文虽然在最后分析了CE和FL的梯度对比,但是却是从样本产生的loss引入的。很多博客和教程、视频都认为难样本loss越大则模型权重会往相应的方向更新,但其实这种说法是知其然不知其所以然,从原理上是绝对错误的。对网络的训练过程不甚了解。相信你看完了这一小节,会对loss的设计有更深的理解。

以分类任务为例,希望难样本能够主导网络权重的更新,而不能让简单样本淹没了不容易分类的样本。我们在focal loss已经分析得到,难样本将拥有更大loss贡献,同时也会有更大的梯度,而后者则是问题的关键:难样本虽然会产生对应方向上更大的梯度,但是巨量的简单样本产生的梯度叠加起来就会导致网络向简单样本倾斜。

第一幅图中,显示了作者统计的所有样本的梯度范数的,注意纵坐标是对数坐标。梯度范数的就是这个样本在loss function上产生回传梯度的模值。不出所料易分样本(左侧Gradient norm很小的样本)的数量比其他都高出了几个数量级,积少成多,简单样本的总梯度共贡献会超过难样本。并且,经过统计还发现,部分特别难的样本(gradient norm右侧,那些在分类中近乎完全错误的样本),也会产生异常的梯度,作者认为这些样本是outlier(离群点,标记错误或由于噪声产生的异常值),所以同时也要修正这个问题。

那有什么办法在这种先天不公平的情况下把一碗水端平?只需要选择一个合适的权重,让他们有相同的梯度贡献,那么问题的答案也就呼之欲出了:用该样本的梯度除以拥有相同梯度样本的数量,最终得到的统计曲线就如中间的图像所示。那么,我们只需要将原来的样本分布乘以得到的调制系数,便获得了右侧的调制后的梯度。用GHM对比CE和FL,显然GHM在易分样本上有很好的抑制效果,同时对于异常多的难分样本(outlier)也同样有抑制效果。

那么应该如何计算样本的密度分布(也就是拥有相同或类似梯度样本的数量,这可以用一个分布函数来表示,但由于现实世界中的采样是离散的,要进行一些处理)?请看论文原文:1811.05181 Gradient Harmonized Single-stage Detector (arxiv.org)。原文中还提出了一种对于bbox回归的梯度均衡化损失函数,并且对梯度密度函数的计算进行了简化和近似,以降低复杂度加快训练速度。


5.2.6.5.5. OHEM系列
  • OHEM

全称为On-line Hard Example Mining,即在线难样本挖掘。此方法一般用在两阶段网络上,通过选择当前网络预测效果差的样本进行反向传播,以提高网络的性能。

实际上,虽然想要解决的问题不尽相同,但笔者认为OHEM应该算是标签分配(label assignment)问题的始祖。

在OHEM被引入之前,传统的机器学习中就有用到改变样本权重的采样方法,最典型的就是Adaboost,不过其建立的是级联检测器。在SVM的训练中,一些简单样本往往离分类超平面很远,这些数据点对于优化目标的作用几乎为零,甚至可能起到反作用。因此,HNM(hard negative mining)被提出用于加速SVM的训练,同时提高性能:经过一段时间的训练之后,只采用那些被错分的、离超平面距离近的样本进行训练;每迭代一次都更新超平面参数(更新模型),然后在新的模型下重新运行HNM,挑选出新的负样本。如此往复直到模型收敛。

然而,倘若对深度神经网络也是用HNM,开销就太大了:我们不能等到把所有样本都计算一遍之后再区分难易样本,不仅训练速度慢,就连区分样本都要花很多时间。因此,结合随机梯度下降和mini-batch法,直接让网络在每一个batch中都挑选出难易样本,并在线更新模型(这是相对HNM而言的“在线“)。实际上,前面介绍的Focal Loss就是这么做的(这也可以看作使用mini-batch训练的必然结果)。现在来看看OHEM的结构:

OHME以Faster R-CNN为baseline;上方是一个read-only head,下方模型的参数会正常更新

选取难易样本并进行更新的pipeline如下:

  1. 对于Backbone输出的特征图,使用RPN获取一定数量的区域提议(即ROI,假设这里取Top-2000个)
  2. 将所有ROI经过ROI Pooling后投入(a)head中,计算每一个ROI的损失;注意(a)在此过程中不会保留梯度信息(Tensor.requires_grad=Flase)
  3. 对这些ROI在(a)中的损失进行排序,取前300个ROI并进行NMS(防止单个目标的损失被反复计算,因为相邻位置的ROI经过pooling后生成的特征几乎先沟通,故网络后会输出相近的损失,从而出现梯度重复),然后将处理后的ROI输入(b)中计算损失并完成反向传播
  4. 把(b)中每一层的参数都复制给(a)
  5. 完成一轮迭代,回到1.

经过这样一番操作,我们就成功选出了那些丰富信息或者对网络来说困难的样本了(也就是上图中的diverse & hard roi sampler)。有些同学可能会问,既然上方的网络已经把所有ROI的损失都计算出来了,并且(a)和(b)的结构是完全一样的,那为什么还要复制出一个(b)来重复执行那些Hard ROI的前向推理操作呢?这是因为网络中的每一个参数在前向推理的时候,在backward propagation之前,梯度都是以累加的形式进行保存。如果希望在对loss排序之后再选出那top-k个损失最大的ROI,你就需要为所有的ROI都保存它们在前向过程中产生的参数(这些参数用于计算反向传播的梯度)。而保存这些ROI和它们相关的数据,空间代价是很昂贵的。另外,还是由于每个参数对梯度都是累加保存这个问题,虽然我们在(a)中已经计算过所有hard example了,但是由于相关参数没有保存,因此在选出ROI后还需要在(b)中再次推理。相比于保存所有ROI和它们对应的信息,直接将网络参数复制一份是更好的选择,这也是一种时间换空间的做法,并且作者在论文中通过消融实验验证,由于要反向传播的ROI数减少了,实际时间不增反降。

RPN实际上已经将那些不太可能包含物体的区域去除掉了,生成的区域提议ROI都是有较大概率存在目标的。因此在第一阶段,大部分简单样本和背景类都被去除。而OHEM则是进一步对ROI进行筛选,去除那些简单样本(作者也通过消融实验验证,简单样本对网络的改变非常非常小可以忽略不记),使得网络能够更专注于“错题”。

对比Focal Loss,OHEM相当于在two-stage的两个阶段都进行的难易样本筛选,最终获得用于计算梯度的那些难样本;而Focal Loss由于没有第一个区域提议阶段,它无法提前确定样本的难易程度,只能直接根据最后的损失大小,赋予不同样本以不同的权重。那么Focal Loss是否可以应用到两阶段网络?答案是没有必要。因为RPN已经将简单样本排除出去,剩下的ROI都是可能存在物体的区域,根据经验一般让这些ROI中的正样本和负样本的比例为1:3,就可以得到较好的结果,正负样本不会出现失衡的情况。一阶段网络没有RPN,导致其必须通过密集检测的方法尽可能地涵盖目标的搜索空间,不能放过任何一个角落以保证预测的召回率,因此导致会有非常非常多低质量的、简单的背景框出现。

本小节开篇提到OHEM像是标签分配的始祖。由于现在还没有讲到更多的anchor-free模型和标签分配问题,我们把标签分配和正负样本不均衡两者的对比放到5.2.6.7,若你已经对此有所了解,可以先跳转前往阅读。

s=stratify,意为分层在线难例挖掘。OHEM提出后效果确实不错,但是它将所有的难样本都一视同仁,并且在整个训练的过程中始终为分类损失和回归损失赋予相同的权重,只按照总loss对ROI进行排序。于是,s-OHEM的作者就从这里入手,进一步探讨细化的问题,提出了分层的样本选择方案。

s-OHEM的基本架构和OHEM是一样的,都是用一个只读网络用于对ROI损失排序,外加一个常规的head用于对只读head排序生成的Top-k的损失最大的ROI进行训练,一轮迭代后将参数复制给只读网络。s-OHEM将网络的训练分为四个阶段,即上图中的s1~s4。实际上分类任务和回归任务的难度是不一样的,产生的损失也不能一视同仁。因此它单独考虑类别损失和定位(回归)损失,在网络学习的初期,两类损失都较大;而经过一段时间的学习,分类损失会较快的下降,此时应该逐渐重视回归损失,以达到更高的定位精度。因此在选择难例的时候,s-OHEM实际上通过下式,在不同的训练阶段选择不同的调制系数(一般在具体实现上令以保证总权重为1):

那么怎么样确认某个任务已经学习得较好了呢?可以利用生成的难例产生的损失分布来进行判别,当loss趋于稳定的时候,就认为一个阶段的学习结束;此时改变抽样策略,修改系数使得网络进入下一个阶段的学习。为了让训练过程更加地平滑,我们可以让两种损失的系数连续地变化,而不是强行分割成4个阶段,这样应该能做到更细致的分类和更流畅的阶段过渡。

s-OHEM应该算是较早提出应当在训练的不同阶段对不同训练任务”因材施教“的方法。它也启发了后续一系列的动态Loss计算策略,以及前面提到的动态标签分配。在s-OHEM之后,loss的不同部分(一般是分类和回归)常常被赋予不同的权重以影响网络的训练。这些关于dynamic和static的讨论,也在一定程度上体现了神经网络设计的进步方向:减少超参数,降低调参难度,在真正意义上让网络自己学,多给模型以自主决定权。


5.2.6.5.6. IOU loss family

在之前训练bbox regression分支的时候,我们只利用了回归得到的角点与GT角点坐标差的模值即L1 loss(若是利用中心点和长宽也类似,用MAE、RMSE、Smooth L1等loss function),仔细思考一下就会发现这存在一些问题:实际评价检测效果的时候使用的指标是IOU,但显然我们loss的设计目标和评价指标不等价,多个检测框可能有相同的Loss但IOU的差异可以特别大,为了解决这个问题IOU loss就自然而然诞生了。

截取自论文原文,这里对比了L2 loss 和IoU loss

IOU loss用IOU来评价bbox回归的好坏,这样就统一了评价指标和训练指标,同时还让bbox regression拥有了尺度不变性(原来的距离loss对于不同大小的样本不具有这个特性!)。

欸那有人肯定又会说了,即使IOU相同,回归效果的好坏也会有差啊,不信你看这个:

以上三种情况虽然predict和GT的IOU相同,但是他们的回归效果肯定不同

而且在刚开始训练的时候,万一预测的框离GT很远,根本没有产生交集那IOU不就一直是零吗?那么有没有什么办法可以改进IOU loss让他可以区分相同IOU情况下回归效果的好坏并解决预测bbox和GT交集为零时如何训练的问题?GIOU给出了答案。首先找到一个能够包住GT和预测框的最外界矩形C。然后计算C的面积减去(GT ∪ bbox) 的面积的差与C的面积的比值,再用GT和bbox的IoU值减去这个比值得到GIoU。

\符号表示集合运算中的减法,定义为C(A∪B)定义为C中有而AB并集中没有的部分

略加思考你就会发现,GIOU loss会使得IOU尽量大,而C\(A∪B)尽量的小,那么在bbox和GT尽量对齐的时候,GIOU值会更大,这就在很大程度上解决了前一张图片中出现的无法区分的问题。下面这张图展示了不同情况下L2 损失、IOU loss和GIOU loss的大小:

对比下来,GIOU拥有最好的评价效果

既然IOU loss已经在这里开了个大坑,大伙就都在想能不能再这些个loss的基础上再挑挑刺,进一步细化评价标准?还真有,CIOU这不就来了,而且这帮作者还直接一下提了俩,除了CIOU还搞了个DIOU,还给不给人活路了,分我也发俩篇撒。

论文作者着重考虑了bbox回顾的三个要素:重叠区域,中心点距离和长宽比。从这里入手首先提出DIOU:在IOU loss的基础上添加了一个bbox和GT中心距离的惩罚项,以此为目标进行训练将会逐渐把两个框拉在一起。但是能保证他们对齐程度一定提高吗?CIOU紧随其后,又在DIOU的基础上额外增加了一个bbox的长宽比的约束,让loss能够更快地收敛。对比GIOU,GIOU的损失函数似乎也在做类似的事情,不过它把这两个工作给糅合在了一起而没有在两个方向上分别发力,颇有些”不偏科也没强项“的感觉。CIOU则把中心点距离和长宽比分开考虑,认为他们的作用是正交的,并对两个因素进行专门的优化,从而获得了更好的效果和更快的训练速度。

这就像线性代数中的特征值分解/正交分解,把原本糅合在一起的特征或属性在不同的维度/基底上分开考虑,最后再叠加到一起。这也是信号处理的惯用”伎俩“了(fourier、laplace等等)。

同时作者指出在NMS的时候使用DIOU可以获得比简单的IOU更好效果,论文还对GIOU的drawbacks进行了鞭尸分析,通过实验指出了IOU family前任们的局限性,感兴趣可以点击上方论文链接进一步了解。后续还有更多的IOU loss,考虑了更精细的情况,请自行搜索。


5.2.6.5.7. GFL

GFL的全称是generalized focal loss,看名字就知道,extend、generalize、complete啊这类的基本都是在原有论文的基础上得到启发进一步得到的结果(这说明多看论文就能发paper)。

观察IOU family的实现方法,我们发现IOU分支是和分类分支独立的两个分支,特别是使用解耦head的结构,训练的时候各玩各。而且前人的ablavation也已经证明定位框的准确性和分类准确度并没有很大的关联,因此在训练和预测的时候,两者输出的结果存在一个gap,一个分类置信度很低的bbox却很有可能在IOU分支中获得很高的分数,最后在计算得分的时候将两者相乘就会得到异常的结果。那GFL的解决方法就是,在训练的时候联合两者的表示(representation,定位质量表示框准不准,也就是前述iou family解决的问题,分类表示如one-hot或者softmax的输出,还有用分布来衡量的方法)。

这样,在训练的时候,把原来的one-hot label转换成分类与质量估计联合的连续label(即one-hot label乘上该预测框和GT的iou作为新的训练标记),将这个label作为分类和质量估计的指导,从而消除训练和推理时的gap(似乎显而易见!但是之前却没有人想到)。

因此对于FL的第一步修改,得到了如下的表达式:

σ就是添加了框质量预测的分类估计值,y是乘上了iou的标签

第二个问题是框的表示,在以往的传统框回归中,实际上是让网络学习一个狄拉克分布(冲激函数),即对一个以图像长宽为横纵坐标,框位置为输出的一个函数;也有一些方法尝试为框的回归简历一个高斯分布,即训练目标是得到以训练label为中心的一个尽量集中的高斯分布。GFL的想法则是不去人为指定或设计一个框的分布,而是让网络自己学习这个任意分布(颇有一种粒子滤波器的感觉)。

想要得到预测输出,根据概率分布函数的定义,就对整个区间进行积分:

因为图像是一个二维离散平面每一个坐标轴是离散的,所以我们把区间划分成n个小区间,将其离散化:
若要回归的更精确,就将区间间隔变得更小,反之增大,极限情况就是一个像素一个区间. 根据概率分布函数的特点,我们希望在积分区间内积分得到1,聪明的你肯定已经想到我们可以用softmax来输出整个离散分布!不过这里还存在一个问题,一个积分值可以对应无穷种分布函数,这就会大大降低学习的效率,使得框的分布不能很快地收敛(每个label都大概率会导向不同的分布,虽然他们的总体收敛方向相同,但是中间有很多个step都相互抵消了)。从直观上考虑,下图(3)是比(1)和(2)更好的分布,因此作者对损失函数稍加约束,提高离label最近的两个位置的权重使得分布能够更快的收敛,否则增加的权重可能被放在距离label处更远的地方,如下图(1)。

不同中分布积分得到相同值(横轴竖线)

最后得到的DFL(distance focal loss)就是如下的形式:

其中yi和yi+1是第i和i+1个位置(这两项看起来和cross entropy也有些相似,其实是在计算预测分布输出和标签的差异),y是标注的Dirac函数(即标注框的位置)Si和Si+1是分布函数在yi和yi+1处的值(也就是前文公式中的P(yi))。那么每次计算两个位置,若长宽都为100像素则需要计算99+99=198次。

不得不惊叹作者的loss function设计之巧妙,最后还将QFL和DFL进行统一,得到了一个GFL的表达式,有兴趣的同学可以自行参阅原文:Generalized Focal Loss 。作者在知乎上也有一篇讲解的文章,戳这里:大白话 Generalized Focal Loss - 知乎 (zhihu.com)

强烈推荐下载原文并且精读,它对于之前的预测框几种表示的洞察十分深刻,且这篇文章的数学方法并不算难而且对FL和IOU loss 提出的针对性的优化思路非常清晰明了,很适合入门后的进阶。随后作者还推出了GFLv2的版本,感兴趣自行搜索阅读。


5.2.6.5. 多尺度特征融合

稍作总结后,笔者在这里再介绍一些目标检测领域的trick和改进方法(其实现在已经不新了),其中的很多已经成为了现代检测网络的标准设计。前一部分主要是关于LF的设计,而这一部分主要是关于关于模型的改进的经典方法与训练技巧,还有一些有启发性的机制。

5.2.6.5.1. SPPNet

由于CNN基本构建模块的特点,在最后输出的feature map往往是浓缩了大量的语义信息(“高级的特征”)但分辨率较低、对细节的保留较少,因此对于小目标的检测效果不尽如人意,且模型对检测对象的尺度不变性大部分也只能来自于训练数据。

SPPNet提出的空间金字塔池化通过对最后一层feature map执行stride大小不同的池化后再进行concatenate,能在一定程度上解决不同尺度物体的检测问题。但是仅仅使用了高级特征(或者说特征其实还是来自同一层),对于不同尺度的融合还是不太好。

spp的结构,用不同大小的grid分割后进行max pooling操作,最后concatenate并投入下一层
5.2.6.5.2. FPN

SSD首先提出在Backbone的不同层输出进行检测,就是希望能同时利用不同层次的信息。但是对于多个feature map的检测也导致了效率的低下,并且低层feature map能够提供的信息太少而直接检测的计算量又大(分辨率高),虽然对于不同尺度的物体检测能力略有提升,但是速度相对YOLO下降的比较多。

几种检测的方法

(a).传统差分图像金字塔,抽值实现降采样

(b).深度学习经典做法,使用顶层feature map检测

(c).SSD的做法,利用深度网络提取feature并在每一层map上检测

(d).FPN做法,top-down融合并在不同的map上检测

那么很自然的我们就会想到,有没有什么方法可以综合底层的细节和高层语义信息,随后利用融合后的信息进行检测,将各种尺度的目标一网打尽?于是便有人提出了特征金字塔网络Feature Pyramid Netwok。和以往的图像金子塔不同,FPN的金字塔由深度网络提取的不同层级特征堆叠而成,并在堆叠的基础上通过top-down的element-wise add实现特征的融合,从而实现对不同尺度物体的检测。

FPN金字塔中的下采样方式,最顶层的特征会经过x2的上采样插值后和底层feature map相加

为了更好的融合特征,也可以把简单的相加替换成一组卷积。

5.2.6.5.3. PAN

注意到FPN只有top-down的融合路径,金字塔顶层无法享受到底层的更多细节,那直接添加down-top的融合不就完事了?PANet的作者直呼有手就行,提出了这个路径聚合网络用于更充分的特征融合(PANet首先被用在语义分割上,分割其实相当于像素级的分类,因此对于细节的要求更加严苛和敏感,特别是边缘和纹理)。

PANet的整体结构如下,红色虚线表示原始图像经过数十甚至近百层的卷积(即原图经过backbone到最后顶部的feature map)得到的P5已经没有什么细节信息了。那么它相比FPN的不同就是(b)部分的down-top连接将底层细节通过较短的路径融合到上层,即绿色虚线所示的路径(P2->N2->N3->N4->N5)。而这里的Top-down连接通过3x3,stride=1的卷积后再相加来实现而不是简单的element-wise add,让网络自己去学习如何进行特征融合(当然如果为了追求速度将卷积换成和FPN中一样的element-wise add也是可以的,或多轻量化网络都是这么做的)。通过绿色虚线路径的特征大概只需要十层不到的conv就能达到N5处,这样的两个部分(top-down和down-top)就很好地解决了利用顶层feature map检测时缺少细节和纹理的问题。

PANet的结构
PANet中down-top连接的结构
5.2.6.5.4. Bi-FPN

不过人们对于极致的追求是永不止步的,在进行了大量实验后,EfficientNet将Neck(一般把检测框架中的特征融合/选择部分称作neck)进行了进一步优化,提出了Bi-FPN(双向特征金字塔)。

左侧白色单元backbone中不同层次的feature map输出,虚线框内是BiFPN的结构;右侧是PAN的结构

Bi-FPN的总体结构和PAN相同,都是top-down和down-top的排列,我们主要关注他们的不同之处。首先是feature map跨过top-down下采样融合部分直接到bottom-up部分的远跳连接,这和ResNet中的结构几乎是一致的, 作用也是相同的,可以分散梯度并防止退化。另一个改变是删除了PANet中红色框的单元直接连接到下一层,作者认为这种单边的连接没有必要添加额外的卷积,因为没有融合的对象,如果进行卷积反而会引入不同层次的特征并增加梯度回传的难度。若有必要,可以通过堆叠Bi-FPN的block实现更多的融合并方便地控制网络规模和深度(g家惯用套路了,比如inception)。

还有其他的思路如把接下来要介绍的注意力机制引入特征金字塔中,为不同的连接路径设置不同的可学习权重、通过GAN由高层语义信息还原底层细节(类似超分辨率)等,感兴趣的同学可以自己探索。


5.2.6.6. CV中的注意力机制

注意力机制最早被使用在机器翻译(或自然语言处理)上中的Encoder-Decoder模型上,让网络在处理不同部分语句时能够聚焦到特定的已经编码的语义向量或输入上。

若想要理解上述注意力机制的来源,请学习和NLP和RNN相关的知识:吴恩达深度学习-序列模型,这是因为在计算机视觉中的注意力和机器翻译可以从不同角度理解。尤其是自注意力机制,强烈推荐李宏毅老师的讲解:李宏毅机器学习课程-self attention

在阅读空间域的QKV模型时,务必先掌握自注意力模型。

而在计算机视觉中,我们不大需要用Query/Key/Value的模型来解释,笔者更倾向于使用特征分解这个词,从特征空间的角度解析cv中的注意力机制。用于计算机视觉的注意力机制一般分为三种:通道域、空间域和混合域。其原理是赋予空间中不同通道或区域以不同的权重,而不是像以往一样进行卷积/池化操作时将空间中的一个位置(空间中的一个位置,对应所有CxWxH中长度为C的狭长的一列)或所有通道视为拥有一样的权重,从而使得网络专注与更重要的信息的提取。

这里会有一个误区,很多同学会认为,网络在训练的时候,只要往正确的方向迭代,不就自然会赋予卷积核中的参数以不同的值,这样难道不会在通道域或空间域产生不同的权重吗?为什么还需要注意力机制?

这里需要特别注意,卷积核上的不同权重参数是定域的,即其只能在其感受野内形成”小注意力“,虽然随着深度的加深感受野逐渐变大,但是其对于浅层和中层的参数更新也是间接的、微弱的。而注意力机制则会关注所有通道/整个WxH的空间的信息,从而选择最重要的部分。因此,也有”卷积是一种特殊的注意力模块“的说法,这么看来也不无道理。

视觉注意力一般分为三种,即通道域、空间域和混合域(前两者都使用)。在分割中还有使用类别注意力(分类特征对于图像分割额外重要),视频分析中也有使用时间注意力来建模目标在时间序列中的关系。

5.2.6.6.1. 通道域注意力

通道域注意力的代表作是SENet,其实现思想和过程都非常简单,仅仅加入了一个即插即用的SE模块分支用于建模不同channel之间的相关性就大幅度地提升了模型性能(最后一届ImageNet图像识别的冠军,超过去年冠军25%)。

SENet的结构
  • SE module通过将上图中的c2xhxw的block利用全局平均池化操作Fsq(取每一个channel的平均值)得到一个1x1xc2的相关向量,再将其投入一个两层的全连接层Fex后得到一个注意力得分,随后让此得分向量和原feature maps进行简单的标量乘法操作,就得到输出。注意力得分即代表了每个通道的重要性(权重)。由于SE modeul的轻量化设计使得其在几乎没有增加什么运算量的情况下大大提升了网络的性能。

若没有学习过积分变换或信号与系统,可以跳过下面这一段。

从本节开始谈到的特征空间分解的角度来看通道域的注意力机制,我们可以类比积分变换的思想:多通道卷积相当于把多个二维卷积核堆叠在一起,将他们拆分来看,每一个二维卷积核其实代表了一种特征空间或像素空间的基,可以对应傅里叶变换中不同频率的正弦分量(如下图),只不过这些分量的选取不是固定的,同时大概率不是正交的,也几乎不可能是完备的(学过信号与系统的同学可以思考一下这个问题)。那么对于检测一个物体来说,这些在不同特征分量上的投影(就是前级feature map经卷积操作产生的不同通道),其重要性也肯定不同。举例来说,如果我们要识别小猫,那么它身上的花纹肯定是一种重要的特征,对应傅里叶变换则应该是一种/多种频率较高的分量(花纹周期性出现),而低频分量就不是我们所关心的。同样对应通道域注意力,虽然我们不知道网络到底学习到了哪些特征(卷积核最终会代表哪一种特征分量),但是不同分量的重要性肯定不是等同的。如SENet这种通道域的注意力,就是让不同通道有不同权重,从而聚焦到那些对于识别特定物体更关键的通道上。

傅里叶变换将图像变为不同权重的不同分量叠加

其实在SENet之前,就有paper认为图像中不同频率分量对检测的重要性不同,因此直接对原图进行傅里叶变换,然后将每个分量分别投入不同的网络,最后再用add或multi的方式聚合特征进行检测。其检测效果也还不错,说不定SENet就是借鉴了这类传统方法。

其实已经在可解释性方面的工作尝试发现频域和CNN网络之间的关系,传统图像处理仍然在发挥其优势,想进一步了解相关工作的同学可以参考这篇文章:频域(DCT,小波变换)与CNN结合,这需要一些信号处理的数学知识。

  • 由于SENet仅仅利用简单的global average pooling和两层perceptron建模了不同通道的重要性,对于容量大的网络肯定是不够的。因此,后续作者又对其进行了改进得到SKNet(Selective Kernel,如下图),字如其名,这是一种能够自己选择合适大小的卷积核的网络。作者通过实验发现卷积核的结构和待检测物体的尺度有一定的关联,因此希望利用SK module让网络建模这种关联。

有人会说啊,google的Inception不是已经做到了让网络自适应选择卷积核了嘛,我们要注意的是虽然Inception同时并行使用了1x1,3x3和5x5的卷积,但是在concatenate之后,这些通道之间的重要性是等同的,而且卷积核的参数在训练结束后也是静态的。

sknet结构

上图中U~和U^分别是经过不同大小卷积核卷积后得到的feature maps,直接按元素相加得到U,随后进行global average pooling和fc(还是SENet的味道),再投入softmax得到注意力得分a、b分别和U~,U^对应通道相乘,选出在对应尺寸卷积核下最重要的通道,最后将上述结果进行按元素相加得到输出V。


5.2.6.6.2. 空间域+混合域

空间域的注意力相对比较直观,人类的视觉系统就有这种聚焦与视野内某个区域的特点,帮助我们快速捕捉最重要的信息。在下图里,我们肯定不会去观察海报的白色部分,因为那里没有任何有意义的信息。

阅读海报时人类感知系统的注意力分布

经过这么长时间的熏陶,相信你应该已经想到如何构建一个空间注意力模块了:先利用池化或卷积从前层的feature maps中提取一个用于生成注意力分数的相关向量,紧接着将此它投入一个轻量化的fc层或conv层,得到注意力分数后和原feature maps相乘即可。不过需要注意的是,空间注意力需要为大小为WxH中的每个点都分配一个权重,因此生成的是一个注意力分数的矩阵。

很好!我们已经有了建空间中不同点间相关性的方法,这其实也是CBAM(Convolutional block attention module)的做法。CBAM是SENet的改进,在原有的通道注意力基础上增加了空间注意力模块,它其实是一种混合域的做法,只不过我们在这里介绍它的SAM模块以引入空间域。

绿色的是通道注意力模块,实现和SENet基本相同
SAM结构,同时使用最大池化和平均池化提取用于注意力的feature map
5.2.6.6.3. Transformer-liker

而空间域注意力还有另一种结构,是继承自NLP中的QKV自注意力模型的:利用原feature maps分别生成Query、Key、Value三个矩阵,然后Q*K得到不同像素的相关性,最后把相关性矩阵和V相乘得到输出。由于直接计算了每个像素之间的关联,所以一般认为其对空间中不同部分特征相关性的建模能力更强。None-local NN是第一个将这种机制引入cv的,在此领域的代表作还有DA-Net,相当于是CBAM的自注意力版,有兴趣的同学可以阅读一下原文:Dual Attention Network for Scene Segmentation.

自注意力的Spatial Attention的结构,其中N=WxH
DA-Net的结构,通过并行的SA和CA模块建模相关性
5.6.6.4. 降低注意力时间开销

但是它们都存在一个比较大的问题:巨大的时间开销。源自计算像素之间相关性的矩阵乘法(Q*K)时间复杂度至少是O(W^2*H^2),这对于实时性是不可接受的。于是就出现了ANN、CCNet和GCnet。

  • ANN(Asymmetric Non-local Neural Network)的思想很简单,一个局部区域内的像素或提取出来的特征和另外一个区域的相关性,应该是比较相近的,除了非常精细的边缘,总不可能每隔一个像素相关性就发生巨大的变化吧?并且检测比分割对于边缘的要求更低。因此它对Key和value矩阵进行了下采样的操作,实际上采用的就是局部线性化的思想:在feature map上选取稀疏的采样点(锚点),利用这些点替代周围的像素。
ANN的思想:Key和Value分别通过采样缩小尺寸,使得时间复杂度获得较为可观的减小

线性化的具体的操作就是利用SPP生成θP和γP(最大池化和平均池化均可,或两者都使用让网络自己学习合适的权重),SPP产生的不同层次特征还有助于多尺度分割和检测。根据上图的结构,作者提出了AFNB和APNB两个模块。APNB共享了Key和Value的参数(类似检测头的参数共享),尽可能在不降低建模能力的情况下减少参数量。而AFNB则是继承了APNB的思想,我们从上图中可以看出θP和γP只要分别保证一个维度为C,S的大小是可以根据锚点的数量而变化的,只需要保持对齐即可(取点多则S大,反之少)。既然这样,用于生成QKV的输入也不一定要从同一层feature map中产生,即下图中AFNB模块中的Xh和Xl分别来自不同backbone中的不同层,只需要让生成的θP和γP拥有和Query匹配的维度即可。作者通过实验验证这种做法可以更好地融合不同尺度的特征,因此让backbone的后两个stage分别作为AFNB的输入。

ANN的架构

从另一个角度理解,也可以将APNB视作Xh和Xl都来自同一个stage的并共享了Key和Value参数的例子。

  • 虽然ANN已经在极大地降低了相关矩阵和注意力分布计算的开销,但是毕竟还是多项式级别的复杂度,有没有什么办法可以进一步减小注意力模块的规模?GCNet(global context)的作者通过实验发现,在backbone的后几个stage,将其feature map上任意一个位置的像素作为query,生成的注意力分布图非常相似,也就是说这种non-local的注意力模块学习到了位置无关的注意力分布。下图是对注意力的可视化结果,蓝->黄->红表示“聚焦程度”。
作者从coco上随机选取了6张图片,发现Non-Local模块对于不同的query point生成了几乎相同的注意力分布

这里需要特别注意的是,某点注意力分数越高表示该点和query的相关程度越高。这也就是说,图像上的任意不同点,对于另外一点计算相关性,他们得到的结果是大体相同的(听起来似乎有些匪夷所思!不过注意这是在backbone的后几个stage,由于卷积的感受野越来越大,feature map上不同点之间的关联也应该随着网络深度的加深而变大)。作者在原文还计算了经过注意力权重加和后的特征向量之间的各种度量距离,发现他们的差距都很小。

那么结果很明显了,原来要计算WxH个注意力分布,现在我们从输入上随机选一个query point(或者说不需要query了)来计算它和其他像素的相关性,然后把这个注意力分布应用到其他所有位置即可:

(b)为简化版本,直接丢弃了Wq矩阵,从计算上来看相当于不关心到底选取了哪一个位置作为query

这时候输出可以写为:

其中zi为第i个位置的输出,Np=HxW,xi为第i个位置的值,xj为其他位置的值,Σ后的是softmax操作,为了让总得分和为1。如果把Wv提取到求和前面(利用交换律),复杂度可以继续降低:

(b)为简化版本,(d)又加入了通道注意力

对边之前的版本,Wv的1x1卷积复杂度从原来的O(C^2HW)变成了现在的O(C^2)。作者又在上图中(b)的基础上加入通道注意力(使用降维的1x1卷积 [ bottle neck,熟悉inception的同学应该知道这个操作,用于降低参数量 ] 并使用layer normal,ization进一步简化原本的SE模块),得到了最终的GC block,如上图(d)所示。因此GCnet也属于混合域的注意力模型。

关于位置无关的注意力分布和模型简化、加入通道注意力与ablation还有更多的细节,若感兴趣可以参阅原文:GCnet

  • 最后一个有趣的很有创意的空间注意力网络是CCnet(Criss-Cross Attention [criss : 纵横交错的] )。CCnet聚焦于相关矩阵计算时的高复杂度,提出的模块结构如下:
一个交叉注意力block需要使用两次基本的交叉注意力module

criss-cross module在计算不同位置相关性的时候只考虑和当前query处于同一行和同一列的像素,单次计算的复杂度变为(H+W)*HW,紧接着还是和V矩阵进行element-wise multiply得到输出。随后再一次进入另一个cc module得到最终结果。

criss-cross module的结构,总体和non-local自注意力相同

聪明的你应该已经想到,使用两次计算就可以间接接触到全局的信息了!第一次相关性计算会将当前像素的信息传递到同行同列的所有位置,第二次计算时之前接受到信息的每个像素又会将再次将信息传递到自己的同行同列。更为形象的例子就是,象棋中的“车”在经过两次移动可以到达棋盘上的任意位置(在没有遮挡的情况下)。

不在同一行的两个像素经过两次注意力计算产生关联

5.2.6.7. Anchor-Free模型崛起

我们在学习Faster R-CNN的时候第一次遇到anchor的概念,它利用anchor匹配正负样本,从而缩小搜索空间,更准确、简单地进行梯度回传,训练网络。但是anchor也会对网络的性能带来影响,如巡训练匹配时较高的开销、有许多超参数需要人为尝试调节等。而anchor-free模型则摒弃或是绕开了锚的概念,用更加精简的方式来确定正负样本,同时达到甚至超越了两阶段anchor-based的模型精度,并拥有更快的速度。这也让one-stage和two-stage的边缘更加模糊。接下来就让我们介绍几个非常经典的anchor-free模型:

5.2.6.7.1. FCOS

FCOS的全称是fully convulsion one-stage object detector。全卷积网络常用在目标分割任务,最早由FCN引入检测领域。FCOS匹配正负样本的方法别具一格,它直接将backbone输出的feature map上的每一个像素当作预测起点(当然输出的feature map和原图有一个映射的对应关系,本质上并没有将图像上的每一个点当成参考点,而是类似yolo,散布了一组格点),即把每一个位置都当作训练样本,只要该位置落入某一个GT框,就将其当作正样本进行训练。同时为了让一个目标不在多个feature map上被重复输出(推理时),FCOS人为限制了每一层回归目标的尺度大小,超过该限制的目标,这一层就不检测。

FCOS的结构

为了删去低质量的预测框,FCOS还在检测头增加另一个中心度分支,保证回归框的中心和GT较为接近。同时和FPN结合,在每一层上只回归特定大小的目标,从而将不同尺度的目标分配到对应层级上。

centerness的计算公式,设置中心度阈值从而保证生成的预测框中心接近GT

可以发现,虽然不再显式的使用anchor对正负样本进行匹配,但是仍然加入了一些预定规则来给参考点分配标签,之前的anchor是anchor box,在某种程度上我们可以将FCOS中的回归位置看作是anchor point,或者步长和边长相等的一组正方形anchor box。


5.2.6.7.2. CenterNet

CenterNet比FCOS更近一步,其论文的名字就叫做Object as points:将目标看作点。它直接在BackBone后接Conv层,输出一个代表物体中心点的Heatmap,在预测时取该Heatmp的topk作为可能的物体中心,再进入接下来的回归和分类分支,回归分支输出的是预测框相对于中心的offsets(width、height两个值或left、right、bottom、top四个值)。

CenterNet的预测示意图

其匹配正负样本的方法自然和预测一样,取GT的中点作为正样本。不过仅仅去中点一个点那是在是太少了,网络可能都不会收敛或是被负样本的梯度淹没。因此以GT中点为Gaussian中心,生成一个“正样本分布”,在此分布范围内的points都会得到正样本的监督信号,只不过离中点越近,one-hot label的值越大(就是将GT的标签乘上一个gaussian权重,也可以看作距离衰减因子)。

来自论文原文,GT中点附近会有一个衰减的惩罚,从而间接地增加正样本数量

虽然CenterNet的论文提到这是一个真正无anchor的网络,然而,我们鸡蛋里挑骨头看看,生成的中心点heatmap在位置上其实和原图也有对应关系,和FCOS一样是原图上的密集网格(网格密度由center heatmap的大小决定)。


5.2.6.7.3. CornerNet

CornerNet和CenterNet有异曲同工之妙,两兄弟的论文题目也稍稍互动了一下,这篇叫:Objects as paired points。

CornerNet的pipline,通过预测top-left和bottom-right角点得到检测框

CornerNet通过预测物体的左上和右下角点的heatmap得到预测框。这里其实有两种方式,一种是只预测检测框,不管里面是社么物体,随后再在每一个检测框上运行分类网络;另一种则是直接为每个分类预测左上和右下两组heatmap,直接得到类别和定位,这也是CornerNet的做法(说不定你用第一种方法试试,也能发篇paper)

CornerNet的架构

CornerNet使用Hourglass作为backbone(是一个常用于姿态估计和关键点检测的backbone),在其后接上两组角点的预测分支,其中每一个分支内又分为三个分支,分别是角点heatmap、角点embedding和角点offset分支。

前述cornerNet为每一个类别预测两组heatmap,因此总共需要预测c组热图,开销还是比较大的。论文还提出了一种用于聚合特征的CornerPooling:

Corner Pooling的示意图

corner pooling将会对topleft角点的同一列的下方和同一行的右侧进行channel-wise的max pooling操作,对于bottom-right角点则是同列上方和同行左侧。从直觉上理解,一个物体的外接矩形必然在边框上和物体有相接的地方,从这些地方提取出的特征就能被Corner Pooling很好地捕捉到:

字如其名,角点embedding将计算每个角点的嵌入,以得到最小距离为目标进行训练就能匹配成对的左上和右下角点。

offsets则输出角点的偏移量,用于获取更精确的定位(heamap的分辨率没有原图高,因此需要offset分支来修正角点位置,可以看作一种插值手段)。


5.2.6.7.4. ExtremeNet

ExtremeNet师从前两个检测网络,同样是通过预测特征极值点的方式来检测物体。通过预测五组极值点的heatmap并使用center grouping的方式将这些极值点分组从而形成检测框。下面这张图阐释了ExtremeNet的检测思路:

也可以称这些极值点为极大的边缘响应

ExtremeNet匹配四个极值点一组的方法是暴力搜索(复杂度为O(n^4),原文提到有方法可以将复杂度降低到O(n^2),其中n为生成的heatmap的WxH),取每个heatmap的topk,四个一组进行遍历组合,若这一组极值点生成的检测框的中心附近恰好对应着预测的center heatmap上的一个极值点,就把他们作为一个检测目标输出。

整个网络的架构如下图所示,同样使用了特征点检测常用的backbone hourglass:

在骨干网路后紧跟着9个分支,分别检测上下左右和他们对应的修正偏移量,以及物体中心的heatmap。可想而知,这个检测网络的速度不会太快。由于物体的外界矩形边界上可能存在多个极值点,或者这一条边恰好和物体的一条边重合,那极值点就可能会出现连成一条线的情况,因此论文中提出了一种Edge aggregation的方法。其实和soft-NMS差不多(NMS则是直接去除分数不是最高的检测框),又和CornerNet的corner pooling有相似之处,它会对每一行列都进行aggregation,若一个极值点附近有其他极值点,直接降低其附近极值点的score。当然,你也可以在检测框生成以后进行NMS,两者的开销比较如何,需要自己计算一下。

和前面两个网络一样,训练的时候用L1 smooth来作为损失函数计算预测框的损失即可。

经过一些修改,这个网路可以比较轻松地扩展到分割任务上,方法就是多预测几个极值点,然后用这些点连接成地多边形来近似一个分割mask,以下不同颜色的多边形是用于分割的结果,白色半透明的bbox是外接检测框:

那么这回,你觉得ExtremeNet是否是真正的anchor-free?似乎它也只是换了一种方式回归bbox,却巧妙地避开了anchor匹配的问题,其实换一种思路,生成四个角点的heatmap然后匹配bbox,这不就是最常见的anchor-based方法嘛。


5.2.6.7.5. FSAF

FSAF(feature select anchor-free )则是的思想也非常的巧妙,可以看作是一个anchor-free和anchor-based网络的结合,其整体架构如下:

图中省略了backbone,原文选用的是ResNet-50

之前的FPN的每一层输出都会接到一个检测头上,常规的检测头都有两个分支:分类和回归。传统的网络为GT选择特征层对进行回归的方法是根据GT box的大小,大目标会被分配到高层,小目标则分配到底层,“大”和“小”的定义都是需要人为设定的超参数,人为设计的规则一定是最好的吗?要怎么样分配最合适的特征层去检测对应的目标?

基于这个问题,在训练时,FSAF让FPN中输出的不同特征层上的anchor-free分支先进行检测,然后输出得到loss,再使用loss最小的那个特征层的anchor-based分支进行回归和分类。这样就做到了自适应的尺度大小分配。


5.2.6.7.6. ATSS

这个方法的全称是Adaptive training sample selection,基本可以说是一统anchor江湖的天下,人家直接通过严密的消融实验告诉你,anchor不anchor无所谓,关键是怎么分配正负样本并采用最好的特征图检测目标!前者也就是后来所说的标签分配问题。

原文对比了两个one-stage检测器,分别是上面介绍的FCOS(anchor-free)和RetinaNet(anchor-based,focal loss),通过控制变量法在MS COCO上验证,得出了不论从从reference point(FCOS)还是anchor box(RetinaNet)开始检测和回归,产生的差异都不大的结论。

RetinaNet在把FCOS用到的buff都叠完了以后,就只有0.8AP的差距了

他们的差异只在于如何定义正负样本:FCOS把落入GT内的参考点都当作正样本并限制不同层回归最大目标的尺度,而RetinaNet还是采用的IOU方法:

两者确定正样本的方法的图示

对于(a),FPN2上布置的一个anchor和GT有最大的IOU因此选择pyramid level2上中心位置的anchor作为正样本;对于(b),每一层上都有数个点(grid)落入GT中,但是FCOS对每一层检测的目标尺度大小有限制,level 1超出了尺寸限制故最终选择Pyramid level 2 来学习此GT。

RetinaNet从anchor box开始回归,预测值是四个offsets;FCOS从点开始回归,预测的是点到四边的距离

因此,作者认为定义更合适的正负样本,如何让对应的拥有最好特征的特征层后接的检测头进行其检测才是问题所在。

前述的FSAF在一定程度上解决了不同尺度目标检测的问题,但是同层的不同的anchor box的长宽比和大小、数量全部是需要设置的超参数,即使用聚类、差分进化或dynamic anchor,也只是尽量取平均值,不能达到最好的效果。

在经过了前面一大段的铺垫后,作者提出自适应的样本选择方法ATSS,对于每个GT,在FPN的每一层上选择距离GT中心距离最小topk个anchor boxes(有几层就有几*k个,而且前面的实验已经得出了anchor不anchor无所谓作者为了方便说明就沿用了anchor的概念)。计算这些anchors与GT的IOU均值 μ 和 标准差 σ,然后选定那些和GT的IOU大于 μ+σ 的anchor(高子里面再拔高子!)作为正样本(同时保证这些点是在GT框内的),剩下的都是负样本。如果一个anchor在上述计算下被划入多个GT,就选择IOU最高的那个。

通过这种方法,就不用设置各种超参数,唯一的参数就是topk的k。经过实验确定,在大于7的情况下也是鲁棒的(用大数定律就能很好地解释,太小的值不能反映其分布特性)。对于不同的anchor大小(其实也就是分割网格的大小),经过实验验证也是稳定的,而且发现即使使用多组不同形状和大小的anchor对于回归和分类也没有帮助。

都说这篇文章的ablation做的特别扎实,行文流畅条理清晰,颇有终结anchor之争的意味。强烈推荐阅读: Bridging the Gap Between Anchor-based and Anchor-free Detection via Adaptive Training Sample Selection (arxiv.org)


5.2.6.7.7. 到底应该怎么确定正负样本

首先,我们在这里区别一下标签分配和样本不均衡问题。Focal Loss、GHM、OHEM都是从前向传播产生的Loss入手,让网络关注那些Loss很大的样本(有可能是正样本也有可能是负样本),loss越大通常该样本包含的信息就越丰富(loss小的可以看作网络已经学得很好了,预测结果八九不离十);而标签分配要解决的问题则是谁应该被分配到正/负样本,这里的谁指的是anchor(anchor-based的模型)或者某个位置/参考点(anchor-free的模型)。

在标签分配出现之前很长的一段时间,标签分配都是通过IOU计算进行的(anchor-based方法),哪个anchor和GT的IOU大,就把他当正样本,而对于anchor-free的方法如RetinaNet,GT bbox的中心落在哪个格点内,就拿那个格点当正样本,FCOS则认为只要中心点落入GT box内就是正样本。因此样本不均衡问题是在正负样本已经分配完之后发生的,标签分配决定谁应该称为正负样本。

笔者认为,可能是研究者发现既然可以利用Loss给不同的样本动态地分配权重,那么能不能设计动态的标签分配方法呢?在FASF之后开始有大量的论文思考标签分配的问题,当时做的最好的也只是ATSS的基于单个batch统计分布进行分配,还是颇有一种“死板”的感觉。旷视提出OTA之后,给标签分配指明了方向:先计算正负样本分配后的损失(为了和loss区别,这里的匹配损失一般称为cost)再根据cost判断哪些anchor或者格点的预测应该被认为是正样本。前面之所以说OHEM是OTA的始祖,正是标签分配继承了“先算损失再选择”的思路。后续在DETR(下面Transformer即将介绍)中提出的一对一匹配,进一步把标签分配问题推向高潮。

对比样本不均衡,我们可以发现标签分配是为了在所有预测中找出最好的哪些,并把GT分配给它们,使得它们的预测变得更好;同时让那些不适合预测这个GT的anchor别来掺和,管好自己的事。样本不均衡则是为了防止网络只会预测背景类或简单类,避免出现“欺软怕硬”的情况。

在6.1.3.4我们会更详细地讨论这个问题。


5.2.6.8. ViT(Vision Transformer)

Transformer是NLP领域的基础架构,也即前述的自注意力机制加上Encoder-Decorder的结构,这相当于CV领域的CNN架构。这里我们不会从最原始的ViT开始,而是直接介绍FAIR退出的DERT模型和其改进版的可形变注意力版本,因为这是在公开数据集上基于transformer的模型第一次拥有了获得超过CNN-based的模型的表现。

不过由于自注意力模块的架构不同于CNN,这导致基本无法使用现有的硬件(如google的npu和nvidia的nn-core与部分cuda-core)和软件(针对3x3conv优化的winograd加速算法)对其性能进行优化。而且拥有相同精度的transformer模型的参数量和计算量往往远超CNN,因此目前对于我们而言,很难将其部署在边缘平台上,在RoboMaster中也没有太大的舞台,故这里对ViT只是进行尽量完整的介绍以便读者能够了解学术界的前沿。

请注意,学习ViT需要一些auto-encoder和自注意力模块的基本知识,这在之前介绍vision attention机制的时候已经提到这一点,有了前述基础,相信你理解ViT也并不会遇到太多的困难,我们可以将vision transformer看作将所有的CNN都替换为自注意力,卷积是一种特殊的自注意力操作(局部),而自注意力又可以看成广义的卷积(kernel和padding都等于图像大小的卷积,并且使用循环padding,即将图像复制并平铺在原图的四周)。

笔者的另一篇文章详细介绍了transformer模型(也就是增强的自注意力):,还有一篇介绍auto Encoder-Decoder模型的总览性的文章:

  • DETR(DEtection TRsformer)

DETR是facebook AI research lab(现在已经叫MetaAI了)发布的一个端到端的检测器,它将目标检测作为一个集合预测的问题,因为它限定了transformer的输出个数和Decoder的输入。端到端的设计使DETR也实现了真正的nms-free,它只会输出限定数量的预测框和其对应的类别。

DETR的检测pipeline

DETR没有太过极端(同时也是为了减小运算量),它仍然采用CNN作为backbone提取特征,之后将其输出序列化并加上位置编码,投入标准的transformer encoder-decoder架构中,最后输出100个prediction box和class(固定100个输出,若途中没有那么多目标就输出no object,即空集)。在CNN之后的结构和最早的ViT并没有什么不同,关键是DETR优化了损失函数和正样本匹配方法,将正样本和预测框进行全局二分匹配以找到最优的标签分配,从而实现上述的端到端方法。

我们直接顺着pipeline,按照结构顺序来分析DETR:

DETR的结构和数据流
  1. 首先是backbone部分,文中提到采用了传统的卷网络但也没有说具体是用的哪一个网络,具体可以看看它的代码。同时我们需要为生成的feature map添加位置编码,因为自注意力计算是位置无关的,导致无法充分利用位置关系,故为每个位置添加位置编码使得transfomer捕捉到这种信息。关于位置编码原文也给出了参考文献和使用不同位置编码得到效果的ablation,将其看作为每个位置的feature map附上其坐标信息即可(可以是element-wise add、scalar mutiply或者concatenate)。

加上了位置编码之后我们就可以把图片拆分成一个个小块(把三维的dxWxH的feature map变成一维的即序列化,因为feature map上的一个像素对应着原图的一块区域,所以这里说是原图上的一个块,一般称为patch),投入transformer中。

  1. 这里使用的transformer结构如下图所示,Encoder中堆叠数个多头自注意力、前向网络和跳连接+归一化组成的基本结构,然后选择一些层的输出作为Query和Key添加到Decoder的输入中(这些mutil-head attention的输入并没有特定的要求,可以是任意打乱顺序的层,类似于FPN中的多尺度特征连接,融合不同尺度的特征)。
DETR中的transformer结构

Decoder不同于Encoder的就是上述的Mutil-Head Attention结构,不仅会用到前层输入还会使用Encoder的中间层或末层输出。同时最重要的是Decoder的第一层输入是一个可学习的Obejct Queries,他和FFN、自注意力中的QKV权重一样会随着网络的训练而改变。它在Transformer架构中的地位,相当于CNN架构中的anchor,并且这是一个动态的、可变的自适应anchor,只不过换了一种形式出现而已。经过学习后,object queries中的每一个单元都会对应到一些特定的目标,论文也给出了预测框中的top 20(限定transformer的decoder输出100个检测结果)的可视化图表。其在MS COCO上检测的结果显示,每一个slot(可以看成是一个object querie的单元,或是预测集合中的一个元素)都会对应到一些特定大小和比例的目标,这就和anchor非常非常类似了。

预测可视化结果,每个结果由一个点表示一个检测框。其中绿色代表小目标,红色代表横向的大目标,蓝色代表纵向大目标
  1. 最后就是head部分,几个FFN(feedforward network,其实就是全连接)后输出预测框位置和分类。这里主要介绍一下训练时的标签匹配方法。这部分匹配和anchor-based的方法相似,要将GT分配给最合适的anchor进行训练,但是由于使用的时end-to-end的策略,在标签分配时自然只能将预测结果和GT进行一对一匹配,而DETR设置的100个框显然远远多于一张图片中含有的GT个数(除了个别小样本数据集,如果要扩展到这些数据集上,修改框个数即可),因此没有匹配到GT的预测会被设为空,不对应任何目标。
  • Deformable DETR

和其他的ViT一样,DETR最大的问题就是transformer模块计算的巨大开销,自注意力(全局+密集)计算的复杂度是,更何况是多头自注意力。我想作者的灵感应该来自deformable conv和pooling,既然卷积可以形变并且可以看作一个局部的注意力模块,那为什么不能让自注意力稍稍不那么“全局”,并且可以自适应的选择需要需要特别注意的那些像素呢?并且之前我们在介绍Vision Attention的时候,GCNet就提出高层次的特征生成的注意力几乎是位置无关的,所以我们并不需要计算所有像素两两的相关性。于是Deformable DETR就自然而然地诞生了。

DDETR的架构

DDETR的整体和思想架构和DETR的区别不算太大,最关键的就是将自注意力替换成了可形变注意力,因此同时需要学习一组用于计算注意力偏移量的参数。可形变注意力不再计算query和其他所有点的相关性而是只计算部分点,那么怎样知道是哪些部分?这就交给offsets进行学习,经过训练offsets能得到对当前query最重要的那些位置。

Deformable Attention模块

feature map通过简单的线性变换生成Value,Query feature同样经过线性变换生成数个头的注意力偏移量和每个位置注意力的权重,最后将那些位置的feature采集起来并乘以权重得到最后的输出。可以看到上图中Reference Point 在Input Feathre Map上的位置已经用相同的粉色标记了出来。

通过替换全局的自注意力,时间复杂度下降到,其中k为偏移量的个数,也就是用于和query point计算相关性的点的个数。此时,虽然自注意力是”局部“的,但是deformable attention仍然可以通过学习大的offsets从而计算两个相离很远的像素的相关性,这里的”局部“应该作”部分“或者”稀疏“来理解,而不是像卷积一样定域的。

由于encoder不再是transfomer一样处理序列化的数据的结构,这里的编码器反倒和Vision Attention的计算非常相似,颇有折衷的感觉,这也说明CNN和Transformer架构处在不断地相互学习和融合的过程中。从他的架构上我们还可以看出DDETR在Encoder结构上显式地建模了多尺度特征的融合,而不像DETR一样只使用的CNN backbone的最后一层输出。至于Decoder,还是使用传统的全局自注意力,将Encoder的最后一层输出序列化后融入自己每一层的V和K矩阵中(同样是element-wise add),第一层的输入同样是一个需要训练的Object Query。这里的特征数量已经没有原来这么多,因此处理速度还可以接受。

关于具体的实现细节,作者开源了代码,请参考:GitHub - fundamentalvision/Deformable-DETR: Deformable DETR: Deformable Transformers for End-to-End Object Detection.


最后的最后,大家可以参照这篇文章:目标检测模型的评测与训练技巧,总结的非常好强烈推荐(虽然是两年多前的文章了但是还是非常棒,不过没怎么涉及到细节而是从宏观的角度上复盘了几年来目标检测的发展。建议作为复习或者扩展再自己详细查找相关资料)。如果能看懂/概览里面的内容并且没有任何压力说明你已经悄悄入门目标检测了。而这两年来的发展,主要是对于标签分配、anchor的思考和backbone的优化与训练调优等,单阶段和两阶段网络的差异被进一步模糊,Transfomer架构也在CV领域大放异彩,深度学习也被戏称为embedding-based的世界,更多的细节和内容就需要读者自己查阅资料了解了。

此处还推荐一个收集了各种检测网络论文、代码和对应讲解(如果有的话)的合辑:awesome-object-detection。大家也可以阅读一些综述,进行知识点的查缺补漏并探索自己的兴趣所在。

神经网络-卷积网络-检测网络部分至此在已经编写的内容中占了整个教程至少三分之一的篇幅,希望读者有所收获有所创作。至于分割任务和视频分析等其他任务,在比赛中用的比较少,在此不过多赘述。不过目标检测和图像分类作为CV的基石,许多trick和模块都是和其他任务相似的。有了目标检测的基础,相信你在其他任务的学习上也不会遇到什么困难。

包括最近大火的diffusion类生成模型,能够识别出数据集中不存在的类别CLIP-based模型,还有meta的不需要任何标注就能为图片进行按类分割甚至按实例分割(是不是又回归到聚类的感觉?)的SAM模型,它们背后的原理并无二致,仍然可以从机器学习的基础模型和算法推导得到。如果你能为深度学习中的trick找到其概率统计和优化的基础,这才表明你真正掌握了它。这里再次推荐李宏毅老师的深度学习课程,不论是介绍什么模型,他都会从最大后验、极大似然以及期望最大化等角度剖析模型的优化目标,而不仅仅是从直观上理解模型的设计

需要注意的是,深度学习领域的发展非常迅速,不像理科和传统工科可能还在教老祖宗的东西,十年前出现的模型如今很可能已经被抛弃,你很难在生成模型里找到flow-baed model的身影,LSTM也快被transformer打得找不着北。如果你有志于CV尤其是learning相关领域的研究,一定要日日新保持对领域前沿的跟踪。

paperwithcode这个网站收集了所有发表在顶会和期刊上CV任务相关的模型和算法以及开源(如果有的话),并且给出了每个算法在对应的一些数据集上的精度指标,也提供了数据集下载的功能,对于做CV和SLAM的同学来说非常方便。

理论部分到此结束,检测网络的实践将在第六章再次展开!



5.3.滤波器、观测器、估计器和预测方法

这些装置之间的区别在于面对不同场景时候的不同解释(是否考虑信号的统计学特性、随机特点、参数是否变化等)。滤波器在信号与系统中的定义为“只让特定频率的信号的通过的系统”,而在时域处理时则可以指滤除我们不想要的信号的装置,只有一个“前向传播”的过程,信号的流向是单向的。

观测器的概念来自控制理论,用于估计一些无法直接测量的状态变量。一般的控制系统中都有反馈通路,而状态观测问题和反馈控制实际上互为对偶问题,状态观测器大部分时候其本身就是一个带反馈的系统,其输入是信号来源和/或系统处理后的输出(此输出常被作为一个增广变量,将被观测器用于下一次迭代)。

估计器的概念和观测器类似,一般是用于对扰动的估计和在不能准确得到系统状态时对其进行估计的装置。

此部分分为两个方面介绍,一方面是传感器数据的处理(如解算得到的装甲板位置信息和从电控得到的信息,一般为时域信号(也可以转化到频域处理);另一部分是对图像数据的处理(二维信号,空间域)。

当开始阅读 5.3 ,笔者默认读者拥有积分变换、概率论、线性代数、微积分和部分信号处理的前置知识。请务必仔细阅读推荐的教程。

基于时域的滤波器是控制理论的核心内容之一,状态估计其实也是机器人学中的一个重要部分,自动驾驶中的SLAM就是它大显身手的领域之一(ESKF,MSCKF等滤波方法以及各种批量优化方法)。放眼RM赛场,比赛相关的技术无不和以上提到的内容有关,因此有必要学习这个重要的内容。在自瞄算法帧率逐渐上升、稳定性逐渐提高的今天,状态估计和轨迹预测将会成为兵家必争之地,是下一个强队之间对抗的隘口。谁能获得更好的预测和估计,谁就能实现更精确地打击。关于这些算法的具体应用我们将会在 6.4 中介绍,这里先不谈实践,着重看看原理和前置知识。

学习过信号与系统和自动控制原理等电类工科同学应该对此部分比较熟悉。这部分算法和代码的理解与设计是运动预测和估计非常重要的基础。这里推荐一个做机器人控制的工程师up:DR_CAN,他的几个系列视频都非常适合了解控制理论和学习。若仅是刚入门RMVisioner,请直接看图像处理部分。

5.3.1. 常用的几种信号滤波器

5.3.1.1 x通滤波器(频域)

一般在处理传感器数据(IMU、测距传感器等)和控制数据序列的时候会使用。我们这里以低通滤波器为例(显然我们的控制数据序列一般是稳定变化而不是突变,当然部分极端、特殊数据除外)。低通滤波器只会允许低频信号通过,从时域角度来看,就是变化平缓的信号量。比如我们从IMU采集的速度序列{5.6,5.2,5,5.5,6,5.7,110,6,5,5.6,5.3,6.1},突然出现的110这个值是较为平缓序列中的一个突变:若对此序列进行傅里叶变换就会高频处有一个尖峰,即代表着剧烈变化的信号。显然这对于IMU来说是一个异常信号,即使是突然启动也不可能有这么大的速度,除非遭到撞击(显然这时候的数据也没有任何用处),那么将此数据序列通过一个低通滤波器,就能够滤除这个异常值。当然我们在实际使用的时候,一般不会将一段时间内的信号序列投入傅里叶变换,而是直接设置一个跳变阈值来筛选这些信号。如果直接把此信号剔除,会出现一个数据为零的空档。一般选择通过插值(顺序平均、指数加权平均等)或者延续上一时刻的信号值来修复这个问题。

对短序列信号进行筛选我们可以得到一个理想的低通滤波器,这是它在频域的表示
5.3.1.2. 中值滤波器(时域)

和低通滤波器类似,用于消除异常数据的影响,这是一种非线性且有损的滤波器。而且由于其加权计算一段时间内的信号数据之和,通过中值滤波器后的数据一般会变得更平滑,有利于消除突变信号对系统的扰动。不过也要考虑有时候确实会有跳变信号的出现(比如当前帧的装甲板消失,在视野边缘出现了新的装甲板),要根据这种情况设计特殊的处理,通过调试设置阈值防止正常数据被清扫。

窗口大小为5和14;可以看到两者在幅值低频处有明显衰减,随后在高频呈现波浪状;相频也呈现一定的“周期性”

还可以采用投票平均法,仍然选取滑动窗口,不过去掉窗口内的最大值和最小值,然后对剩下的数据取平均。另有其他改进的方法如高斯平滑、指数加权平均、权重衰减平均等方法,能够适应更多的情况、更快的应对数据的变化等。

显然,这两种滤波器都需要采集一段时间的序列数据并进行相应计算和筛选,这是有时间滞后性和开销的,本质上都是低通滤波器,因此一定程度上会牺牲数据的实时性。从控制系统的角度考虑此问题,实际上就是反馈装置(传感器)的带宽设计,若带宽高的确可以提高系统的快速性但却会对高频噪声过于敏感导致高频误动作(电机抖动);若减小带宽则以损失响应速度为代价提高对异常数据的抗性。

我们常说,控制系统实际上是在以某个选定条件约束下的最优化问题,需要在数种性能和参数中取得权衡。这似乎也是机器学习中非常经典的“天下没有免费的午餐”定律?


5.3.1.3. Kalman Filter(卡尔曼滤波器,时域)

KF家族从被提出以来可谓是运用的最多最广的滤波器了,并且其发明者Kalman教授以他的一己之力将控制领域的研究从频域转换到时域(以状态空间设计为基础的现代控制理论)上来,虽然他在后半生一直热衷于频域分析的研究。Kalman滤波器最出名的轶事就是在Kalman博士访问NASA后,那里的工程师认为这是一个极佳的时域修正方法,并且直接在阿波罗6登月飞行器的轨道矫正系统上使用了扩展卡尔曼滤波器。

相比于传统的时域滤波器和频域滤波器,KF运用马尔卡夫假设(从概率和信息观测的角度),只需要保留前一个状态的信息,占用内存小且速度非常快。

这一小节主要推荐一些非常棒的教程。关于KF的工程应用和直观理解,Matlab官方出品的这个系列视频很好地讲解了运用过程:Kalman Filter。另外若想从概率统计和数学角度初步理解KF,可以进一步参照这篇可可爱爱的教程:How a Kalman Filter works。

若是结合控制和概率论解释,并且完整地进行数学推导,请看bilibili up主 DR_CAN的:卡尔曼滤波器系列视频。另外,他从工程的角度讲解了状态空间法中的观测器设计,这能让你更好地从控制角度理解KF,请戳:龙伯格状态观测器的设计。

若你学习过自动控制原理(准确的说是现代控制理论,主要用到的是关于稳定性分析和线性代数的知识),想从李雅普诺夫稳定性角度分析为什么KF是高斯噪声影响下的最优线性滤波器和其收敛性、抗扰能力,请戳卡尔曼滤波器原理介绍。

如果想进一步了解状态估计和机器人学中的概率论相关知识,可以参阅《state esstimation of robotics》和《probabilitics robotics》这两本经典的作品中相关的内容,里面详尽地介绍了KF为什么是高斯意义下地最优滤波器以及作为一个贝叶斯估计器的扩展,KF是怎样演化成这种形式的。

对不同领域来说,卡尔曼滤波有不同的推导方法,如从最大后验估计、协方差最优的角度(这两个是最常见的推导方式)。对于KF和EKF需要特别注意的点,可以参考笔者的文章:卡尔曼滤波和EKF线性化等的一些细节。

KF从多个信息来源获取数据后的融合过程图示
  • 由KF的构成我们可以发现,它能够综合多个数据来源(观测模型)和系统状态转移的模型并根据他们置信度(由协方差的大小来表示)来对各个信息来源赋予不同的权重最终融合得到预测结果。而EKF、UKF都是对非线性系统作近似线性化而产生的滤波器,对于工作点附近非线性程度较小、容易线性化近似的系统(容易指的是求导生成的Jacobian矩阵较好计算,不会包含很多非线性函数)有着很好的效果。显然在比赛中大部分的控制数据和传感器数据都是能够比较好地用近似线性化来拟合的,比如在传感器帧率高、处理速度快的时候我们常常假设对手机器人的运动是匀速直线运动,因此KF系列便顺利成章地作为我们使用滤波器的首选。

  • 若我们想用KF来修正之前的数据(如t-1,t-2时刻),那么卡尔曼滤波器相当于一个插值或平滑数据的装置;若想通过KF来融合当前采集的 数据并且消除噪声,那么它是一个估计器,和前者一样都是贝叶斯方法,即利用后验信息来修正概率模型;若想通过它来预测t+1时刻的值即通过修改状态转移矩阵用历史数据来推测将来值,这时候就是一个预测器(单纯从名词直译来理解)。

  • 有了以上教程的铺垫,我们就可以通过修改KF的状态转移矩阵来预测将来的数据了:以装甲板的运动为例,我们一般对解算得到的装甲板的中点进行运动学建模并假设装甲板的运动模型(一般假设为匀速直线或匀加速直线)和观测模型(也就是相机的成像模型,可以使用小孔成像)的非线性程度较小。这样我们便能选用EKF来对其进行线性化。我们可以把当前帧和之前若干帧的装甲板位置的世界坐标系坐标相减并除以相机的帧率的倒数得到其运动速度,或者进一步求差分以得到加速度信息。在得到这些信息后,把他们填入状态转移矩阵并设置一个合理的时间间隔(曝光时间+单帧处理时间+数据发送到电控的通信时间+电机响应时间+子弹飞行时间等,一般需要实车调试进行修改),EKF便能根据转移矩阵和置信度计算出此确定的时间间隔后装甲板的位置了。我们会在 5.3.2 进一步介绍运动预测方法,在 6.4 也会详细讲解如何利用KF对装甲板的运动进行预测。

  • UKF则是EKF的改进版,全称为Unscented KF。当系统模型的非线性程度较大的时候,我们就很难用局部线性化的方法去得到贴近实际的近似分布,因此我们会在参数未进入系统前随机生成服从高斯分布的一组点同时给它们赋予不同的权重(为什么要这么做?),然后将他们一起投入到此非线性变换模型中,并在变换结束后统计这些点的均值和方差,利用这些数据重建一个高斯分布(点参数估计)用于近似此模型的分布。显然,使用的随机点越多则近似越准确(大数定律),但是消耗的资源也越大,这点和下一个要介绍的PF是完全一样的。不过,在实践中的经验来看,UKF对高维模型的近似并不是很好,一来是高维空间的分布更加复杂难以通过线性系统(在这里我们使用高斯分布)近似,另一方面是计算开销也随着维度增加而迅速上涨,维度爆炸是这两种方法(UKF和PF)共同的、最大的问题。

好在我们的系统并不复杂,最多只有速度、位置、加速度三个维度(加上加加速度和加加加速度也是没有任何问题的,如果你觉得者能够提升你模型的精确度并且这些数值的噪声足够小的话,或者你能够找到其他能够增加置信度的先验信息和其他可用传感器数据)。

UKF的流程图解

图源知乎-"fishmarch"的专栏'概率机器人'

根据上图我们可以把UKF的迭代过程分为以下四步:

  1. 选取服从Gauss分布的随机点

  2. 和数据一起投入变换

  3. 根据权重和新的均值方差得到新的Gauss分布,并假设此高斯分布就是经过变换后状态的近似分布

  • 此外,卡尔曼滤波器还有各式各样的变种,如SCKF(状态约束)、ESKF(误差状态)、AUKF(增强无迹)等,有兴趣的读者可以自行了解相关的内容,无非是从不同状态变量的关系之间入手增加额外信息或约束以进一步逼近真值,或是通过更合理的方法来近似非线性部分。

5.3.1.4. Particle Filter(粒子滤波器,时域)

PF可以看作是UKF的进化版。UKF要求用服从高斯分布的一组点经过转换之后去通过采样来得到新的高斯分布进而近似真实的状态分布;而PF则是不再追求用高斯分布去近似真实分布,直接用一组点经过模型转换后再采样,用此后验数据来近似任意分布。它们的区别在于UKF是通过一组假设在通过系统的转换后仍然服从高斯分布的采样点来求其参数μ和σ,是一种参数估计方法;PF是用已知的采样点数据去求未知的任意分布,或者从另一个角度来说,就是干脆不给出后验分布函数(不用分布来描述,就是一堆离散化的点),直接根据粒子的权重进行融合得到状态的一种可能假设。根据大数定律(又是它!),想要通过采样近似一个分布,样本越多则结果越接近真实分布,因此PF在“撒豆子”时会使用比UKF多得多的样本点,所以它本质上是一种非参数化的蒙特卡洛方法。

UKF(上)和PF(下)最直观的区别:UKF用根据一组点的权重进行极大似然估计,重建高斯分布;PF直接给采样点分配权重,即图中竖线的密度,之后越密集的点附近被重新采样到的概率会越大

在PF的初始化过程中生成的这些样本点经过观测矩阵、状态转移矩阵变换后,输出一组经过非线性变换得到的“畸形”的分布点。现在,我们要对这些“畸形”的分布点进行重新采样以估计系统的状态,不过,我们不再像EKF、UKF一样去用这些采样得到的点来还原分布了,我们直接把这些采样得到的点作为后验信息(通过对这些点进行加权,得到关于系统的状态信息,这时候不再讨论“分布”,上面提到用后验数据来近似任意分布,既然是任意的分布,那么也不存在一个既定的模型需要通过这些点来计算其参数即进行参数估计,这些点的信息本身就代表着一种分布)。

那么应该如何赋予权重并重新采样?这一步就对应着KF中的观测,我们会根据观测得到的数据为上面的粒子赋予权重(通过观测来得到各个粒子所处对应状态的概率),接着把那些概率小的粒子排除出去(这些概率小的粒子附近位置被重新采样的概率更小),在权重高的粒子附近根据权重重新采样。这样就完成了PF的一次迭代更新,pipeline自然就重新回到第一步。如此往复循环,滤波器输出就能收敛到最“贴近”实际状态的分布了。

这里有一个关于粒子滤波器的简短视频,能够帮助你从直观上理解PF:Uppsala University:Particle Filter Course。上面的介绍都是倾向于通过直观的解释帮助你理解,如果需要采用严谨的数学推导从概率统计、贝叶斯估计的视角入手,也请参阅相关的教材。笔者还没有见到从控制和观测器的角度解释这样的非线性滤波器(主要是笔者的对这方面的知之甚少),若你知道一些相关的材料,请联系我!


接下来两个滤波器对数学和信号处理的知识要求较高,没有相关基础的萌新 RMer可以跳过这两个滤波器的介绍直接看结论。

5.3.1.5. Least Mean Square Error Filter(最小均方误差滤波器,时域)

经典的频域滤波器要求对信号与噪声的频谱完全得知,并且两者不能有交叠,否则会造成滤除不干净或是影响原有信号的问题。而估计器则是通过已知信息求在某种标准下的原信号的最优估计。LMSEF是维纳博士提出的以最小均方误差为最优准则的一种线性滤波器,奠定了最优滤波器设计的基础。其对于参考信号(或目标信号,即我们希望得到的信号)和噪声信号的要求是平稳随机信号并且统计特性已知(如果统计特性未知或是会改变,请看下面的自适应滤波器)。维纳滤波的最佳适用场景是处理已经被记录下来的信号,或是能够长时间观测的信号。

假设参考信号为,我们采集到的信号序列为(可以将其看成一个向量),噪声为(也看作向量)。我们设经过滤波器后的估计值为,对于信号卷积形式是

这里的就是对于采样的输入信号的冲激响应;如果将估计值写成的线性组合形式则有:
这里就是在不同时间点的值,也就是对 序列不同值的加权系数。进一步的,设误差序列为:
则均方误差为:

使得均方误差有最小值,将的表达式展开,我们将得到一个半正定的二次型:

,因此通过取一特定的w会有一个全局最小解,我们的目的就是利用已知的信息求取一个.

接下来的想法就很简单也很直观了,我们直接对误差函数求导,并让其导数为零,解出的就可以让误差函数最小了:

令上式等于零,解得:
式中的两个期望,就分别是采样信号的自相关和原信号和我们采集到信号的互相关。如果已经得知这些信息,那么就可以求取全局最小解了。从前述的推导就可以发现,我们实际上是在采集到一个序列信号:
之后,再对输入信号进行滤波从而获取原信号的最优估计。因此,当我们面对已经完成采集的图像(相当于在拍摄瞬间完成所有光强信号的定格,序列是一个空间序列,又叫空间域)、音频(扬声器产生了失真,需要进行还原)、dian等,就可以使用wiener filter对其进行处理。这是因为我们已经获取的关于这些信号的所有可知信息,只需要假设噪声形式就可以直接计算相关矩阵(虽然不是无限区间)。

不过实际上我们无法得到一个信号序列的这些理想参数,因此通常利用实际输入信号的采样对自相关系数矩阵(一个对称方阵,元素分别为,之间两两的相关系数)和互相关系数矩阵进行无偏估计(指被估量的期望和实际值的期望相同),用此统计量来代替真值。该公式也称为维纳-霍夫方程,是最小均方误差准则下的滤波器系数最优解。利用部分信号的统计信息来进行滤波也启发了自适应滤波,稍后我们会进行介绍。

显然,对于一个动态的任意信号而言,若需要实时进行处理,我们获取该信号与输入的互相关还有输入信号的自相关这些先验信息难度又更大了,我们只能获取部分的数据并且得到的估计基本上有较大的偏差。由于输入过程取决于外界的信号、干扰环境,这种环境的统计特性常常是未知的、变化的,导致我们无法在获取信号的过程中实时求取.

如果不知道被估计量的任何信息则无法使用LMSEF。但可以对这些先验进行假设使其符合一定的形式,和实际情况越贴近则滤波效果越好。

同样,以其他准则为最优的滤波器或多或少也会受到需要统计特性已知或平稳信号的限制,这也促使人们开发出了自适应滤波器。此外,由于要进行矩阵的求逆运算,若序列数据的长度很大或是数据的维度和规模很大,我们会遇到“维数灾难”,求解的难度乘指数级上升。

wiener filter和linear filter的对比

相对于一般的线性滤波器(由虚线BC表示的理想滤波器),维纳滤波的响应(上图中的斜线)在噪声和信号的重叠处就开始衰减,使得噪声对原信号的影响减小。其倾斜程度和曲率取决于原信号和噪声的强度比。

5.3.1.6. 自适应滤波

自适应滤波以前述的wiener filter为基础,这里我们也从统计与优化和控制理论两个角度来介绍它,顺着上一篇介绍维纳滤波器的文章的思路继续进行推导。

  • 统计与优化

前述的wiener filter在求取最优解的时候受到的限制很多,不仅要求噪声和目标信号的统计特性已知,还需要采集无限时间的信号序列(从零到正无穷)后以得到最优解,而在实际情况下我们一般利用一段时间内的信号序列对做出无偏估计。不过,这也只是权宜之计。首先倘若要对序列信号进行实时处理,采集一段时间这个要求就无法满足。其次若对信号和噪声的统计特性完全未知甚至它们是非平稳态的,那么问题相当于无解。

不过,我们可以从维纳滤波器的近似求解中看出一些蛛丝马迹:利用一小段时间内的信号对误差的期望进行最小化,如果再对这个结果进行推广,缩小这个“一小段时间”的长度进行估计并保存这次的结果,认为是当前最佳同时在稍后进行迭代,不再追求最优解而是尽量消除噪声并尽可能快速的收敛,不也是一种很好的方法吗?

那要怎么才能知道要如何更新呢?对于全局最优解,我们要求一个使得下式为零:

并且之前我们已经得到,我们要优化的误差函数拥有很好的性质:它是一个凸函数,其上没一点的梯度都指向最优解!很直觉的想法就是,我们让每次往最优解迭代若干步,即,其中k为调制因子(学习率),这不就是梯度下降法吗!如果在每一次采集完成后都进行数据更新,就成为了随机梯度下降法。很多的优化问题如果无法一次性求解,往往会诉诸迭代的解法,逐步逼近最终的解。对于极限情况,我们每采集一次数据就进行一次参数估计,即每一步都更新 。

不过这里还有一个问题亟待解决,要计算处代价函数上某一点的梯度,还是需要输入信号和原信号的一些统计特性,因此我们让代价函数就等于误差向量,认为当前的样本的损失可以替代总损失。学习过优化或者神经网络的朋友应该对此应该不陌生了,那么求梯度就有:

其实也就是把式(1)中的后半部分用代替然后用结合律和前半部分相消(当然这种视角不太严谨,比较直觉,因为矩阵乘法没对齐,时间戳也不是对应的)。

所以我们只需要之前的加上当前梯度乘上一个系数就可以达到参数更新的目的了。对于稳态信号,逐渐迭代就能够收敛到最优解附近(根据推导可以发现每次的误差一定会比上次来的小);而对于非平稳信号,也可在数十步之内逐渐跟上,但不可避免的有相较于稳态信号稍大的误差。

对于k值的选取,经过稳定性计算可以得出以下条件:

其中λ为输入信号自相关矩阵的最大特征值,感兴趣的同学可以自行查找资料,大概证明过程是为了保证代价函数会随着的迭代递降,根据递推公式,只需要保证特征值小于1就会让滤波器的输出值收敛。越大收敛得越快,但也容易跑飞或者使得累积误差变得的不可接受。不熟悉特征矩阵、对角化以及幂矩阵运算,请戳这里:MIT 18.06 linear algebra P23

所以最终权重更新的公式为:

搜索关键词:自适应滤波、泰勒展开、特征分解、学习率。

虽然已经找到了求解的方法,但它肯定还是有不足之处,比如收敛速度慢、震荡比较剧烈等。针对这些缺点人们又提出了NLMS、AP、RLS优化等,有些是针对梯度下降的(可以参考之前神经网络部分的adam和momentum);有些是针对噪声和误差统计特性的;感兴趣的同学可以自己搜索。

几个参考资料:adaptic noise canceling、

  • 控制理论

待更新。笔者还需要进一步查找相关资料。

  • 应用举例

  • 降噪

网络上能找到非常多自适应滤波器用于降噪的例子,尤其是AEC(自适应回声消除)常常被作为样例,但是却存在大量的缺解、讹误和疏漏,甚至在根本没有弄清楚原理的情况下就大肆复制粘贴添油加醋。笔者在初看的时候就是一头雾水满是疑惑,在中文网络上越查便越发头大。经过长达三天的查证和学习,在此归纳总结。

如下图所示,为带有噪声的输入,为和噪声相关的参考信号,那么我们的目的是将作为自适应滤波器的输入,其输出能尽量逼近。那么经过一段时间的迭代后,就会逼近我们想要的原信号.

注意不要把这里的和前文的符号记法混淆。

然而我们发现,如果不为零,那自适应滤波器在进行迭代的时候始终会得到非零的差值,这样不就相当于输出会逐渐逼近了吗?这也是非常匪夷所思的事情,怎么可能用一个滤波器从噪声信号中还原出原信号呢?

其实,在利用adaptive filter进行noise canceling的时候,有三个前提条件必须得到满足:

  1. ,这要求原信号和噪声信号不相关
  2. ,即要求用于参考的信号和噪声信号相关,并且相关程度越大越好
  3. ,希望原信号是一个均值为零的信号

有了这三个条件,我们再来看前述的内容就好理解了。的期望为零而又不恒等于零,则在渐进收敛的过程中,在一段时间内为正一段时间为负,这样对于滤波器的参数来说,其实是保持动态静止或者说于最优值附近轻微波动的(当几乎收敛的时候,根据参数的更新公式,当误差为正则向正方向更新一步,误差为负则向反方向更新一步,总的结果是保持不变的,但在还未收敛的时候,总有一个方向上更新的距离会超过另一个方向,这个更新步长较大的方向就是趋向解的方向)。

因此,我们避免了"不断更新使得经过滤波器后的输出逼近”这种给谬论。比如语音信号、心电信号、还有其他许多电信号都有这样的性质,把直流值当成零参考点,其震荡信号的均值为零,这就很好地符合了第三条要求。

笔者也有看到将自适应滤波视为去相关操作的解释:若输入和参考信号有关,就能够利用相关性动态计算参数使得输出不断逼近参考信号,而对于无关信号(在这里就是,没有被噪声污染的原信号)则不去理会。在此例中参考输入是一个和噪声高度相关的信号,而原信号即需要被还原出来的信号则于参考输入无关。

我们也拿典型的AEC应用进行说明,在此场景下,我们希望消除麦克风采集到扬声器中的声音所产生的回声(你应该也有这样的经历,在线上会议或免提通话时,我们的声音在对方的扬声器中播放时又被对方的麦克风采集到,随后这段信号再次由信道传播回我们的扬声器中播放出来,形成回声),那么期望的应该是麦克风采集到的去除回声之后我们说话的声音:

AEC的应用场景示意图,图源本节末的链接

我们会发现,这里其实增加了一个DTD(double talk detection)模块,他用于检测当前双方是否在说话。这又有什么用呢?前述进行noise canceling的时候必须满足三个条件,在实际情况中第二条是比较容易实现的,但是1、3满足起来就比较困难了,因此利用DTD,回声消除的模式如下:

  • 远端语音存在,近端语音不存在:滤波、自适应滤波器系数更新(对方说话我们不说话)
  • 远端语音存在,近端语音存在:滤波(两方都说话)
  • 远端语音不存在:什么都不用做(没信号过来,没必要做任何动作)

很容易就能发现,我们只有在原信号(我们说话的声音)为零的时候即第一种情况,才会对滤波器的参数进行更新,这就避免了原信号和希望消除的噪声信号之间的耦合导致参数更新不准确的问题。在这种情况下,自适应滤波器的作用原理反倒和接下来要介绍的系统辨识有些相像了:其实我们在利用自适应滤波器去模拟回声路径,让输入经过自适应滤波器后的输出和输入通过环境反射交杂后产生的输出尽量接近。

关于AEC更详细的综述,可以参考这篇文章:声学回声消除原理与实现

  • 系统辨识(应用于控制等)

对于系统辨识的应用,我们希望自适应滤波器可以模拟出待估系统的行为。这里,,滤波器的输出和原系统的输出作差得到误差信号,将此误差输入自适应算法,经过数次迭代将会逐渐逼近参考信号。也就是说,虽然系统内部结构不同,但是自适应滤波器复刻了的传递函数。

在模型十分复杂的情况下,我们就可以使用adaptive filter去逼近模型的黑箱。用一个直观的低维实例来说,就是在用一条直线去逼近一个复杂的曲线。当然,你可以用更高阶数的自适应滤波器从而达到更好的效果,或是在此之后增加其他的非线性拟合部分。

  • 估计(预测)

很多读者初看这个应用,会觉得有些无厘头,为什么能够利用过去的值得到当前值呢?请记住,自适应滤波器虽然使用限制相对维纳滤波器放宽了许多,但是仍然要求信号在短时间内能够保持一致的统计特性或不至于变化太多,如果是一个剧烈时变的信号显然我们无法通过历史数据序列来预测将来时刻的值的。

对于已经学习过R循环神经网络的同学来说,我们可以将自适应滤波器看成是单层RNN,滑窗范围就是参与计算的历史序列长度 。不过两者稍有区别,自适应滤波器的参数会一直保持更新,而RNN需要先训练再使用,或者我们将前者看作是一种一直处于学习状态的RNN。


5.3.2. 预测方法

也可以称为目标运动估计或运动学建模?

运动预测是自瞄击打装甲板的关键一步,如果没能预测目标的移动,那么我们的解算输出用于只会跟在目标的屁股后面,使得打出的弹丸始终落后于目标,扑向空气划过一道淡绿色的曲线。同时,运动预测和运动学建模也是目标跟踪领域的一个研究方向,利用此技术能够提出精确的区域提议,使得候选区域数目降低,加快处理速度。

我们这里将按照循序渐进的原则,介绍运动预测的发展和方法。

5.3.2.1. 朴素方法:简单物理规律的应用

这种算法就如标题一样简单,我们直接根据物理规律对物体的运动进行预测。假设物体处于匀速直线运动或匀加速直线运动,利用两个时刻物体的位置差求差分得到速度,或进一步对速度求差分得到加速度,用此值乘上需要超前的预测时间(超前量),就可以得到目标物体在这段时间之后所处的位置了。

倘若只使用两帧之间的位置差分,噪声的影响可能会比较大,可以考虑使用多帧的滑窗进行直接平均、或EMA(指数加权平均)等加权平均方法,尽量降低噪声的影响。

从上面的描述就可以发现,这种模型只对运动状态变化较慢的目标有效,即使我们是算法处理帧率再高计算得到的差分再准确,那也只是基于匀速或匀加速模型得到的结果;虽然对于有任意速度或加速度的物体在短时间内我们都可以将其建模为CV或CA,然而一旦目标的运动状态变化速度超过了可以接受的上限值或预测的时间超过了上限值,准确率都会大幅度下降。(不过话说回来,所有预测方法都是有一定的约束的)

5.3.2.2. KF及其优化

这些方法本质上都是基于模型的方法,即给定一个确定的运动模型,然后根据已知信息推算目标的状态并融合一个可用的观测,以获得更高精度的结果并更新模型的参数。

  • 使用卡尔曼滤波器或其他观测器,都是为了达到消除噪声从而获取更好的位置估计的目的,这样融合出来的差分和二阶差分等相对前一种方法会更加可靠。

若采用匀速模型,则在状态变量中加入速度,update步应该分为两部分:

  1. 将用于更新状态变量的状态转移矩阵中对应速度的位置的值设为,即两帧之间的时间差
  2. 增加一个新的状态转移矩阵,在对应的位置上设置的值为,这是需要超前的预测时间,在RoboMaster视觉任务中, 包括算法处理时间、下位机通信时间、电机动作时间、单链移动时间和子弹飞行时间。

第一步得到的状态变量会一直继续使用下去,这将作为当前位置和速度的最优估计并不断保持更新,和前面朴素方法介绍的多帧平均的作用一致;而第二部分计算出来的值,是用来提供超前位置的,每次都根据第一步的结果重新计算。

对于协方差矩阵的设置,我们从两个极端入手考虑:若将预测值的速度协方差设定的非常大,则我们几乎只相信测量值,即两帧之间的差分值,那么这就不是一个匀速直线运动了,得到的融合结果可能会呈现折线状,需要稍微进行平滑以获得连贯一些的控制数据;将与预测值的协方差设定的很小的时候,那么我们会更相信预测值,认为其是一个近似的匀速直线运动。

当然,我们也可以把加速度加入状态变量中,用二阶差分来计算加速度,同样可以采用前面平均速度的方法来平滑加速度。不过二阶差分本身就容易带来较大的噪声,对于帧差的准确度要求更高,尤其是我们使用SolvePnP算法常出现定位点抖动的情况导致帧差出现高频噪声。从观测器的状态反馈权衡的角度考虑,相当于给了一个很大的增益,其带宽会扩展但是容易被高频噪声所干扰。

  • 上述的方法本质上还是基于匀加速或匀速这种较好处理的模型,只不过采用观测器尽量减少噪声,还有一些稍微复杂一些的模型采用了不同的假设和约束,使得预测的效果更好。我们在此简要列出几个供读者进一步参考。

首先很容易想到的就是其他constant模型:既然有平动那肯定也有转动的匀速模型和匀加速模型吧?这就是Constant Turn和Constant Angular acceleration。那能不能把转动和平动结合起来?肯定能啊,这就有了CTRV、CSAV、CCA,这些模型在自动驾驶的运动学建模中都用得非常广泛。

再比如将加速度或速度建模为服从一定分布的思想也是很巧妙的。singer模型假设目标的加速度符合一个的均值分布,其服从一阶随机微分方程,可以用目标机动时间常数衡量一段时间内加速度的相关性。哈工威的视觉有一场圆桌介绍了他们的singer模型,参考:哈工大威海hero:神经网络优化与运动预测算法分享。Matlab中也有一个相关示例:Singer acceleration motion model

当前模型(current model)对singer进行了改进,考虑了加速度的均值,不像前者将加速度均值设为零,其加速度是一个条件分布,将根据一段时间内观测到的加速度均值进行计算更新,因此又被成为均值自适应加速度模型。

  • 最后就是不同模型的集大成者,可以看作机器学习中的集成方法——IMM(Interact multi model),它综合了多个单一模型(就是上述的那些),分为四步进行运动预测和更新,相信已经学习过卡尔曼滤波的你应该大概能猜想到他的流程,IMM最大的不同就是增加了第一步,让模型进行交互计算相互转移的概率:
  1. 模型交互,计算目标运动模型从一种模型切换到另一种模型的概率(一会看了后面三步会更加清晰),然后为所有模型两两计算转移概率并乘以每个模型的输出得到新一轮的初始状态。
  2. 以第1步得到的初始状态(1个)作为每个单一模型的输入,然后用各自的模型进行预测和量测更新。这一步和KF完全相同,而且实践中大部分单一模型的状态转移和融合都是使用EKF或UKF。
  3. 在第2步更新完之后,利用在更新时每个模型预测值和测量值的差值,计算每个单一模型的似然函数值进而得到该模型的置信度。也就是说,单一模型的预测值和真值(其实是测量值)差距越大,说明这个模型和实际情况相差的越远。
  4. 根据第3步得到的每个模型的置信度(权重),对所有模型的结果进行加权融合得到输出。

IMM实际上就像一个超大杯的KF,里面有许多小KF,小KF的输出又再次在大KF中进行融合,从机器学习的角度看应该算是一种stacking ensemble,每个小模型的输出再投入一个模型中进行集成。比较巧妙的设计就是IMM的第一步,它没有直接用上一轮迭代中的输出值作为新一轮迭代的初始状态而是添加了一个模型转移概率,使得不同单一模型之间的交互更充分。归根到底,IMM也是一种贝叶斯方法,利用先验信息和后验估计来获取极大似然估计。若希望看到详细的数学推导,请自行查找相关的论文。

IMM一轮迭代的流程
5.3.2.3. 基于运动行为和环境交互
  • 这种建模方法就是考虑更多的、约束性更强的先验信息,比如汽车在十字路口要么转弯要么直行,行人过马路会走斑马线(虽然有人不走);在RoboMaster比赛中的例子也有很多,机器人往公路区走时就有可能要飞坡了;往环形高地走必然是要上坡了;检测到小陀螺状态,那就是处在对方机器人处于自转状态了......(第六部分会介绍反陀螺)

  • 对于这些先验信息,我们可以用状态机的方式进行建模,一旦判断进入某种状态,就采用一种应对方式。虽然看起来简单,但如何确定处于哪种特定的状态也是其编程难点。

车辆和行人在十字路口的行为,图源《无人驾驶中的决策规划控制技术》
5.3.2.4. RNN
  • 虽然说遇事不决量子力学,但在算法的世界里,要是难以建模怎么办?神经网络呗!将目标的运动离散化,得到的就是在空间位置中的一个序列,那么我们只需要将之前位置的三维向量作为输入(也可以添加其他信息比如速度和加速度,但是笔者认为速度和加速度这么简单的、通过差分运算就能得到的信息,网络很容易就可以学习到),让网络输出一段时间后的位置即可。

  • 我们也可以结合上一种方法,让RNN输出当前的运动行为和状态,然后再用模型的方法进行进一步的计算;或是反其道而行之,将可能的运动行为输入RNN中,使得RNN的预测输出更为精确。不过显而易见,想要融合更多信息,代价就是数据集获取和制作难度的上升和运算速度的下降。

  • 也可以尝试当下火热的Transformer以及其后继RWKV等,由于其不限制序列长度且不需要像LSTM和RNN一样“记住”之前的信息,使得它对序列信息的建模能力更强。添加时间注意力能让网络聚焦到关键的运动信息上。


5.3.3. 在图像处理中应用的滤波器:

这部分本来准备在OpenCV中地常用函数部分介绍的,但后来又想到在前面已经稍微涉及到了那些函数的作用,那么在这里就从信号处理的角度对二维信号的分析进行叙述。这一小节概括的内容可以算是真正的图像处理,即利用一定的方法增强/提取图像的特征、消除可能的干扰等,也是比赛中传统视觉算法的实现基础。

我们也可以将图像处理、机器视觉和计算机视觉进行简单的对比,图像处理更像是传统的信号处理,把图像视为空间上二维的亮度序列信号,通过逐像素操作或卷积操作等,提取图像的基本特征;机器视觉则是着重关注测量,需要得到的是严格定量的分析。如某个物体的长宽、面积、角度还有距离相机的位姿和距离等,大部分应用场景是工业自动化生产,一般需要用到图像处理的知识;计算机视觉的抽象层次最高,希望计算机拥有和人一样的理解能力,不单单是进行测量,而要抽取出高级的语义特征等。拿一张含有水杯的图片作为这三个任务的输入,图像处理输出的可能是利用canny边缘检测后得到的水杯轮廓,机器视觉算法也许会计算出水杯的真实高度,而计算机视觉程序则能告诉我们,图像中有一个水杯,甚至可以利用目标检测技术将其精确定位出来。

本节就介绍最基本的任务,图像处理。

5.3.3.1. 频域分析

对于学过信号与系统的同学来说,Fourier Transformation已经是我们的老朋友了。既然它能处理一维序列,自然也能处理二维信号,图像就是一个二维信号,我们当然可以用它来对付图像了。我们着重关注的是二维的DFT(离散傅里叶变换),希望傅里叶变换能帮我们提取出图像的频率信息。对于时间序列,频域上的幅值低对应着不同频率正弦波的强度大小,频率越高则说明变化越剧烈;那么我们直接将这种关系推广到二维空间信号,频率高低就对应着明暗过渡的缓和程度,对于物体轮廓和边缘,亮度的突变就代表了高频信号,而平缓的区块则对应了低频分量。

下图中展示了三个基本频率分量(还有一种往右边倾斜的分量),频域和空间域仍然保持着对偶关系,高频分量在频域中间隔大,低频则是集中。对于表示信号非常关键的相位信息,也一起保存在频域中。不过由于要完整的表示出频域分量,需要四个维度(时域中的每个维度都对应一个复数,这是一个空间),因此一般将相位谱和频率谱分开表示。

图像的基本频率分量,以及他们在空间域和频域的对应关系

那么,回到最根本的问题,为什么我们需要傅里叶变换?因为我们想要利用变换域的一些优越的性质——比如卷积在频域只是简单的相乘操作。之前我们介绍了Sobel算子和Laplcian算子在频域的实现都非常容易(微分变成了了标量乘法),低通滤波器、高通滤波器、带通滤波器更是手到擒来。线性时不变系统(在这里则是“线性空间不变系统”)的优良性质让我们可以单独处理每个信号分量,再通过可加性结合所有成分。

对一张图片分别应用高通和低通滤波器,图源MIT opencourse 6.819

你可以设计特定的滤波器来提取需要的图像分量,在之前介绍Vision attention的时候,我们就提到一个通道的数据就相当于原图的某个“频率分量“,只不过CNN的卷积核参数是利用反向传播自我学习出来的,而不是用傅里叶变换计算出来的。从这个角度上看,傅里叶变换产生的特征(正弦分量)就要比神经网络训练出的特征更”低级“一些。同样,我们可以取二者之长,比如给予每个正弦分量一个可学习的系数,让网络通过反向传播训练此系数,得到一组组合滤波器,这相对最基本的DFT能更好地提取图像特征,又可以网络降低训练的开销不需要单独学习卷积核上的每一个参数,提高运行速度(DFT的运行速度已经被优化到极致了,打过OI或者玩信号处理的同学一定听说过/掌握了快速傅里叶变换FFT)。

眯起眼睛或取下眼镜,你将获得一个低通滤波器,看到作家的人头画像

相比要求信号平稳且噪声特性不变的FFT,你可能还听过小波变换(wavelet),信号处理中它的应用非常广泛,它在一些场景下能得到更好的结果。

5.3.3.2. 空域滤波

在信号与系统、控制理论中我们常常从时域和频域分析对象,那么在图像处理上我们也可以从空间域对图像进行操作。最常见的比如差分算子、二阶差分算子、高斯核等大家应该已经见怪不怪了;对于单一像素的逐个操作如阈值、双阈值、直方图归一化还有对比度增强等,也可以看作是非线性的滤波;还有之前介绍的自适应滤波,同样可以用在图像降噪去雾上。

  • 降噪

因为CMOS损坏、增益过高或是在低曝光、低亮度条件下拍摄照片时画面常常伴随噪点或椒盐噪声,最简单高效的方法就是利用均值滤波、中值滤波、高斯滤波平滑/去除这种噪点。如果是每个像素上都伴随随机噪声,根据其特性可以选用定制的自适应滤波(动态改变窗口大小和参数选择)或其他的相关滤波方法处理。不过,我们需要权衡效果和时间开销,如果前处理就花费了大量的时间,对算法的实时性是非常不利的。

上图为随机噪声,下图为高斯噪声,图源MIT opencourse 6.819
  • 特征提取

用于梯度计算的差分算子已经是老熟人了,提取边缘可用sobel算子和Laplace算子,比较少见的有提取倾斜边缘的roberts、只处理垂直方向的prewitt算子,上述的这些卷积核都是根据不同情况选用的,当然还有增加预处理和后处理的Canny边缘检测。这里列出他们的形状、适用情况及其优缺点。

  • Roberts算子,从核的设计就可以看出,比较适合处理倾斜的边缘,对于本身边缘较为明显的图像,处理效果较好,但是边缘响应较宽而且会受到噪声影响,若要缩小响应宽度可以增加后处理,减小噪声影响则增大卷积核:
  • Prewitt算子,专注提取水平和竖直方向的边缘,鲁棒性相比前者更佳,但是提取出的倾斜边缘会呈现锯齿状,对于亮度变化不那么剧烈的图像也比较有效:

  • Sobel算子,相当于使用了高斯插值的边缘微分算子,对于离中心更远的像素具有更小的权重(很直观,比如下方的核中,左上角对卷积核中心点的贡献应该比正上方小)。由于加入了高斯滤波,噪声容限相对prewitt有所提升,若希望得到更平滑的结果,同样可以扩大卷积核:

  • Laplacian算子,对于离散的图像信号来说是二阶差分算子,能够更明显的凸出图像边缘:

    这是四方向的拉普拉斯算子,还有8方向(考虑两个斜45°角)的卷积核如下:
    如果想要凸显效果或减弱效果,读者可以自行尝试给边缘和中心设定不同的值,利用OpenCV中的filter2D()函数进行测试。

  • Canny边缘检测在各处有非常多的教程,OpenCV documentaion中也有示例,这里就简略过一下:

  1. 高斯滤波,把噪声去除,显然差分运算对冲激噪声是非常敏感的

  2. 用梯度算子求出边缘响应

  3. 非极大值抑制,根据把边缘缩窄/删除误检边缘

  4. 双阈值操作确定响应上下界,把范围内的响应留下

  5. 聚类?强响应附近的弱响应也会加入,而弱响应附近的弱响应直接删除

5.3.3.3. 形态学操作

形态学操作是对二值图的轮廓和外形进行变换的运算(其实也可以运用于灰度图像,会有插值效果,但应用最多的还是二值图,直接凸显形态特征)。这部分内容是传统识别算法的老朋友了,我们手动设计一些特征并用形态学操作进行增强,或削弱噪点。

  • 膨胀

将结构元素应用在某个像素时,把当前像素替换成结构元素覆盖位置上的最大值,对于二值图来说,就是变成255,扩张边缘。

  • 腐蚀

和膨胀相反,替换为覆盖位置的最小值,收缩边缘。

  • 开运算

先腐蚀再膨胀,消除高亮度的狭长区域(和你的结构元素设计有关)或高亮噪点。

  • 闭运算

和开运算相反,消除黑色空洞区域,或将断裂的轮廓闭合。

  • 顶帽

原图减去原图的开操作,消除亮度较低区域下的高亮区域。

  • 黑帽

和顶帽操作相似,是原图减去闭运算,可以得到高亮部分的较暗区域。

  • 形态学梯度

一般指的是膨胀和腐蚀图像的差值,前述两种“帽”操作也属于形态学梯度。

  • 统一接口

OpenCV的形态学操作api为morphologyEX(),提供了上述所有操作的接口。

  • 自定义结构元素

OpenCV提供了getStructuringElement()函数方便我们自定义结构元素,你也可以使用Mat制作更复杂的结构元素。

关于这部分内容,网络上和教材中有大量的介绍,在此不过多赘述。关键是理解形态学运算的原理, 知道算法是怎么运行的即可。



5.4.目标跟踪

为了简化分析和学习过程,本文所指的目标跟踪皆为单目标跟踪。其任务是在给定某图像序列初始帧中的目标位置和范围(cx,cy,h,w)的情况下,预测后续帧中该目标(cx,cy,h,w)。

5.4.1. 基本实现方法和原理

目标跟踪的pipline大致如下:

  1. 提取初始目标的特征
  2. 根据输入的初始位置与范围在之后的帧中依据运动模型和其他准则生成诸多候选框
  3. 根据某种准则由此目标的特征计算候选框的得分
  4. 选择一个得分最高的候选框或对多个候选框进行信息融合进而得到最为下一帧的预测目标

从以上的流程我们可以把目标跟踪划分为几个不同的研究领域:候选区域的提案方式(用什么运动模型?)、初始目标和候选目标的特征提取、候选目标的评分机制、模型更新(如何在运行过程中学习对象的运动模型从而优化后续的跟踪效果)、融合预测值从而获得更好的结果。在现行的几种模型中,我们常根据pipelin中1、3的不同把目标跟踪算法分为生成式和判别式。

  • 生成式的算法通过提取初始目标的特征构建一个表示模型用于描述目标的特征,在之后帧中用此表示模型来分析候选框,和初始目标进行对比从而得到得分。其本质上是一种模板匹配,在提取初始目标特征的时候得到一个用于匹配的模板,随后在候选区域上应用该模板进行匹配。每次匹配结束之后,提取特征生成新的模板。也就是说,特征提取仅限于目标的ROI。

  • 判别式模型则是将跟踪问题看作分类/回归问题,希望习得一个判断函数(映射),将候选框投入此映射得到分类或得分。其本质上是利用初始帧的信息训练一个分类器/回归器,然后在之后的跟踪过程里不断更新分类器/回归器的参数。生成式和判别式的参数更新一般都是递归式地(熟悉的感觉有没有!递归滤波器它又回来了!实际上在早期的跟踪任务中,卡尔曼滤波器和粒子滤波器等常用于生成式模型中特征的匹配和预测框生成)。因此,判别式模型会对非目标区域和目标ROI都进行特征提取,并生成一个分类器,训练目的是让目标ROI区域的响应最大/判为正类,其他背景区域的响应小/判定为负类。

分类方式类似于机器学习or深度学习的两大类问题:和,分别代表了生成模型和判别模型。对于这里的跟踪算法,已知上一帧的目标的特征x,创建一个模型,使得下次和则为给定上一帧图像的ROI和这一帧图像ROI一起作为x,输出y为yes/no或者一个0-1的分数。

在 5.0 部分我们已经介绍了为什么要使用目标跟踪算法,其高速的特性使得我们可以降低算力消耗,甚至可以在用目标检测得到第一帧后同时运行目标跟踪和目标检测算法并对两者的结果进行融合(目标检测依赖gpu算力而目标跟踪除cnn方法外主要使用cpu,实测cpu在运行视觉算法的时候占用率大概只有20%-30%,不过这貌似和之后的CNN-Based方法有点套娃了),大大提高置信度。在实际工业应用中,目标跟踪、目标检测和重识别也常常配合使用。

目标跟踪示例,可以看到多种颜色的候选框,左上角是从开始跟踪到当前时刻经过的帧数

本节我们主要介绍几种经典的跟踪方法。


5.4.2. 光流法(Optic Flow)

光流法是典型的生成式模型。顾名思义,光流要追踪的就是“光的流动”,即计算或估计两帧连续图像中相同点的移动。稀疏光流将会计算视频流中数个关键点在之后帧的移动轨迹;而稠密光流则是对图像上的所有像素都计算移动距离,这将会在二维平面上得到一个向量场。

稠密光流和稀疏光流的对比 -图源知乎https://zhuanlan.zhihu.com/p/74460341
  • 传统的算法包括LK(Luca-Kanade,经典的稀疏光流算法)和Farneback(稠密光流,这个算法对数学功底尤其是线性代数的要求比较高,是作者的博士论文,请参考Two-Frame Motion Estimation Based on Polynomial Expansion)以及他们的变种和改进,新兴的算法包括基于深度学习的光流估计和基于运动场能量的算法(最小作用量?)。

两个传统算法都有一个假设:物体亮度不变。以稀疏光流为例,我们希望特征点在两帧之间的亮度不变,这样才能较好地追踪它们的运动,在当前帧和之后帧的像素点间建立联系。算法的具体流程和推导可以参考这两篇文章,已经解释的非常好了:

  1. 稀疏光流KLT - 知乎 (zhihu.com)
  2. 光流估计 - 知乎 (zhihu.com)
  • 而使用CNN进行光流估计的开山之作是FLowNet,它针对的场景是稠密光流。相信经过之前的学习,你应该已经很熟悉这类算法的流程了:我们希望网络的输出是一个二维向量场——那么输出的尺寸和输入相同(这似乎和分割任务完全一致!),并且每个像素位置上应该有两个值,这样就能代表一个二维向量,表示两帧之间该点的动向。那么网络的结构就呼之欲出了,我们需要一个全卷积网络或反卷积网络,形如曾经红极一时的U-Net。其实也可以看作一个Encoder-Decoder的结构。

不过,应该如何进行标注?难道要人眼逐帧辨别像素的运动?这个工作量有些太吓人了。FlowNet的作者非常有创造力,他将一些其他物体以图层的方式叠加到背景下,作为第一帧图像,随后对这些物体进行“稍稍”的仿射变换以形成轻微变形和位移并再叠加到相同的背景下,得到第二张图像:

右侧图像显示了两帧之间移动的像素点,不同颜色表示位移矢量的方向,颜色深浅表示矢量大小

这样就可以在几乎无人工成本的情况下得到大量的label data,作者还用这种方法制作了一个数据集,就叫“flyingchairs”。

后续FlowNet还推出了2.0版本,主要是加入了级联结构,实现光流估计的coarse2fine过程,预测得结果变得更加精确并且在速度远超传统算法的情况下精度实现了SOTA。

  • 更多关于深度光流估计的论文,可以参考这个系列文章:2019-2020基于深度学习的光流技术调研 - 知乎 (zhihu.com)

  • 光流法有一些较强的假设,在自动曝光的相机配置下有时会出现异常。当场景明暗变化剧烈或进入黑暗区域、过曝区域,光流法都会丢失目标(当然一些其他方法也有类似的困难)。相比之后在传统手工特征中要介绍的特征点法,光流法对于场景中的纹理和细节要求不高。

对于RoboMaster比赛来说,稠密光流没有很大的用处,我们不需要知道图像上每一个点的运动趋势,这毫无意义(当然哨兵除外,稠密光流与直接法是视觉SLAM中一种很好的建图方法)。稀疏光流对于我们来说更有价值,如果对装甲板进行跟踪,我们可以直接跟踪装甲板灯条的四个角点,在之后帧中都不需要进行后处理就可以直接进行PnP解算了。光流不仅可以匹配两帧连续图像中的不同点,还可以一定程度上代表图像中物体的运动趋势,所以在视频分析中常常作为比原始视频更高级的特征或辅助特征,输入神经网络中。


5.4.3. 以KCF为例的相关滤波(kernel correlation fitlering)

KCF是典型的判别式方法。而相关滤波方法的始祖是MOSSE(minimum output sum of squared error filter),它首次将信号和通信领域的方法引入目标跟踪。相关滤波是用于衡量两个信号相似程度的方法,因此从名字中就能看出,它通过对初始目标在线训练一个滤波器来检测候选区域和初始目标的相关性,响应最大的即是当前帧要跟踪的目标。相关滤波类算法的最大优点就是其无可比拟的速度(150FPS+),在KCF横空出世之前,跟踪算法要么勉强达到精度需求但是根本无法在线运行,要么准确度低得令人无法接受。

KCF对数学知识的要求较高,如果你认为自己的数学功底还不错,或曾系统地学习过矩阵分析、信号处理、复分析和傅里叶分析,那么你可以挑战一下论文原文的阅读。虽然其背后的知识较为复杂,但是代码实现起来却出奇的简洁清楚,可能这就是数学的魅力吧。

这一小节首先总览一下相关滤波方法,然后按论文的结构,从每一个组成block和总体的pipeline对KCF进行介绍。

5.4.3.1. 相关滤波

(有点难啊不想写)


5.4.4. NN-Based

这部分内容在我们前面介绍光流法的文章中已经看到了一些端倪,根据前面的知识我们应该可以轻松地得到一种基于CNN的跟踪器:输入前一帧图像和初始的位置还有当前帧图像,输出即为预测的bbox。当然,这只是CNN-based方法其中的一种。聪明的你应该可以分析得到,刚刚提到的pipeline代表了一种判别式的方法的端到端框架,我们利用深度网络提取深度特征生成滤波器(卷积核),然后再次利用深层的网络作为分类器和回归器对浅层网络滤得的特征进行判别并输出当前帧中的目标框。

一定要区分生成式模型和判别式模型的不同,如早期目标检测算法中常用的HOG/SIFT/ORB/LBP/SURF等就是先对模板/原图进行特征点提取和并根据特征点周围的特征制作特征描述子,随后在后续帧图像中再次运行相同的流程找到特征点及其对应的描述子随后对两帧之间的特征点根据描述子的相似程度进行匹配。因此生成模型的核心是特征提取+特征匹配,而判别模型则是对可能的候选目标进行分类或回归(打分)。在 5.5 中我们会以一种经典的特征点匹配算法为中心,详细介绍生成式模型和传统的目标检测以及其他拥有良好的尺度/亮度/旋转/平移不变性的特征。

在前面我们已经见识到网络强大的特征提取和表示以及对非线性映射的拟合能力,下面给出在目标跟踪中能用能用网络替代的模块(包括全网络的架构):

5.4.4.1. 特征提取模块

传统跟踪算法中直接使用像素级特征的大有人在,也有使用HoG、HoColor这些稍稍高级一些的特征。他们的优点就是计算起来非常快,缺点也很明显:表达能力相对较差。而深度特征的表示能力在这几年来的飞跃式提升是大家有目共睹的,自然而然我们可以使用CNN对初始帧进行处理得到feature map。有改进者结合了深度特征和简单的梯度/颜色特征,加入一个置信度模块:对于简单的样本就直接利用速度最快的HOG等特征进行初步筛选,如果一看就是背景直接跳过即可;若出现置信度不高的情况,再进一步使用CNN进行提取。这也是很多级联方法的惯用trick,如经典的Haar人脸检测器就是利用这种方法避免让所有输入都通过整个流程。

HCF(hierarchical)是典型的CNN特征提取器+correlation filter方法

特征提取器有两种方法,离线和在线。离线的提取器一般就是在ImageNet等训练集上进行预训练,然后直接用于生成深度特征。而在线学习则是和其他传统方法一样,从前一帧的GT附近截取数个ROI并在其上进行在线训练。也可以采取中庸之道,使用pre-train+在线fine-tune的方法。

cnn的卷积操作应当为典型的空域卷积,前面介绍KCF的时候我们就知道频域中的许多性质可以帮助我们加速运算——卷积在频域变成了简单的乘法。其中,在线方法可以在频域利用傅里叶变换的性质(复矩阵对角化/循环矩阵对角化/FFT等)加速提取特征和训练。

针对目标尺度和角度的变化,可以采用多尺度训练/FPN等目标检测中常用的技巧来进行针对性优化。至于亮度/对比度等不变性,对于网络来说都是小菜一碟,只需要在训练时装入经过数据增强的样本即可。

5.4.4.2. 候选区域生成

不同的跟踪方法有不同的候选区域生成策略,最简单粗暴的方法就是对整张图片都运行(这就没有任何”策略“可言了)。这一个步骤也可以对应到目标检测网络中的回归器。

相关滤波方法一般是采样GT周围的ROI(例如KCF就是把GT扩展2.5倍,利用循环矩阵在频域良好的性质进行计算),生成式方法则是寻找数个特征点/群,将那些有较多成对匹配点的区域作为ROI。之前在 5.2.6.6 中介绍vision attention的时候提到有通道域和空间域的注意力,那么网络可以利用attention机制对ROI中的特定区域进行筛选,减小搜索空间。

再进一步就可以利用 5.3.2 中介绍的运动学建模和运动预测方法,利用历史的时间序列数据建模物体的运动,KF、PF、IMM等方法在目标跟踪中都是可用的,不过由于深度信息的缺失(仅仅在二维平面上估计运动,丢失了轴的信息),处于复杂运动状态下的物体很难用线性预测器预测。因此可以使用网络替代这些组件中的部分模块,用网络的输出提供协方差/噪声建模等信息,更好的建模非线性因素。

既然是序列信息,RNN/LSTM/Transformer必然也是可用的,并且可以增加时间注意力机制(对比通道域和空间域,在RNN中是另一个维度),建模历史不同时刻帧对于当前跟踪的重要性。

利用LSTM建模物体的运动从而生成ROI
5.4.4.3. 分类器/回归器(fine-tune)

之前介绍的判别式算法,其滤波器(可以看作分类器)都是在线训练的,而对于由神经网络构造的判别器和回归其,更合适的方法是离线训练。对于分类器,较好的方法是参照人脸识别中常用的few-shot learning,设计一个判别函数,利用triplet损失或Siamese网络来进行训练。不了解这两者的同学可以参考吴恩达深度学习第四章-Siamese网络以及吴恩达深度学习第四章-Triplet损失。

对于Siamese(一般称作孪生网络),我们可以并行训练两个网络,分别用于提取gt和候选区域的特征。

一般来说就是在提取模块上再进一步,其实CNN的特征提取和映射很难分清界限,你可以认为前几层是在提取,后基层是在映射,这里我们暂且把他们直接拆分为2个部分即提取+映射。

然后,对cnn输出的最后一层特征进行embedding(再提一嘴,也可以直接当作最后输出的特征就已经是嵌入在某个空间中了),并利用一些尺度度量的手段如欧式距离、海明距离等等来计算两个向量的距离/估算相似度。最后输出一个分数,高分代表相似,低分代表不确定/不相似。许多时候我们也不需要直接训练两个网络,而是直接训练一个网络并在另一个网络上共享参数,即下图中的”share“所示的虚线。

一个孪生网络的示例,图源李宏毅机器学习课程sliding

在triplet loss中,我们期望相同的同类别样本的距离尽量小同时希望不同类别样本的距离尽量大,因此设计的损失函数如下:

其中是anchor即参考样本(正类),是正样本,是负样本,就是所有成对的训练资料(在训练中可以取任一样本作为anchor,然后取另一和anchor同类的样本作为positive,最后取任意一个其他类的样本作为negative)。加入的是一个极小的值用于正则化,防止网络”走捷径“:在最小化同类样本距离的同时却使得正样本和负样本的距离很接近零,即前一项损失几乎是零,第二项只要稍稍大于零就完成了训练目标。最后,网络输入两张图像(gt和ROI),输出他们之间的距离,根据这个距离进行判别即可。

对于回归器而言,其目的是输出当前帧跟踪框的位置。但是上面介绍的分类器都是运行在某个区域(ROI)上的,意思是我们要先确定用于提取特征和匹配的框,然后再运行分类器,然而一般可能的预测区域都是通过 5.4.2.2 中介绍的方法得到的。那么在这里,我们则是执行类似目标检测网络中two-stage的方法,候选区域生成作为“RPN”,后面再跟一个回归器用于对候选框的位置进行微调。不过对于判别式的跟踪而言,倘若回归器生成的bbox太过于贴合目标,有可能还会适得其反:不同于目标检测,目标跟踪中背景信息也是很重要的特征(特别是对于只提取上一帧 bbox中特征的方法 ),tracker应当很好地分离背景和前景,并且获取他们之间的差异,倘若完全没有提取到背景信息跟踪器将会很难区分下一帧的目标和背景。

5.4.4.4. 端到端方法
  • 学习过之前目标检测的同学应该对这个词语非常熟悉了。我们之所以这么执着于end2end,正是因为其框架和pipeline的简洁性和同一性。上图所示的就是SiamFC,一个端到端的跟踪网络。其实本质上是一个相关滤波的框架,只不过所有模块都使用CNN实现:上方的分支是特征提取器,将前一帧的bbox送入网络进行卷积运算得到6x6x128的滤波器,下方则是输入当前帧整张图片,还是通过提取特征(参数共享的思路),得到22x22x128的feature map;最后用第一个分支得到的滤波器对feature map进行卷积(也可以说是相关运算)运算,可以得到一个heat map代表原图上各个位置的响应,自然响应最大的地方就是我们要找的目标了。

参数共享和第一个分支中利用生成的特征作为判别式算法中的filter的思想是这个网络的最大特点。在SiamFC之后有大量工作顺着其思路进行扩展,但是都是基于相关滤波展开的,比如添加attention、增加响应之间的交互、更好的特征提取方法等。

  • 另一种思路是由SORT(simple online and realtime tracking)提出的tracking by detection,字如其名,利用目标检测实现跟踪。不过这个方法最开始其实算不上end2end。其框架包括一个用KF实现的预测器、一个目标检测网络和用匈牙利算法实现的匹配模块。这其实是为了解决多对多跟踪即MOT而提出的方法,不过其思路很值得借鉴故在此简单介绍,对于单目标跟踪同样可以使用该思路。

检测过程如下:首先用第一帧的目标初始化KF,随后在第二帧中通过检测网络得到当前帧中的数个目标,这时候就需要用匈牙利算法对检测网络输出的bbox和前一帧的gt基于cost最小的原则进行一对一匹配(可以参考5.2.6.8 中DETR的相关部分);完成匹配后利用两帧的差异更新滤波器的参数,至此完成一轮迭代。

显然,SORT的性能非常依赖检测网络的准确度,若检测框数量多余或少于前一帧,就需要进行特殊处理。而且其预测方法也过于简单,没有增加更多的先验,导致对快速物体的跟踪效果不是特别好。因此后续提出是DeepSORT对SORT进行了改进,引入了当前帧和前一帧的IoU、对物体运动更详细的建模、用于衡量两帧目标相似度的Siamese网络等方法改善SORT的性能。

DeepSORT的检测pipeline

当然DeepSORT也还是有很多问题,后续也提出了很多针对性的改进:

  1. Re-ID网络(即Siamese+KM)可以直接和检测网络融合,做到真正的端到端
  2. 若一个物体在很长一段时间都被很好的catching,应该可以给予其更大的权重
  3. 不单单使用tracking by detection的方法,追踪器可以弥补检测器的漏检
  4. 使用除KF外更好的运动模型,参见 5.3.2 的运动建模部分

对于RM比赛,由于目标较为稀疏并且我们的相机采样率极高,甚至可以直接使用tracking by IOU的方法,直接使用两帧之间目标的IOU来判断当前检测框内的物体是否是同一物体。

现在主流的、SOTA的目标跟踪方法基本是使用深度学习的tracking by detection的方法,如byte-track,DeepSORT等。引入时间序列分析也是提升跟踪性能的有力方法,只不过额外的开销可能导致无法实时(一些使用time-attention的transformer等)。感兴趣的可以自行搜索论文和相关领域的进展。



5.5.参数自适应方法和稳健特征

不论是传统的图像增强算法还是神经网络,总有一些参数是需要手工调整的,而这些参数得设置是否恰当,往往会影响整个算法的表现。手动设置的参数往往是针对部分情景和特定目标的,若环境或任务发生改变,之前的参数很可能会失效。这也是为什么我们都追求一种能够适应尽可能多的场景的参数设定方法,亦或是尝试元学习以指导参数的自动选取,减少人为干预。

不过即使再怎么追求hand-crafted free,我们似乎无法逃脱设置参数的魔咒——用于选取参数的算法在另一种角度来看仍然是一种参数,也就是深度学习中常说的超参数,而超参数之上亦有参数,这就是meta learning。如果我们希望不要人为设定meta learning的参数呢?那便有了meta-meta learning,我们可以不断增加meta的前缀直至无穷无尽... 即使是下面介绍的自适应方法本身,也算是一种prior knowledge,或者约束。

因此,参数自适应的方法就显得尤为重要了。我们希望能够在分割图像的时候自动选取合适的阈值,希望在分类图像的时候得到错误率最小的置信度阈值,希望在获取图像特征的时候划分得到最富有语义信息的部分······

本节将会介绍传统的图像处理中常见的参数自动选取和阈值自适应技术,以及运用了这些方法的特征点检测/目标i检测算法如LBP/FAST/SURF/ORB/SIFT/HOG。此节的内容也会更贴近实际应用,会稍微涉及一些比赛中的实用coding方法。

理论上来说,这一部分应该放在神经网络和CNN之前,因为CNN其实算是借鉴传统方法发展起来的,其特征表和提取都算是传统方法的泛化结果。5.5 也是许多同学在学习CV的时候常常忽略的一环,常常是只学习了DNN和CNN,跑了几个数据集和经典网络就认为自己掌握了计算机视觉的基础知识。实际上这部分内容才是CV的基石之一。这部分内容相比“一上来就yolo”更加理论、更加“数学”一些,也许可以为你的计算机视觉学习提供一些更有力的直觉。


5.5.1. 阈值自适应与不变特征

对于黑白图像(彩色图像同理),亮度/像素点的值是最基本的特征。一般的计算机视觉任务都需要对输入进行预处理如标准化、提亮/暗化、锐化/模糊等,所以首先来看看像素级特征和形态学特征。

5.5.1.1. 亮度自适应

由于相机曝光/光圈/外界光线环境的变化,输入的图像常常有着不同的视野亮度。倘若我们希望在预处理中对所有图像进行整体提亮/暗化以达到相同的亮度水平,显然设置一两个一成不变的参数是不够的。不同的 图像要达到相同的 效果,就提醒我们应当使用自适应技术。 最简单的方法是对所有像素值求平均,然后使得所有图像的亮度均值相同,这样在稍后的处理中就可以用一个相同的参数对付所有图像了。

不过增减均值有时候并不是最好的选择,若我们希望在稍后得到图像的形态学特征,常常需要进行二值化处理,这时候阈值的选取就非常关键了,如果部分图像整张都偏暗或偏亮,阈值稍大可能导致特征被删除,太小又可能留下过多不需要的形状;特别有时候我们希望增大图像的对比图突出边缘和纹路等细节,那可以使用直方图均衡化的技巧,其将一张分布不均匀(常常是有大块明暗区域)的图像的直方图变得更接近均匀分布。这能够大大提升图像的对比度,使得明暗的差异增大,让我们更好地分离需要的部分。其基本思想是对在直方图中密度大的区域进行展宽,而对密度小灰度级进行缩减。

Histogram equalization的例子,图源知乎https://zhuanlan.zhihu.com/p/29772177

经过HE的操作,原本不明显的边缘和纹理也都变得更清晰了,最重要的是现在可以方便的选择阈值进行二值化操作了。

根据预处理目的的不同,我们也可以把没有达到均值部分的像素值与均值的差看作是某种损失,若相差越大就要施以更大的”惩罚“,即在稍后将此像素值增大/缩小得更多一些。比如L1 loss或L2 loss都是可用的,也可以采用一些更平滑的曲线。


5.5.1.2. 局部阈值自适应

前述的亮度自适应方法在一定程度上可以解决图像亮度变化造成的影响,但是考虑这种情况:我们分割出一张报纸上的文字(黑色字体),但是报纸的一侧暴露在阳光下,而另一侧则在黑暗中。那么即使进行了直方图均衡化,由于其亮度的”两极分化“,高亮和强暗区域仍然是糊成一团。这时候就应该选用局部阈值或局部直方图的方法了,其本质是使用比全局统计特征(即直方图)更精细的分类。

使用局部动态阈值和单一阈值的对比

可以看到,采用动态阈值后很好地解决了这个问题。其实现过程也应该很容易想到:使用一个mean slide window或者gaussian slide Window处理整个图像,对部分区域统计均值/加权均值并由此设置均值的1/5或1/4(这个值仍然是人为设定的,取决于对细节的取舍,高斯窗则是做加权均值统计)作为阈值,便可以得到上面的效果了。

使用局部直方图均衡化对比全局均衡化,中间下方的纹路差别更大了

局部直方图均衡化可以取得的效果和动态阈值相仿,不过和全局直方图均衡化希望得到形态学特征相比其更注重的是纹理和细节:对比上图中的global equlize和local equalize的结果就可以看到,若只希望得到外形轮廓即形态学特征,应该使用全局直方图均衡化,中间下方的沙坑轮廓非常明显;而要获取所有纹理特征,则应该选用局部直方图均衡化。

其步骤也同样非常简单,取滑动窗口和一定的步长(你看吧,滑动窗口的大小的步长以及是否加padding仍然是需要人为设定的超参数,使用不同的参数可以得到不同的效果,可以自己想想试试看),对滑窗内进行均衡化即可。

动态阈值分割前景背景的经典方法还包括大津法,其思路为使得分割后的前景与背景类间方差最大,因此算法就是要找到一个合适的阈值以实现前述目标,一般用暴力搜索即可,从0-255遍历阈值得到类间方差的最大值。大津法对分类边缘的处理效果较好,因为一旦有伪正类或伪负类被划入对应类别,类间方差都会极大地减小,这使得损失函数对于分类错误的敏感性很高。

OpenCV对上两者都提供了接口,被统一写入cv::Threshold()内,通过传递不同的阈值类型参数就可以选用自适应阈值或大津法。


5.5.1.3. 双边滤波

我们常使用一些降噪滤波器如高斯、中值、均值等对图像进行平滑处理以去除噪声,但是它们都有一个缺点,就是容易损伤高频信息,随机噪声、椒盐噪声等都属于高频噪声,而前述的几个滤波器本质上都是low pass filter,所以高频纹理和边缘的损伤是在所难免的。那么我们能否开发出一种参数自适应的滤波器以避开边缘,只在平坦区域寻找噪声?这就是双边滤波的思想。它同时考虑空间域(xy两个维度)和值域(像素值的高低,即图像的"第三维"),力求减轻对边缘的模糊。

高斯滤波器的计算方式如下:

其中高斯模板如下:
而是归一化系数,即每个位置上的高斯权重之和,除以此系数保证所有位置的权重和为1。直观上理解,高斯滤波器以距采样点的距离的函数为加权系数,离得越远的像素贡献越小,越近的像素贡献越大。但双边滤波同时考虑值域的信息,在滤波的时候,还要看看窗口内的像素和中心像素是否相似,因此其计算方式如下:
其中为:
代表了中央像素和窗内其他像素的相似程度,即考虑了值域上的距离。和当前像素越相似,那么这个像素就会有越大的权重。这样,在边缘的时候,处于边缘一侧的像素都和中心像素相似,因此有较大的贡献;而另一侧因为在像素值域上相差甚远,因此权重很小,基本不会影响到滤波后的像素值。当然,同样要进行权重归一化,只需要对的每一项都乘上即可:
看看使用不同的的时候,滤波的效果:

纵向变化的是值域方差,越往下越大;横向变化的是空间域方差,越往右越大

不过由于值域项的调制作用,对噪声的滤除效果必然有所下降(噪声一般是平坦区域的一个凸出像素,以它为参考点滤波的时候,周围的像素肯定是和噪声不太一样的,因此权重减小)。为此,你可以添加更多的非线性项,比如规定当中间的像素和周围都不同(都大/小)时,赋予中间点更小的权重等,这自然会增加运算量。效果更好的自适应去噪方法包括NL-means和它的改进BM3D,它们都拥有非常优良的去噪效果,不仅考虑了邻域,还将图像中其他与当前滤波的patch(窗口)也考虑进来,分别从频域和空域对噪声进行估计。显然,在得到更好的滤除效果和保留细节的同时,你总得付出更大的代价:时间开销和空间开销。No Free Lunch Law再次降临!

图像复原、图像降噪、多光谱融合和超分辨率等领域,依托传统图像处理方法的同时引入深度学习,也是CV的一个重要但不那么卷(好像也有点卷了?)的方向。2016/17年一篇仅使用简单的CNN+encode-decoder结构就能去除图像噪声的论文(忘记叫什么了,知道的可以联系下笔者)引爆全场,只不过是让一张图片通过一个bottleneck一样的结构(就是编码器-解码器的架构)其上的噪声就消失的无影无踪。至此,类似U-Net的结构便在上述领域内层出不穷,各种编码方式、前处理(一般是从频域、小波变换等入手)、跨域融合就被加入进来,DNN逐渐就占据了这些方向的SOTA宝座。感兴趣的同学可以进一步学习相关内容。


5.5.1.4. 形态不变特征

注意区别这里介绍的是形态学特征(morphology feature,指的是图像中具有一定形状的图形)而非前述的一般特征,不过形态学特征是作为一般特征的子集。请结合具体的例子进行理解,如RoboMaster比赛中的装甲板识别任务。形态学操作和相关知识可以参照《学习OpenCV3》中的相关内容。

在处理形态学特征的时候,我们可能希望寻找一些由不同区块的轮廓组成的目标,在比赛中常见的即装甲板灯条的匹配,其在正常曝光下从远处看是两个对称的梯形,而我们一般将其视作一个轮廓进行处理。当我们希望匹配处于同一个物体上的两个或多个形态特征(对于装甲板识别就是找到处于同一块装甲板上的灯条),应该寻找对旋转/尺度/平移都有不变性的特征。

以灯条匹配为例,装甲板距离相机的距离会导致灯条长度的改变,但是不论距离远近,处于同一块装甲板上的灯条的长度比始终应当保持不变(根据 4.1 介绍的相机小孔成像模型),这就是一个尺度不变特征,显然灯条的宽度比也同样是一个很好的衡量指标。而由长宽相乘得到的面积,也可以作为一个筛选/匹配的准则,不过应该使用面积比而不是面积(面积比和长/宽比似乎在一定程度上有一些耦合,不过可以对筛选顺序进行合理排序,面积比可能更容易清理哪些非灯条的目标,在 6.1.1 我们会再进行介绍)。

至于平移不变性,这对于非定位和non-global-based的特征来说都是可以得到保证的,因为要利用形态学特征进行匹配,首先就应该定位这些轮廓。

最后是旋转不变性,角度是候选的常客,只要是目标在空间中进行的是刚体变换,变换后的特征仍然会服从角度一致。但是对于out-of-plane的投影变换(相机成像)来说,物体的任何性质都不会被保留。但是对于畸变程度较小的镜头和真实尺度较小的目标,在数倍焦距外的相对角度被投影到2d平面后往往只会产生轻微的形变,因此之间的相对角度仍然可以作为一个筛选的标准,一般只要适当放宽筛选条件即可,这也适合对匹配目标进行初步筛选,删除那些极端值。

两个灯条的长宽比等几何性质拥有的尺度不变性特征

5.5.2. 统计特征和global-based方法

5.5.2.1. 基本模型

早期的计算机视觉任务(即深度网络出现之前)最关键的部分仍然是特征工程,即特征的提取和表示。以分类任务为例,我们需要找到一个好的特征表示,然后利用分类器(常常是SVM)进行类别区分。下面这张图很好地展现了识别的通用方法:

我们常常青睐具有不变性的特征

若能够提取到对于某些变换具有不变性的特征,对于识别的准确度将会大大提升。最常见的不变特征包括亮度不变、对比度不变、尺度不变、旋转不变。而对于在 5.1 中反复提及的特征的级别,我们可以提出以下的关系图:

“高级”的特征拥有更好的抽象表示

进一步抽象,我们还可以得到所谓的语义信息(semantic info),即用人类的知识描述的可以抽象出某种概念(concept)的特征。像素级的显然是最低的特征层级,通过对像素进行操作如卷积、聚类、形态学处理等,我们可以得到稍微高级一些的特征如纹理、边缘(可以回看介绍CNN的部分);再往上就是形状、轮廓;最后逐渐上升到概念如苹果、汽车等。越高级的特征往往拥有更好的可分性,因为它们只属于少数几个不同的类别,而不像低级特征一样,为许多不同类别的物体所共有。但是高级特征却缺乏细节信息,这也是CNN中常用FPN进行特征融合的原因。

在本节余下的内容就将介绍在CNN和Transformer出现之前我们是如何提取并表示特征然后由此实现各种计算机视觉的任务的。通过这部分的学习相信你也可以找到深度学习方法和传统方法之间的联系,并更好地融会贯通。

最原始的图像识别/分类方法就是对像素级的特征进行匹配,没错,我们说的就是模板匹配。首先获得一张目标的图像作为模板,然后将此模板在输入图像上进行滑动,并逐窗口计算相似性。模板匹配利用的就是最低级的像素级特,显然是没有任何的不变性的,即使目标和模板有微小的不同,计算得到的方差都大得惊人。

模板匹配中常用的计算匹配误差的算法

5.5.2.2. 直方图特征

那么如果将特征稍微提升一个层级试试呢?这就是下面要说的直方图描述子。我们将一幅图像分割为数个网格,并将网格中的像素取平均得到的值进行统计,得到一个直方图。得到的直方图可以看作是一个离散分布,亦或是一个特征向量:

不同的熊猫和考拉对应的直方图统计结果,可以看到它们各自的直方图大体上是一致的

这样,我们就可以利用SVM或与anchor vector(参考向量)的内积来获取衣服图像的分类了。从三张熊猫的图片中我们可以发下,虽然它的头发生了轻微的转动,但是统计直方图总体上保持一致,是中间低两边高的形式,计算两个直方图的KL散度或欧氏距离,显然也不会差太多。稍微想想你就能发现,直方图统计拥有良好的旋转不变性和较小的平移不变性。

若是有颜色的图片,只需要分别统计RGB的直方图,将三者concatenate就得到了其特征向量。若希望拥有光照不变性,最简单的方法就是对直方图进行归一化/标准化,因为直方图统计的恰恰是gloabal feature,因此全局的亮度变化对于归一化后的特征不会产生太大的影响。对于不同角度顺逆光拍摄的照片,还可以恰当地使用前面介绍的直方图均衡化方法提高不同图片之间的差异性,防止直方图出现聚集。

不过,这种特征的缺点也很明显:它显然还是太过低级了,并且要求在分类的时候目标恰好占满整个画幅以弥补其不具有尺幅不变性的缺点,若没有占满则会引入大量的无意义的背景信息(虽然有时候背景特征是很关键的一个部分,可以提供和前景有关的语义特征和潜在信息,这种低级的统计特征显然无法捕获到两者的关联)。


5.5.2.3. gradient-based feature

顺着直方图的思路,还可以使用什么稍微高级一点的统计特征?联想图像处理中常用于获取边缘和纹理的方法:求梯度——梯度直方图(histogram of gradient)的方法就呼之欲出了。我们可以首先提取全图的梯度得到梯度图(一般在此之前先做一次高斯滤波,因为微分算子对噪声非常敏感),然后将梯度的方向划分为若干个bins(如每隔45度一个格点),最后增加该位置梯度位于对应方向上bin的值。

考拉和熊猫的图像用sobel算子卷积后得到的响应

梯度直方图和一般的直方图的不同在于,梯度包含方向和模值,因此无法像像素值直方图一样用一维的向量来表示。当然,可以使用二维直方图进行统计,不过这会带来不同直方图间方差过大的问题。因此,我们采用这样的方法,将梯度的模值累加到属于其方向的bin内,得到梯度直方图。

例子如下,求梯度后得到方向(左侧)和模值(右侧),如左上角的(10,1),其落在0-40°的区间,故将其值累加到该bin中。

和像素值直方图一样,可以进行归一化,得到的向量就可以作为特征向量用于训练分类器了。虽然使用的特征比原来稍微高级了一些,但是缺少尺度不变性以及容易受到背景影响的缺点还是没有得到改善。那么应该如何区分背景信息和前景信息,提取最重要的特征呢?我们将在下章介绍local-distribution的时候再提。


5.5.2.4. PCA-face

这里再介绍一种经典的全局特征方法,PCA-face。看这个名字大概就能猜出他的过程:利用训练集中的数据进行主成分分析,然后将输入在top-k个分量上进行投影得到特征向量,并用该特征向量和不同的人脸计算内积选取距离最小的那个作为分类输出。不过稍有不同的是,它巧妙地将一张图分解为均值和残差部分,首先得到整个训练集的协方差矩阵,然后对协方差矩阵进行特征分解并取最大的k个特征向量作为basis,这样就可以把输入减去均值之后在“脸基”上进行投影得到输入图片的特征向量(注意此特征向量和特征分解中的特征向量含义不同)。

一张的图像本身作为二维输入,首先将其展平成一维的向量,计算所有张图像展开后的向量的均值,再将所有向量减去均值便得到单张图像的残差。随后使用个的向量计算维的协方差矩阵,即。因此特征向量仍然是一个长为的向量,将其切分成原来的形状就得到图像。

PCA-face的图示

特征分解得到的对应最大的那些特征值的向量就是那些最能凸显不同样本的特征,即数据集中的样本在该分量上的差异最大。因此选择这些向量作为face basis,使得输入在这些基上的投影有最大的差异程度,从而区分不同的人脸。这是将传统机器学习算法应用到cv的一个典型实例,其中蕴含的特征工程的核心思想非常值得学习。同样,这种方法本质上还是基于像素级特征实现的,其特征也没有什么不变性,只有当人脸的大小一致并且都处于图像的正中央才能取得比较好的效果,这里提到它主要是为了介绍其实现思路。


5.5.3. local-distribution方法

考虑到global-distribution的缺点如没有局部旋转不变性和平移不变性、对背景信息的处理也做得不够好,那我们就从局部统计入手,图片划分成数个不同的区域进行统计。

5.5.3.1. gradient histogram

一种直观的改进思路将全图划分成数个网格,在网格内分别统计梯度直方图,然后将统计得到的向量拼接成高维向量(相当于得到了一个全局描述子/全局特征),或利用集成的方法对每个向量训练一个分类器(boost、stacking、bag都是可用的)。这样就能在获取一些不变性,利用固定位置的局部信息组合来替代全局统计信息。当然,网格不宜太细否则容易受到噪声或个体特征的影响,使得分类器过拟合,太粗则和全局统计特征差别不大,也达不到效果:

还是前面的例子,每个网格中不同向量的长度代表在该方向bin内的累加值的大小

加入了网格之后,分类器对于局部旋转和位移的敏感性将会降低。既然可以用微分算子作为滤波器提取边缘,自然可以用其他的滤波器模板以产生更丰富的特征响应,下面就介绍使用了多种卷积核的GIST。

注意,local-distribution方法虽然使用了局部信息,但是最后生成的特征向量仍然是一个全局特征,因为这个特征描述的是整幅图像而非图像中的某个物体或目标。


5.5.3.2. GIST

gist意为要旨,要领,原文提出该方法的初衷就是进行场景分类,如森林、沙滩、街道、室内等。这样的方法显然需要一些全局特征对场景进行恰当地表示,同时为了保证特征更加鲁棒,采用局部信息聚合到全局的方式进行处理:

gist的图示

首先生成图像金字塔,原文选择的是4层;然后在每一层上都分别从六个方向应用Gabor滤波器,生成共24张响应图。对于每张相应图,都切割成4x4的网格,将网格内的响应值取平均,最终得到4*6*16(尺度*方向*网格)个维度的向量。对此向量运行PCA得到80维的全局特征表示。后处理我们都很熟悉了,用此特征训练一个分类器或直接计算和anchor vector的距离进行分类。

Gabor滤波器是一种小波滤波器,小波分析是傅里叶分析的进阶,同样在频域处理信号,在继承Fourier Transform优点的同时,其强大的局部分析建模能力和对非平稳信号的兼容性使得它大放异彩。这方面的内容需要较多的泛函分析、数值分析知识,有兴趣的同学可以自行查找资料。

在这里我们直接将Gabor滤波器视作和常见的卷积核一样即可,只不过它的特殊设计使他具有一些优秀的性质。对应到卷积神经网络,就相当于我们通过FPN生成了4个尺度的输出,并且每个尺度的feature map都有6个通道(使用了6组卷积核);而取平均的操作也非常的熟悉,这不就是average pooling吗!可见ConvNet的架构设计并非是无中生有从石头里蹦出来的,而是从这些传统的算法中扩展得到的。当然,在聚合特征的时候你也可以使用其他方法,比如之前介绍的直方图统计等,这都是对算法效率和精度的权衡。我们从特征工程的角度思考,采用直方图可以更好地建模局部信息,但稳定性会下降;采用窗口平均则可以拥有更好地抗噪性能,但是对局部信息的保留,就没有前者来得多了。

到这里,CNN似乎已经初具雏形,接下来要介绍的非常经典的HOG算法则更有卷积神经网络的影子(虽然它提出的时间比LeNet-5来得更迟),或者说,cv的发展都是同根同源的吧。


5.5.3.3. HOG

HOG的全称是Histogram of Oriental Gradients,即有向梯度直方图,论文发表时用于进行行人检测,直接上pipeline:

直接按照流程进行介绍。

  1. 先对输入进行gamma和RGB通道的归一化,5.1 中已经介绍过这种方法了。一方面可以减少噪声的干扰,另一方面提供一定的光照和颜色不变性。

  2. 梯度计算,用一种微分算子处理全图得到响应。若要加速可以使用分离的1*3和3*1算子。

  3. 获取梯度直方图。这里和前面介绍的gradient histogram一样,将图像划分为数个网格,原文选用的是8*8的像素为一个cell,对每个cell进行梯度直方图统计。这里HOG对梯度直方图进行了一个小的改进,原本是当某个梯度方向落在一个区间内,就将其统计到这个bin里,而HOG则是将直方图划分为9个方向(0-180,分度为20°,不考虑梯度的正负,因为作者发现考虑正负效果更差而且还更慢,就是这么耿直),若梯度落在两个刻度之间,就计算它在两个方向上的投影(是不是比之前直接丢到bin里合理多了)。

  4. 直方图归一化。以2x2的cell为一个block,对block内的直方图进行归一化,直接看下图:

计算示意图,绿色的为一个cell,蓝色为一个block;图源https://zhuanlan.zhihu.com/p/85829145

在提取了梯度特征后再次对局部的输出进行标准化,这不就是现在NN架构中常用的Instance Norm吗!因为梯度对局部对比度和亮度变化非常敏感,因此我们不仅在输入进行标准化,在得到梯度直方图后选择再进行一次标准化。同时,选取的block大小为2x2的cell(至于为什么选取2x2的cell作为一个block而不是其他值?作者尝试了2x2,3x3,4x4,5x5最后发现2x2的效果最好,有爆调参数那味了~),在一定程度上像是在进行特征聚合或特征交互,在此过程中,相邻cell的信息将会相互传递。

其实这个标准化应该叫做”conv instance norm“,常见的BN、LN、GN、IN的图示如下:

H,W这个维度就是一个输入,而上述的标准化又只在一个实例的局部进行,因此笔者个人认为这个名字取得应该还算比较合理。HOG采用的这种方法在AlexNet上被采用,但是由于后续推出的BN效果远超这样的conv instance norm,因此这种方法在今天已经销声匿迹。

  1. 收集HOG特征。直接把上一步的normalized slide window后得到的值全部拼接在一起(一个block里面4个cell,一个cell有9个bin,normalized window在滑动的时候不加padding),以上图为例,将整幅图像划分成cell的个数为 8x16,就是横向有 8个cell,纵向有 16个cell。每个 block有2x2个 cell的话,那么cell的个数为:(16-1)x(8-1)=105。即有 7个水平block和 15个竖直block。再将这 105个block合并,就得到了整个图像的特征描述符,长度为 105×36=3780(斜体来自和图源相同的知乎文章)。

  2. 利用训练集中产生的所有HOG特征向量(如上一步中得到的3780维向量)训练二分类SVM判断是否是行人。

再次利用批判性的思维审视一下HOG,可以发现它的架构比起GIST与ConvNet更加相似,并且还好巧不巧用上了一些后来才提出的trick。它也充分考虑了局部信息,利用gradient这个局部信息采集各个位置的特征逐步聚合,最后得到一个用于分类的向量。不过从第五和第六步就可以发现,要利用SVM进行分类,其窗口大小必须是一致的,要检测一幅图像中不同位置和尺度的目标,就要对应地使用不同尺度地滑窗进行逐位置检测,时空开销都非常大。而且,由于采用了微分算子提取特征,其对于前景和背景一视同仁,还是没有办法将他们分离,因此会产生非常多对于我们的分类任务来说是无用的特征。

有效利用背景信息的例子:

通过视角和几何这些上下文/背景缩小搜索空间的大小

使用概率的方法建模人出现的概率,对于没有任何约束的情况(左上),图中的每一个区域都需要搜索,对于使用了几何信息的情况(右上),因为行人一般拥有固定的宽高比,只需要关注那些呈现一定形状的区域(怎么判断这些“区域”?通过聚合纹理和轮廓可得边缘,或者显示的建模,如KNN进行分割和合并);加入视角作为约束条件(左下),就可以排除图像上方的检测框,人不可能飞在天上,并且还可以通过街景和道路的方向以及视角效应(在远处收敛)布置近大远小的windows;最后,在使用了视角和几何约束后,可以看见我们只需要布置少量的anchor即可。

由于我们没有关于待检测的物体会在何种尺度、出现的具体位置的先验知识,即使将特征抽取区域划分成一个个小块,也没有办法避免背景特征和无用特征被引入,而在 5.5.4 中,我们将会提及解决无用特征的方法,利用位于物体上的有效局部信息建模我们要检测的目标,而不是像前面这些方法一样对整张图片或窗口进行建模。若能像下图一样先确定用于抽取特征的一个patch,然后再在patch内对比特征,不就能有的放矢,大大提高效率和准确度了吗?

特征点检测和局部特征抽取的通用框架

5.5.3.4. ConvNet没有不变性!?

我们已经阐述过使用不变特征的好处,那么AlexNet在ImageNet竞赛上大杀四方的时候,是否就意味着DCNN已经拥有了我们所需要的一切invariant feature?让我们一一揭晓:

5.5.3.4.1. 旋转不变性

从直观上来说,CNN不具有旋转不变性,这是一个显而易见的结论,物体在旋转之后通过卷积核显然会得到不同的响应;之后更深层的卷积也会受到第一层输出的影响,因此最后生成的特征(比如softmax之前的一层,我们可以把最后一层的Fully connect+softmax当作分类器,前层都当作feature extractor)显然和旋转前不同。那么为什么我们的分类器和检测器仍然可以正确地完成分类、进行识别呢?这是因为我们的训练集中含有旋转过的数字,或我们使用了数据增强中的旋转操作。在《深度学习》(花书)中给出了一个数字识别的例子:

中间的一排表示用于匹配不同模式的filter

对于每一层的卷积操作,我们一般都拥有多组卷积核,而对于经过不同角度旋转的输入,对应不同通道的卷积核就能学习到这种旋转。如左上所示,左侧的卷积核对于向右倾斜45°的数字9会产生较大的响应,中间的则是对稍微向左侧倾斜的9有较大的响应;因此,在之后的最大池化操作中,这些响应都会被归结为”这里有一个9“而输出。那么这些不同的filter为什么会对不同的9有响应呢?那不就是训练出来的吗:我们的数据集中有不同转角的数字9,经过数轮反向传播之后不同通道不同组的卷积核自然而然就学习了某一种旋转模式。

当然,”旋转不变性“也不一定是从卷积层学到的,很可能是在全连接阶段,拥有不同转角的输入在经过非线性映射后投影到了特征空间中相近的位置也是一种可能的解释。最有力的证明就是进行实验,对每一层的feature map进行可视化,画出heat map。感兴趣的同学可以阅读本章末的paper。

但是不管怎么说,神经网络最后的输出应该是相近的,只是在单个卷积核提取特征的时候,由于使用的是卷积操作因此并不具有roration invariant特性;那数据增强和filter组合技算不算赋予了CNN以旋转不变性呢?这就是一个智者见智仁者见仁的问题了。

虽然一般我们没有显式地对物体的旋转进行建模,但是数据增强的确让网络在最后学习到了旋转不变性(data aug和filter group的组合),倘若我们为网络增加一些先验知识,并且直接建模旋转这一变量,网络是否更容易应对旋转的情况?这就是Spatial Transformer。这里的transformer不是自注意力的那个,代表的是变换的“那个” transform。

左侧为一般卷积,右侧为加入了ST模块之后的输出

Spatial Transformer会通过反向传播学习一组仿射变换的参数(平移+旋转)用于将图像中关键的部分“摆正”,在变换的过程中可能会出现拉伸或压缩,原文选择了双线性插值的方法。同时ST还可以切除没有用的背景信息,这又有一点注意力机制的意味(实际上也确实是一种隐式的注意力)。不过,我们还是需要不同转角的输入以训练ST module,否则单单输入垂直的目标,网络仍然无法学会这种变换。

此外,在 5.2.5.2 介绍池化层的时候,提到了pooling会使得当前层提取的特征拥有一定的旋转不变性和平移不变性,能够应对轻微的变化。然而有时候pooling反而会破坏这种不变性,我们稍后会介绍其中的缘由。

5.5.3.4.2. 尺度不变性

和旋转不变性一样,神经网络不具备尺度不变性的原因相同,除非使用数据增强。从最早的spatial pyramid pooling到最新的PAN,我们可以FPN的发展轨迹略知一二:如果CNN拥有scale invariant特征,为什么还要费尽心思地进行多尺度特征融合?

对不同尺度特征的感受能力还和卷积核的感受野有关,扩大感受野也是膨胀卷积(dilate conv,atrous conv)提出的目的。YOLOF(you only look one-level feature)就另辟蹊径,主张利用不同stride的膨胀卷积提取不同尺度的特征并进行融合,从而抛弃了FPN的设计。不过当下大热的Vision Transformer通常也不需要利用FPN建模不同尺度的特征,似乎自注意力能够自动聚焦到不同大小的目标和特征上,因此也引领了研究的新潮流。

YOLOF在四个res block里的3x3卷积采用了不同dilate rate的空洞卷积
5.5.3.4.4. 光照、对比度、颜色不变性

如果有,为什么还要用数据增强?但是,输入归一化(这也算是现代神经网络的标配了)和BN从原理上来说,似乎可以给CNN提供这些不变性。笔者猜测,之所以继续使用光照、对比度、颜色变化的数据增强,是因为这些变量常常是协变的,单单用BN很难解决这种像素级的耦合变化。

首先需要辨析一下不变、等变(同变?)的概念。不变表示不管输入如何变化输出始终一致;等变表示输入平移或发生某种变换,经过函数的输出同样会进行类似的变换。

5.5.3.4.3. 平移不变性

CNN是否具有平移不变性是这几个问题中最具有争议性的一个。数篇论文尝试通过理论分析、实验的方法解释这个问题,却得到迥异的结果,大家的争论主要集中在以下两点:

  1. CNN中的平移不变性到底代表什么?
  2. CNN是否具有上述的”那种平移不变性”?
移动轻微移动图像后,classifier输出的置信度变化剧烈

图源https://zhuanlan.zhihu.com/p/77543304

  • 纯卷积拥有“平移同变性”

卷积作为线性算子,对于图像这个线性输入进行操作显然具有“空间不变”性质(对应于LTI的时不变性质),因此,对于平移后的图像进行卷积,产生的响应只会在空间上平移对应的间隔。因此,若CNN全部由卷积构成而不加入pooling和fc,那么图像发生平移,最后输出的feature map也会平移相应的距离,但是响应大小仍然不变。

我们称这种性质为平移同变性,即输入图像的平移只改变响应的位置而不改变大小,但是显然输出特征是和原来不同的,两者相差一个位移。故不是”平移不变性“而是同变性。 那么拥有特征拥有平移同变性有什么好处?那就是更容易被分类器划分(虽然不如平移不变性那么容易,因为还需要平移一个距离,即增加bias)。

另外,前面在介绍旋转不变性的时候我们提到,pooling可以选择响应最大的特征。如果一组filter对应某个特征(当它遇到特定的边缘区域、角落或颜色斑点时将拥有很大的响应),那不管是该组的哪一个滤波器有激活,我们都可以检测到这个特征的出现(presence),一旦数个特征的组合出现并且有较大的响应值,我们就任务图像中存在这些特征的组合所对应的物体。若这种猜想成立,也同样佐证了平移同变性的重要性,并且说明严格的平移不变性并不是那么重要的。

显然,平移同变性只是让网络有能力学习到不同位置的物体特征,我们同样需要使用数据增强的方法让特征出现在feature map的每一个位置。

  • pooling是同变性消失的罪魁祸首

pooling确实可以提供一定的平移不变性(注意和同变性的差别),但是处在边界的时候,pooling的境地就很尴尬了:

上方为原始”图像”,下方为左移一个单位后的图像

图源https://zhuanlan.zhihu.com/p/77543304

不失一般性,我们使用一维的信号举例。上图中的shift0代表原图,其pooling步长为2,输出为0101;shift1则是信号左移一个单位,但是由于原本的响应恰好处于边缘处,导致输出变成了1111,完全丢失了平移不变性和同变性!当且仅当移动的距离等于pooling的采样步长时,平移同变性可以保持(相同的像素移动到了下一个pooling窗口,因此响应的位置改变但是值不改变)。

可是上面不是提到只要特征出现分类器就能正确分类 吗?那么pooling之后特征只是在位置上发生了偏移,并不影响我检测这个物体啊?还是拿上图的例子说明,假设网络认为0101代表一种特征组合的模式,即某种物体具有这样的特征组合,那经过shift后组合就变为了1111,很可能就是另一种模式了。不过好在神经网络的参数量较大,同时拥有多个通道用于组合不同的模式,有较强的抗噪能力,即使在一处发生了错位,也不会影响最终的结果(不过其实从本节开始处的动图来看,平移的影响还是比较大的)。

除了pooling,strid而不等于1的卷积同样会丢失特征的平移同变性,除非位移值和stride相同,原因和pooling相同;更不用说dilate conv、deformable conv这些不规则卷积了。

上面的解释都是定性的,若要从定量角度分析,则需要一个同变性丢失程度的度量:shiftability。在Making Convolutional Networks Shift-Invariant Again这篇论文中,作者采用余弦相似度来衡量平移不等的程度:

上式中的即卷积操作,则表示height和width方向上的移位。其实就是对比先移位后卷积和相反操作得到的feature map的相似性。既然有了度量方法,就该思考如何减小pooling带来的影响,请看下面。

  • 抗锯齿:anti-aliasing

喜欢玩游戏的同学可能在游戏设置中看到过这个调节选项,学习过数字图像处理/计算机图形学的同学多少都了解这个概念和采样定律的关系。这篇文章的思路非常有趣,从传统信号处理的角度入手解决pooling带来的问题。它将pooling视作下采样操作,根据Nyquist’s law,倘若采样得到的信号要保有原信号的所有信息,采样频率至少为原信号最高频的两倍。空间域的卷积等价于频域的乘法,这也是采样定律应用的核心。但max pooling显然不是卷积操作也不是线性算子,无法在频域被解析描述(或者说没有频谱/频谱没有函数式),故丢失了前述的平移同变性。

pooling的size和stride相同,也可以看作是一种欠采样的操作。欠采样会带来频谱混叠的问题,再次回到本节初的那张识别小鸟的动图我们会发现,置信度周期性地变大-变小,而离散频谱正是周期性的!

不同图像在对角线上移动的时候置信度的变化

经典的解决频谱混叠的方法就是加入抗混叠滤波器(低通滤波器),average pooling其实就引入了一定的模糊效果(均值滤波器的频率响应是一个低频处增益接近1,高频增益呈小幅度波浪状的图形,介绍中值滤波时曾给出其频率响应图,但由于max pooling的效果更好因此平均池化在现代CNN已经几乎被弃用。

传统的目标检测或图像分类在进行下采样的时候一般要先进行高斯模糊(高斯金字塔、拉普拉斯金字塔等多尺度特征提取方法)而在CNN中直接对feauture map进行低通滤波再最大池化,会混入过多无用信息和噪声带来的响应效果导致效果很差,因此作者提出一种巧妙的结合MaxBlurPool:

max pooling的等价表示,由于对stride=1的max响应进行的下采样导致aliasing
论文提出的MaxBlurPool

首先观察MaxPool的等价模式(结果等价但是运算量不等,没被采样到的点其实什么也没做),即使用stride=1的pooling得到输出,并在其上隔点采样得到最终的响应,如Baseline展示的那样,红色的响应直接被丢弃了,这也导致位移在不为stride的整数倍时没有连续性。既然在pooling之前增加抗混叠滤波器的效果不好,作者就尝试在隔点采样前利用一个低通滤波器对中间数据进行卷积进而将未采样到的响应信息聚合到采样点上,也可以看作是一种插值/平滑。经过Anti-aliased pooling后,feature map的cos similarity应该会有显著地提升,因为已经不像之前那样,当响应越过边界后就直接算入另一个像素,而是引入一个平滑的过渡。

通过实验证明,这一通操作确实有效,作者还对VGG-16的输出的平移不变性进行了可视化:

右键打开原图放大看,由蓝到红表示平移不变性的度量,即前述的cos similarity

这一小节没有严格的理论证明,为了做到更通俗易懂,大部分是笔者根据论文的公式/描述以及自己的理解编写的。这其实是关于DNN可解释性的一个非常有趣的话题。后续会从信号与系统/数字图像处理的角度用公式从采样定律和频域分析的角度严谨地推理关于“池化”为什么会丢失平移不变性这一问题,敬请期待。

  • 正确使用padding?

padding会影响网络的平移不变性/同变性是由SiamFC(在5.4.4.4 介绍过)提出的。

其实我也没有太理解SiamFC中不使用padding的原因,论文作者认为使用padding会给特征引入位置相关性(位置编码)导致网络会“走捷径”直接根据弱边缘响应而不是特征来确定目标位置。

不过我猜测对于基础的分类任务来说,padding确实会影响平移同变性。从单层padding可能看不出来,但是随着网络的加深,卷积核的感受野不断增大,会将添加的padding(尤其是补零padding)信息聚合过来,那么处在feature map边缘的响应和处在feature map中间的响应受到padding的影响就会有所不同(显然是边缘受到的影响更大),也许这就是SiamFC中想表达的意思?

另外,由于图像不是一个在二维平面上无限伸展的信号,并且不具有周期性,因此使用循环padding虽然似乎可以提供平移同变性,但实际上会导致网络把图像认为是一个上侧/下侧循环连接,左侧/右侧循环连接的圆柱体(球体?立方体?),引入本不存在的相关性(一幅图像的天空和草地不会在相反的方向连接在一起)。

本小节暂不用正文形式,以引用形式作为TODO暂放。待笔者进一步了解后修改,若你有自己的理解欢迎email/评论交流!

最后稍微总结一下,若将CNN的输入到输出视作一个end2end的问题(不看网络中的任何一个module而将其视作一个整体,包括数据增强等trick),考虑输入进行各种变换但是输出的标签保持不变,那么卷积神经网络应该是具有这些我们期望的性质的;但是单独分析网络中的某个结构如卷积层、池化、全连接层,那么就要仔细想想了。另外,再次注意甄别平移不变性和平移同变性的区别!

一些关于CNN的 invariance性质的 paper:

Stride and Translation Invariance in CNNs

CNN: Translation Equivariance and Invariance

On Translation Invariance in CNNs: Convolutional Layers Can Exploit Absolute Spatial Location

Tracking Translation Invariance in CNNs

Why do deep convolutional networks generalize so poorly to small image transformations?

Quantifying Translation-Invariance in Convolutional Neural Networks

选出的文章都是不支持CNN拥有平移不变性和平移同变性的,这里的不支持指的是在特征层面不具有平移不变性和同变性,而非最后输出的结果。


5.5.4. 特征点选取与特征描述子

和前面的全局统计特征不同,特征点检测+匹配方法则是另辟蹊径,只会选择那些很有“特点”的区域,然后用一些方法描述这个点,也就对应着特征点选择和建立特征描述子这两步。不论是global-distrubition还是local-distribution最终生成的都是一个全局描述符,而point-based的方法则是得到数个特征描述符,然后在两帧之间匹配特征点或在输入图像上匹配我们要寻找的物体的特征点。前言不必多,我们会一边介绍特征点检测的方法,一边将它与全局特征描述进行对比。

虽然 5.5.2 和 5.5.3 出现的算法当下都已经被卷积神经网络或者Transformer替代,但本节介绍的方法仍在CV中占用重要的地位,如图像拼接(OpenCV提供了stitching库)、图像检索(以图搜图或图像特征匹配)、相机标定以及双目相机极线匹配(OpenCV提供了ccalib3d库)、稠密3D重建(OpenCV提供了reconstruction库)、视觉SLAM中的路标定位与回环等都大量使用了特征点检测和匹配技术。

在OpenCV中利用 SURF对模板(左侧)和目标图片(右侧)进行特征点检测和匹配,每种颜色的线段代表一对匹配点
特征点检测在图像拼接中的应用,摄影爱好者就可能知道一些超分辨率图像是怎么来的

5.5.4 将特征点检测算法拆分为三个部分,分别是:

  1. 寻找特征点

  2. 对找到的特征点建立特征描述子(特征向量)

  3. 根据特征描述子进行匹配

因此,以SIFT算法为例,我们将它按照上述步骤拆分为空间极值检测(1),通过极值点(即上一步检测到的特征点)附近的梯度直方图建立特征描述子和描述子的主方向(2),最后将描述子两两进行主方向配对,通过者再计算特征向量的欧氏距离,取最小者为匹配成功(3)

需要注意的是,本节在介绍上述内容时,并没有按照某中特征点检测算法的各个步骤进行,而是力求从每种方法被提出的动机、能够解决的问题以及提出的时间顺序进行介绍。当然在最后的 5.5.6 我们会详细地介绍当今最常用的特征点检测算法ORB,按照论文编写的内容对其进行介绍。

现在也出现了一些深度特征点提取方法,字如其名,使用神经网络提取具有旋转不变性、尺度不变性等的特征点。在许多数据集上,深度特征点的效果已经超过了手工特征点,其最大的优势就是在保证不变性的同时最大程度泛化。另外,其召回率相比hand-crafted特征也有很大的提升,感兴趣可自行搜索论文。


5.5.4.1. 特征点选取

不像传统的方法直接对全局或在划分的网格内无差别提取特征,point-based的方法首先要选取一些很特殊的点,并以此作为参考进行特征提取(实际上是以点为中心的一小块区域)。这和基于CNN的目标检测中的anchor非常相似,我们只选取那些可能出现目标的区域进行特征提取(如Faster R-CNN只在region proposal上进行ROI Align,然后送入回归分支进行bbox精修),那么对于特征点选取来说也是一样的,这可以帮助我们大大减小搜索空间,让算法只提取那些“关键”的区域。

虽然这里的特征点和现在常说的人体关键点如17关节、5个面部关键点检测稍有不同。不过两者其实是有一定的相似性的,现代关键点检测网络的架构和目标检测基本一致,和point-based(5.3.3 介绍的CenterNet和CornerNet等)则是如出一辙,不过关键点检测网络有一些额外的设计和专用模块用于建模点与点之间的关系。关键点检测是根据图像特征之间的关系将关键点定位出来,在此过程中提炼了深层的语义信息如鼻头、嘴角等;而特征点检测则是一个筛选的过程,在大批量的候选目标中挑选出很独特的点并通过一些方法建立特征描述符。

面部50关键点检测,图源旷视face++开放平台

对于feature point detection(也称interest point detection),后续的pipeline一般是和模板图片进行对比、对特征点进行匹配,从而实现图像的对齐(图像拼接找连接点、相机标定找棋盘格角点)或分类任务(早期的计算机视觉)。而key point detection后续操作包括表情检测(利用关键点之间的位置关系,微笑时嘴角上扬/关键点之间的距离减小或增大)、动作捕捉(人体关节检测,可以用于AR/数字孪生等)。

通过选取一些很有特点的点,我们就可以避免引入过多的噪声和无用甚至会帮倒忙的背景信息,从而更有效地抽取对于分类/检测来说更有用的特征。

我们感兴趣的点一般具有以下特点:

  1. 纹理丰富(方便稍后建立特征描述子)
  2. 和其他点的差别大(便于区别不同的特征点)
  3. 在图像中能够被准确定位(角点、斑点,边缘和平坦区域不可取因为无法被定位,或者说有无穷个这样的点)
  4. 具有不变性(光照/对比度/旋转/平移/大小改变等变化不能够影响这个点的检出,即使上述条件变化,这个点仍然应该作为interest point被算法检测到)
5.5.4.1.1. Harris角点检测

Harris角点检测作为特征点筛选的一种,旨在选取独特的角点为后面的检测和匹配作铺垫。由于常常是数条(两条及以上以不同角度相交)线段的交点,角点一般在多个方向上都有剧烈的梯度变化,这便符合我们所要求的特殊的、能够区别于图像中其他区域的性质。

平坦区域的梯度对于任意方向的移动都没有很大的变化,边缘区域的梯度只在垂直边缘移动时发生变化,而角点则在任意方向上移动都会引起梯度变化

在图像这种离散空间,我们都是直接通过差分的方式求取梯度。首先定义为小窗口在x方向上移动u单位,y方向上移动v单位引起的响应变化:

是位于点处的滑窗权重函数,此窗口可以是权值全部为1的平均窗也可以是Gaussian等其他让窗口不同位置有不同权重的窗口,即表示窗口内每个位置的重要性不同。窗口大小也是可以自己定义的,如3x3、5x5等,窗口越大鲁棒性越好,但是检测出的精细度也会降低。 就是 处的像素值(实际上I表示的是intensity,即像素强度/密度,如果使用插值和平滑则可以达到更好粒度更细的检测效果,不过一般就取离散值)。这样看来,实际上就是在计算参考点移动后产生的梯度变化。那么为了让窗口移动时有最大的变化即梯度变化最大(先不管方向的事,下面会提到),我们就要最大化下式:
直接在处把像素密度函数进行一阶展开得到线性近似:
再展开,就只剩下两个方向的偏导数了,像素密度函数被消去:
把窗口权重函数也加上,然后写成矩阵形式:
定义如下,就可以得到微小移动后的强度变化量的近似:
很巧妙,刚好是一个对称矩阵,一定可以对角化。上式不就是一个二次型?什么时候取得最大值?不管u和v是啥,主特征值越大就越大。直接看矩阵的每个元素也能瞥见一些端倪,对角线上的两个元素是两个方向上的偏导二次方,副对角线上的两个元素是两个方向偏导数的乘积,即滑窗在遇到x和y方向上梯度变化都很剧烈的区域时就会有很大的响应。而二次型的最小值在最小的特征值处取得,那么为了让两个方向上的变化都剧烈就只要找出的两个特征值都大的区域即可。

只要两个特征值都足够大并且相近就认为窗口内有角点出现;图源Cornell University Computer Vision Course

不过直接算上面那个式子的特征值太麻烦了,还要用求解一下才行,后人改进了一下,用下式替代:

经过上式改进的算法被称为Shi-Tomasi算法。只要算出来的大于你设定的阈值,就认为两个特征值的大小满足条件了,行列式和迹比特征值好算得多,还不用考虑特征值重数。感兴趣的同学自行展开一下上式。上式可视化之后如下,是不是和上图很相似?

R的等值线是在一族双曲线,黑色表示特征值反之则大,此图的原点在左上角,前一张图则是左下角,注意区别

OpenCV用于标定的calib3d中,用于识别棋盘标定板的函数就用到了Harris角点检测,运行Harris后,若某个区域的响应有高度一致性(很规律),就认为是标定板:

标定板的角点响应十分整齐

一些其他的角点检测算法包括如下几种,它们各有优劣之处,提出思想有相同也有不同,感兴趣的同学自行搜索,大部分都在OpenCV的imgproc库内提供了封装好的函数:

  • Kitchen-Rosenfeld 角点检测
  • Moravec算子
  • Forstner算子
  • KLT角点检测
  • SUSAN角点检测

需要注意的是,Harris角点不具有尺度不变性,以下面这张图为例,在不同的尺度或不同分辨率的图像上检测角点的时候,计算出的梯度显然是不同的,以最下方的尺度看来,这个角点已经可以被一个线段近似,导致其表现得像边缘而不是角点,是不是和微积分以直代曲的思想有点相似?可惜这个性质对于特征点检测来说可不是件好事。而在大尺度上的窗口也很可能会忽略较小的波动和弯折,这也会导致小角点被忽略。

Harris角点在相同图片的更小/更大尺度上可能无法检出
在特征点检测阶段,尺度不变性的直观理解

如果像上图一样,在两张图中找到的特征点无法重合,即在不同条件下物理意义下相同的点不能被稳定检测出来,后续的匹配也就无从谈起了。因此,不要忘记我们寻找特征点的目的,我们首要是希望能在经过变换的图像上找到和原图一样的特征点,其次才是找到那些比较有特点的点方便我们进行匹配。那么要想在不同图像尺寸和不同尺度的特征上能够重复、稳定地检测出特征点,就需要引出尺度空间和图像金字塔的概念。


5.5.4.1.2. 尺度空间及其上的极值检测

尺度空间有一套严谨的数学理论,虽然在提出之初依赖一些仿生学和直觉的知识,但是在其发展过程中形成了可解释性和假设条件都很强的理论。若要从严谨的推理和计算中理解尺度空间,请参考这篇综述:Introduction to Scale-Space Theory (jhu.edu) ,作者在文中详细论述了为什么高斯核是构建尺度空间的唯一选择、尺度空间的性质和构建尺度空间的快速近似方法等;以及这个讲义Image Pyramids and Scale Invariant Image Description(不得不说国外大学授课的讲义做的是真的好啊)。有趣的是,尺度空间的构建和热扩散方程竟然有相同的形式,而我们知道,傅里叶从热方程中导出了Furrier Transformation,而数字图像处理的又是以傅里叶分析为基础展开的。有泛函分析基础的同学可以进一步探寻它们之间的关联。另外,高斯函数的傅里叶变换仍然是一个高斯函数,只不过乘上了一个频率变换因子。这里给出一种推导的视角:傅里叶变换的特征函数(特征向量)是指数函数。

5.5.4.1.2.1. 检测目标:斑点

尺度空间理论和图像金字塔是传统计算机视觉方法解决尺度不变性的基石。本节笔者将通过一些直观的认识帮助你理解尺度空间中的尺度和分辨率(即应用在图像金字塔)的差别和联系,并通过尺度空间来找出尺度不变的斑点作为特征点(对于二维图像来说,斑点为的局部极值,可以看成像素值上的一个“孤岛”,其中间的像素值比周围都要大)。

这就是我们要找的斑点(blob),这里举得是一维的例子,二维blob就不难想象了

blob和corner一样,属于能够被精确定位的特征点。和角点一样,斑点由于其四个方向上都有剧烈的梯度变化,我们可以直接使用拉普拉斯算子对斑点进行检测,拉普拉斯算子对斑点能够产生更显著的响应。角点可以看作广义的斑点,把斑点看作圆,角点就是少了一个扇形的圆。相比角点,斑点代表的区域可以很大也可以很小,相比角点来说更加地稳定,对噪声的抗性也更强。在5.3.3.2 中我们提到,由于二阶微分算子对噪声非常敏感,我们一般采用高斯滤波器先平滑图像,再进行二阶微分的卷积运算。于是我们就决定,使用高斯+拉普拉斯算子对斑点进行检测。

由于高斯函数、微分算子对于卷积来说都是线性算子,我们就可以先将高斯函数的二阶导数求出来,再用这个二阶微分的高斯核和图像进行卷积以达到相同的效果,大大提高效率。

原信号和高斯、一阶Laplacian高斯、二阶Laplacian高斯以及经过三者滤波后的信号;imgsource:YouTube-first principle of cv

从上图可以看见,一阶Laplacian高斯算子对于边缘的响应是一个类高斯函数,有一个明显的峰,而二阶Laplacian of Gaussian则产生一个明显的过零点。而斑点则是一小块区域,在各个方向上都有边缘。那么斑点遇到二阶Laplacian会产生怎么样的响应?答案在第三小节揭晓。

5.5.4.1.2.2. 尺度和分辨率的定义

首先需要辨析一下尺度和分辨率这两个概念。一个物体的尺度是指它的大小,在物理意义下,一般以标准单位米/毫米/微米等衡量。对于计算机视觉,尺度是指某个物体占据的图像像素数,以及它们与现实世界中的对象的关系。为了定量描述图像物体的尺度大小,我们可以规定:图像中某个实体的尺度是实体占据的像素区域内可以包含的最大线段的长度(当实体是非凸多边形的时候允许线段穿过凹陷部分)。而分辨率是指“解析”细节的能力,在物理意义下通常表示为可以区分两个实体的最小距离。对于声音,分辨率表示时间采样率,并确定可以区分(或解析)的最高频率。对于图像,分辨率表示用于表示实体的空间采样率,并确定可以解析的最小实体。

网络上有大量的讲解SIFT的所谓的“教程”,都没有讲清楚尺度空间中的尺度是如何定义的,更有甚至由于不理解尺度空间,写下了“尺度空间的尺度和物体的大小不是一回事”这样的谬误,请注意甄别!在查阅相关资料的时候,尤其是在中文网络,用搜索引擎尝试检索尺度空间、图像金字塔、SIFT、尺度不变特征变换等关键词得到的结果,笔者可以肯定有90%以上都是具有误导性甚至完全错误的。

在下图中,一组与不同距离拍摄的教堂照片中,红圈的真实尺寸即物理世界中的尺寸是保持一致的,但是由于相机距离教堂的距离不同,根据小孔成像“远小近大”的成像规则,在图像上就显现出了不同的大小。从刚刚的尺度定义,直观地、定性地来看,下图左侧的图片中整个教堂就算是大尺度的物体,其轮廓属于大尺度的特征;屋檐拐角、顶部的十字架都属于小尺度物体,其上的特征更是属于尺度极其微小的一类。

教堂的一角,我们希望在不同尺寸的图片上都可以检出这个特征点

显然近处的拐角附近拥有更丰富的纹理、细节与边缘信息,更远处拍下的照片由于相机的分辨率有限导致采样率低于纹理信息的频率,高频的细节信息就被丢弃了。

不同距离会导致相同的图样在被采样时呈现不同的频率,以5.3.3.1中的那幅频域和空域对应图为例,用同一个相机在不同距离拍摄相同的条纹,在最终成像的时候,保存在图像中的条纹间隔由于“远小近大”而发生改变。类似的,上图左侧图像中就无法采集到屋檐拐角处的高频的细节。

5.5.4.1.2.3. 利用高斯核构建尺度空间

还是以上方教堂的图片为例,为了保证尺度不变性,我希望屋檐拐角这个物体不管在什么尺度下(如上图的三种情况)都能被检出,那么它们就应该有一些能够被直接感知到的相同的特性,或是我们通过一些手段把三幅图之间的不同之处给抹去。尺度空间就能够综合这两者,帮助我们实现目标。根据我们前面的学习,考虑使用三个不同尺度的模板去匹配这三种不同的情况(对于目标检测网络来说,这就是anchor,对于slide window传统方法来说,就是不同尺寸的滑窗)。那么对于我们的检测目标斑点来说,这个“模板”应该被替换成一个filter——前述的Gaussian+Laplacian的组合。“模板”的不同大小,就对应着二阶高斯核的的不同。

一方面,当高斯核的增大,其滤波模板的范围就增大,能够覆盖的范围也随之变大;另一方面,越大的意味着加权程度越高,细节的损失和模糊程度也越大。这就很好地达到了我们的目的,即感知相同的特性并抹去不同之处。而在尺度空间中,这个参数就被称作尺度,我们只会在某个尺度上观察到相应大小的特征,更小尺度的特征将由于高斯滤波器的低通特性被模糊。直接看下图:

以一维信号为例,最下方是未经过高斯滤波的信号,上方的每一个信号都是用逐级增大的高斯核滤波后生成的。细小的高频特征经过第一级滤波就所剩无几,随着高斯窗的加大,截止频率越来越低,只能保留一些低频特征。右图是左侧信号被施以不同尺度的LoG(Laplacian of Gaussian,即前面说的拉普拉斯算子+高斯算子)处理后产生的过零点(即边缘响应),在LoG的尺度小的时候,它可以敏锐地检测到微小地边缘和细节;随着尺度的增加,空间上临近的小边缘逐渐汇聚成一个较大的边缘(局部极值被融合抹去,只剩下更大范围内的极值,即“大局部“),最后消失。先看看LoG的三维图:

下图是用不同尺度的高斯核模糊后再用拉普拉斯算子处理得到的图片:

尺度小的时候,低频的特征都被保留下来,瓷砖上的细小纹路还能在图像得到清晰的反映;随着尺度的增大,慢慢只剩下啊更大尺度的纹理和边缘;在最大的尺度上(这里是),顶部两块瓷砖的边缘都已经被模糊。而用一系列不同尺度的高斯核卷积原图生成的对应图像的集合,就被称为一个尺度空间,其中的每一幅图像中都包含了属于那个尺度上的特征。

相信你现在已经了解尺度的定义和构建尺度空间的意义了,我们不需要改变图像的分辨率,只需要用不同尺度的高斯核处理图像即可以得到一系列包含不同尺度特征的图像。现在回答第一小节提出的问题:LoG怎么检测斑点?请看下图:

img src:YouTube-first princple of computer vision

上图一共有四组不同大小的对照,每一组实最下方得到的信号是规范化的LoG处理原信号后得到的响应。由于斑点的两个边缘的过零点很接近,它们的下侧峰发生重叠直接合二为一,生成一个很强的峰(响应)。观察每组对照的最后一行,我们会发现,不管blob的尺度有多大,总有一个尺度上的LoG能够对此斑点取得最大的响应,而取得最大响应对应的LoG的尺度,就被称为这个斑点特征的尺度。该尺度也被称为这个特征的scale signature(尺度签名?意思就是独一无二),而能够在尺度空间中以尺度不变性检出特征点的函数被称作signatue function。换句话说,不论斑点特征属于哪一个尺度,我们都能通过一个对应尺度的标准化LoG”模板“将其检出。

不同尺度的规范化LoG可以检出不同大小的斑点,右侧纵轴为尺度,横轴为不同尺度的blob

再通俗一些,假设我们用上图最大的规范化LoG”模板“覆盖到下方最小的斑点上,虽中间确实会产生稍大的峰,但总响应肯定是不及覆盖最大的斑点时产生的响应,因为除了模板中间最高的”尖峰“部分,剩余的”山坡“都被浪费掉了。而我们用小的LoG”模板”去套最大的斑点,就会导致山旁边的“护城河”也和斑点重叠,使得山峰和山坡上产生的响应都被抵消了(下图),因此也不会有很大的响应。只有当斑点的大小恰恰合适,即刚好覆盖到“山坡”和“山峰”,又没有触及到“护城河”的时候,才会产生最大的响应。也就是说,只有某个斑点遇到其对应尺度的标准化LoG时才会产生最大的响应(这时候是在纵轴上对比,即尺度轴,而非图像的二维平面)

把标准化的LoG倒过来看看,周围有一圈反向的“环形山”

我们再看个例子,5.5.4.1.2.2处的教堂图片,要想把红色圈中的屋檐拐角定位出来,在不同的尺度上就要用不同直径的红圈!

要是还有点不明白的话,视频讲解戳这里:Detecting Blobs | SIFT Detector - YouTube,这里也推荐一篇对尺度空间斑点极值检测的尺度不变性讲得比较好的博客:对SIFT算法尤其是尺度不变性的理解-ChuanjieZhu的博客

至于为什么要使用规范化的LoG即乘以尺度的平方,直觉的解释是,随着尺度的增大,高斯函数也变得越来越平滑,它的峰值响应会逐渐减小。虽然我们仍然可以在斑点处利用LoG检测到极大值,但是相对于其他地方来说这个极值显得太不明显了,不够稳定且容易被噪声干扰。这是二维高斯窗的公式:,每次求偏导都会将幂指数上方的乘到前面的系数项,使得函数值成比例缩小。为了保持不同尺度的LoG拥有相同的性质——,标准化的LoG在不同尺度上对应不同的斑点就将拥有等比例缩放相同的响应。这样我们就能保证在较大尺度的时候,遇到斑点仍然拥有比较尖锐的响应。要从数学上推导,参见Feature Detection with Automatic Scale Selection

5.5.4.1.2.4. 通过DoG近似LoG

虽然标准化的LoG好使,可以很容易地帮我们检测出不同尺度的斑点,但是算起来太复杂了,二阶导数的计算量有点大,为了提高计算速度,最常见的做法就是近似。于是有人发现尺度空间中响铃的两张图片直接做差,就可以得到LoG的近似了,数学上的推导就不写了:

从直觉的角度理解,DoG实际上包含了在生成尺度空间的过程中被模糊掉的那些信息:尺度空间中相邻的两层图像之间相差的就是一次高斯滤波,做差之后这些信息就进入了DoG中。这样看来,DoG似乎保留了图像在尺度变换中的残差,这和ResNet让网络拟合残差、Diffusion model每次扩散去除噪声的思想又有些相似。

对比一下标注化的LoG和DoG:

是不是很像?
5.5.4.1.2.5. 筛选极值点

根据我们的已有知识,终于可以开始寻找那些尺度不变的极值点了。首先构建尺度空间,我们就会得到一个3维的stack(两维为xy,另外一维是构建尺度空间的时候使用的)。如下图所示,我们在每一个像素点处进行极值点检测,对比其像素值和周围26个像素的大小,如果比周围的像素都大/小,那么就是尺度空间中的一个极值点。

“x”是待检测点,筛选需要对尺度空间中的所有点运行,复杂度是26hw*(t-2),t为使用的尺度个数

下图是对尺度不变的直观理解,这里有两幅不同尺度的图片,我们利用不同尺度的标准化LoG处理“x”位置的那个角点,会产生下方的响应值,四组图片中使用的LoG的尺度逐渐增大。对于图像中这个相同的但是尺度不同的角点(“x”处),总有一个尺度的LoG会对其产生最大的响应。那么不管这个角点被缩放到什么大小,我们都能通过在尺度空间中检测极值将其检出,这就实现了尺度不变性。

上图的角点由于尺度不同,其极值出现在不同的尺度中,但是不管怎么样,它们都被检出。

实际上,SIFT算法也建立了图像金字塔,其目的是为了减小构建尺度空间的开销,因为在相同分辨率上进行大核卷积(即尺度很大的时候),时间开销是非常大的。我们在5.5.4.1.5将会介绍图像金字塔和尺度空间之间的千丝万缕。

说到这里,要是你对5.5.4.1.2尺度空间中尺度不变的斑点检测还是有些晕,我们再缕一缕思路:

  1. 我们想要找的interest point是斑点

  2. LoG可以检测边缘,响应表现为一个过零点,而斑点可以看作各个方向上都有边缘的一个区域

  3. 由于在不同方向上都出现过零点,处于边缘内侧LoG的响应峰会重叠在一起,从而对斑点产生很大的响应(内侧峰形成“山”,而外侧的响应峰则形成“护城河”)

  4. 某个尺度的斑点会对这个尺度的归一化LoG产生最大的响应

  5. 不管斑点被缩放到哪一个尺度,总有一个尺度的LoG可以给它以最大的响应,这就表现为利用LoG检测处的斑点具有尺度不变性

  6. 建立尺度空间,在尺度空间中利用DoG近似LoG,通过寻找尺度空间(3维)中的极值,将斑点检出


5.5.4.1.3. 极值检测的改进与积分图近似
5.5.4.1.3.1. 亚像素级特征定位

在前面一个小节中寻找极值的时候,实际上只运行了一轮预筛选,仅仅对比了其相邻像素的差值以初步确定候选点。因为是在离散空间中找极值,这个候选点可能不是真正的极值点。为了更精确地定位极值点,我们将通过插值来寻找亚像素精度的极值点。以一维的情况为例,看下图:

竖线即每一个像素(采样点),极值点可能出现在采样点之间;图源不知道哪里,侵删

那怎么样才能得到上图中的连续的曲线?显然得不到,不过可以近似,请出老朋友泰勒,我们在刚刚得到的初步候选点附近,对DoG进行展开:

其中为相对候选点的偏移量。我们在尺度空间中进行展开,因此是一个三维向量,让上式求导后等于零,得到下式:
就是展开式的导数为零的解,这样就得到了插值后的真正的极值。不过别急,显然我们的也是离散空间中用差分计算的,并不是连续的精确解,那怎么办?仿照牛顿法解方程,我们多迭代几次就行。在迭代的过程中,我们还可以删除那些不稳定的点。为了提高效率降低复杂度,在迭代后算一下新的极值点上的响应:
如果响应值小于0.03就认为太小了,直接舍去;如果迭代得到的偏移量在加上之后超出图片范围,也直接将这个极值点舍去。如果在任意维度的值大于0.5,说明它偏离到尺度空间其他采样点上了(xyσ的间隔都取1,即插值点在此时已经离别的点更近而非展开点),那么就移动到新的展开点进行计算。可以设置固定的迭代轮数,也可以设置阈值即当迭代后产生的偏移量小于此阈值时停止迭代。

经过上述操作我们不仅得到的更精细的亚像素级特征点,还能在迭代过程中将那些不稳定地、响应不够强烈地特征点筛除。

5.5.4.1.3.2. 二维前缀和:积分图

学习过数据结构/参加过OI/ACM的同学对前缀和应该很熟悉。对于序列数据,要想多次计算其某个子串的和,就可以利用前缀和这种结构,它只需要的时间就能得到任意字串和。网络上有大量相关资料,在此不过多赘述。对于二维数据,我们也可以仿照前缀和,在两个维度上计算某个子矩阵的元素总和,这种方法被称作积分图。这里给出一个介绍上述两种方法的教程:前缀和 - 知乎 (zhihu.com),若你已经了解积分图,直接跳过看下一节。

积分图的建立类似动态规划,在完成建图后只需4次寻址+三次加/减就可以得到任意矩形范围内元素和
5.5.4.1.3.3. DoG的超级近似:更快速地快速计算LoG

先看看高斯函数在不同方向上二阶导数的形状(即LoG在某个方向上的“边缘分布”,固定xy中的一个轴):

再仔细看看上面的图片,再想想为什么要使用积分图?这是因为我们希望用积分图上的矩形来近似替代LoG的计算!若要进行标准化,仍然是计算后乘以尺度的平方。看下图就更清晰了:

对于LoG,直接用相同权重的大块矩形和替代原来的卷积操作

上面得到得是LoG在各个方向的分量,实际要得到LoG的话,把不同分量加在一起即可。用上图左侧的矩形块计算LoG就不再需要进行超高复杂度的卷积,这种模板被称作box filter(盒式滤波器)。现在要建立尺度空间,就比原来简单得多了:对于图像输入,首先建立积分图,这需要花费w*h的时间。随后,利用不同尺度的box filter对原图进“滤波”,就构建出了尺度空间。最后,用前面介绍的方法在尺度空间中寻找极值点即可。

复习一下微积分,拉普拉斯算子的表达结果是这样的:,因此要获取拉普拉斯-高斯响应,我们就要把纵向和横向的盒式滤波器的响应加在一起。对于这一项,在构建尺度空间的时候是不需要使用的。

SURF(speed-up robust feature)是首先使用盒式滤波器构建尺度空间的算法,实际上该算法使用海塞矩阵Hessian Matrix的特征值检验来检测尺度空间中的极值点。并且利用盒式滤波器生成尺度空间时,在任何大小的开销都是常数时间级的,因此SURF不需要使用图像金字塔。阅读SURF论文你会发现它仍然产生了几组(octave)不同分辨率的图像,这是因为盒式滤波器在增大的时候(即尺度增大),并没有添加相应大小的padding。

相比规范化LoG,通过计算Hessian矩阵能够得到尺度空间中更稳定、重复性更好的特征点。感兴趣的同学可以直接参考SURF的论文,摘要和背景中有关于Hessian矩阵和LoG在尺度空间中的异同的介绍。

一篇从数学上介绍为什么Hessian能够用于特征点检测的博客戳这里:Introduction to the Hessian feature detector for finding blobs in an image - Milania's Blog,讲的非常好,而且这位作者的前端功底非常牛批,甚至在博客界面给读者提供了拖条可视化。

在此我们简要地讲述一下Hessian矩阵检测特征的直觉,在多元微积分部分大部分同学都应该学习过相关的知识,直接看Hessian的形式:

对于f为高斯函数的情形,此时H的迹实际上就是LoG!那么Hessian的物理意义是什么呢?还是先考虑一维情形,一阶导数正负代表变化方向幅度代表变化快慢,而二阶导数正负代表变化趋势快慢(曲率的大小),正负代表弯曲方向。二维的情形就复杂了,往各个方向移动会有不同的曲率,对于二元函数的一阶导数,我们会使用方向导数来描述某个特定方向的导数,对于曲率我们同样可以使用方向曲率的概念!计算方向导数,我们需要用这个函数的梯度乘以一个单位化的方向向量,那么对于方向曲率我们又应该怎么做?看看多元函数的泰勒展开:
一阶代表函数值当前变化方向带来的附加值,二阶代表函数变化趋势带来的附加值,也就是曲率变化的影响。那么要计算方向曲率,通过(c表示curvature)就行了,这又再次转化为了一个二次型问题!是一个对称阵,接下来该怎么做应该很清楚了:对角化,利用特征值来解决问题。虽然矩阵元素的含义和特征值的意义不同,但后续的解决方案是不是和前面利用梯度变化来解决Harris角点检测的问题完全一致?那矩阵有什么含义呢,他的两个特征值就代表了曲率变化最大和最小的两个方向。要用Hessian矩阵检验斑点,我们就应该找到那些所有方向上曲率变化都很大的区域,话不多说直接看图:

橙色向量就表示检测点出Hessian矩阵的两个特征值的方向(实际上是特征向量的方向,长度为特征值);图源https://blog.csdn.net/sy95122/article/details/80825846

对于边缘,显然只有一个方向的曲率变化大,即垂直边缘的方向。所以SURF最终采用的检验尺度空间极值的方法是找到尺度空间中满足下式的点:

其中是用于修正盒式滤波器误差的系数,因为盒式滤波器将相同的系数赋予所有位置,而在边缘处响应实际上要弱一些,因此用此系数修正。可视化的效果和5.5.4.1.1中的双曲线图是一样的!上式为单个尺度的响应,在为每个尺度都计算出响应之后,用5.5.4.1.2.5中一样的方法找尺度空间极值。


5.5.4.1.4. 基于像素比对的FAST特征点检测

FAST在Fusing Points and Lines for High Performance Tracking中被提出,是为了解决目标跟踪中图像特征在短时间内会发生巨大变化故无法保证视角不变性的问题而提出的。显然传统的特征点检测算法都是无法提供视角不变性的,因为所有特征都是在线提取并用后销毁。FAST特征虽然不如SIFT、SURF稳定,但是速度足够快,这样就能保证两帧之间视角不发生太大的变化。

虽然利用盒式滤波器替代归一化LoG构建尺度空间已经让速度提升了接近十倍,但寻找特征点的总耗时仍然是秒级的,而FAST(Features From Accelerated Segment Test)的提出直接把寻找特征点的时间干小了两个数量级。前面的方法都是通过梯度、曲率等需要微分算子作用的特征来描述像素的变化,这种方法一方面可以提高特征点的稳定性(因为梯度和曲率相对于像素值要稍微high level一些),另一方面是为了方便构建尺度空间并在尺度空间中找到LoG等的自适应尺度模板(即尺度不变地找出特征点)。而在图像上的微分实际上都是通过差分实现的,那为什么不干脆就直接通过像素间的差值来描述变化?只要保证这种方法足够稳定就行,况且直接检查像素,速度要快得多。

从像素值的层次来看,角点或斑点和周围像素应该有显著差别,那么我们需要一种定量的方法描述像素如何不同,FAST就直接对带检测点及其周围的一圈像素点进行测试:

如果想提高精度,还可以对斜角上的像素进行插值

测试点和半径为3的一圈16个像素值进行比较,当16个像素中出现至少12个连续的像素都大于或小于测试点的像素值时,就认为这是一个潜在的特征点。为了提升特征点的稳定性,还为比较设置了阈值,即将条件收缩到当这些像素和测试点差值的绝对值大于阈值的时候才认为通过。由于要求是12个连续的像素点,FAST还能够进一步加速:先看3、7、11、15号像素,只有当这四个像素有3个满足前述条件的时候,才对剩下的点进行测试。因为至少12个连续的点一定会包含这四个点中的至少三个。经过这样的预筛选,FAST的速度又进一步提高了。

为了避免较大尺度特征点附近的集中响应(多个相似的集中响应用一个就能代表全部)和强响应附近较弱的响应(不稳定),在运行检测的时候同时计算每个测试点的响应得分:

上式就代表这个点和周围像素的差异,取更亮像素和其差值之和以及更暗像素与其差值之和的最大值。然后,对于每一个测试点及其的3x3邻域内的其他响应点运行NMS,比较它们的分数筛除弱响应和集中响应(重复响应)。

后续对于FAST,还有人提出利用决策树的机器学习方法判断测试点是否为特征点。可以把测试点附近的16个像素点按照下面的方式进行分类,我们便会得到一个16维的分类特征,其中为上面介绍到的阈值:

而最终分类的目标就是判断该测试点是否为我们要找的角点,因此我们只需要建立一颗2-state的决策树(实现上一般常用ID3-tree),树叶即对该角点的分类。先通过其他的特征点检测算法制作数据集,然后用决策树训练一个分类器即可。由于大部分时候都不会触碰到最长的树枝,平均复杂度应该是即只需要8次测试就能确定角点的分类。(多少带点瞎猜的成分)

至于检出时的尺度不变性,FAST是通过建立图像金字塔,然后在不同分辨率上的图像中分别运行一次角点检测实现的。

5.5.4.1.5. 图像金字塔和尺度空间的区别和联系

这两个概念非常相似,甚至在一些地方一起出现(SIFT会构建一组金字塔,在金字塔的每一层都生成一系列的尺度空间),但实际上两者的内涵有所不同。我们直接来看看他们的使用方式:

  1. 图像金字塔将图像按照成比例的方式连续缩放,生成一组尺寸为指数增加/减少的图片。金字塔中相邻的两张图片的长宽均相差一个倍数。然后,使用等大小的滑窗进行特征匹配或特征提取(即提取特征所用的窗口/patch大小在所有层都相同)。如下图所示,我们选取了一个固定大小的红色窗口用于模板匹配,然后生成了一系列等比图片构成金字塔。随后在每一张图片上都运行模板匹配算法。最终在每张图片上都能匹配在原图上尺寸不同的大雁。
  1. 尺度空间则不会缩放图像,而是缩放构建空间时使用的高斯核的大小。这里的“大小”指的是Gaussian滤波器的像素尺寸,即其覆盖/作用的像素面积。下图就是在同一张图片上使用不同大小的高斯核构建的一组图片,它们构成了该图像的尺度空间。倘若我们希望在尺度空间中寻找一些特征,例如通过LoG寻找边缘和blob响应,则在大的尺度下就应该使用大的LoG(大小同样以像素计),因为在较大的尺度之下,纹理和细节都被高斯核模糊,只存在低频特征。反之,则应该使用小的LoG。

分析下来,实际上它们是解决相同问题的两种思路。一切都是相对的:图像金字塔让我们的模板/特征提取窗口的大小保持不变,改变图像的比例以获取原图上不同大小的特征;尺度空间则保持图像大小不变,生成一系列不同大小的特征提取算子。两者的不同之处在于尺度空间的每一层分辨率相同,而金字塔从下到上的分辨率是逐渐减小的,显然在图像金字塔上进行特征提取和检测,开销要比尺度空间小得多。在构建图像金字塔的时候,由于图像处在离散空间,上采样/下采样都需要用到一些插值的方法,如最近邻,线性/双线性插值等,而尺度空间则可以直接选择不同大小的高斯核,构建连续的尺度空间表示。由于效率较低且理论复杂,尺度空间在现代图像处理的应用中的占有率逐渐被图像金字塔取代。

SIFT在实现的时候就使用了图像金字塔+尺度空间,倘若继续增大尺度,构建尺度空间的开销就太昂贵了,在尺度空间中每添加一层都需要在原图上运行一次高斯模糊,并且生成的新的尺度图像分辨率并不会下降(如果添加了padding)。SURF则通过box filter对LoG的近似,使得其构建任意尺度空间中任意层图像都只需要常数时间(使用了积分图),所以没有选择使用图像金字塔。

而不满足尺度空间的算子(有很多非线性的算子不具有signature function的性质,无法在尺度空间中获取单个极值)、不需要提取微分特征(如FAST这种直接比较点对的)就会选择使用图像金字塔。

另外,图像金字塔还常用于coarse-to-fine的特征检测和提取方法,之前介绍的光流法中就可以使用图像金字塔以追踪两帧间隔较大的图像中的相同点,我们在之后会再次提及。

5.5.4.1.6. 再次检验invariant feature

首先是尺度不变性,这可以通过signature function在尺度空间中匹配找到那些极大响应;对于没有构造尺度空间的方法,则是采用图像金字塔。再看看光照不变性,图像整体的亮度变化相当于给所有像素值增加/减少相同的强度,微分算子计算的是像素的差分值,同增同减这种共模信号会被直接忽略,checked!还有对比度不变性,对比度的改变只会导致响应的强弱发生变化,对于不同对比度的情形可以使用5.5.1中介绍的自适应阈值方法减小对比度改变的影响;即使仍然使用固定阈值,也只会导致那些不那么稳定的特征点被筛除(即降低对比度的阶数,导致像素被“拉近”从而减小响应),稳定的特征点(和周围像素差别很大)仍然得以保留,checked!对于FAST特征点,对比度的改变也不会产生影响,“强者衡强”的差分思路再次展现了他的为例。最后就是旋转不变性,高斯算子、拉普拉斯算子以及组合生成的LoG都是完全对称的,不管图像怎么旋转,都将产生稳定的响应,checked!


5.5.4.2. 建立特征描述子

在找到特征点之后,我们就要为特征点生成特征,即其”身份信息“,通过这个特征,它才能和模板或其他图片上找到的其他特征点之间进行匹配。在寻找特征点的时候,我们所叙述的那些invariant feature说的是不管这些变量如何变化,我们都能够稳定地将它从一众像素之中拎出来。而现在要建立特征描述子,我们所期望的不变性则是在光照、角度、位置、尺度等外部条件发生变化的时候,对于一个物理上相同的特征点(比如找到的角点、斑点等),我们应该要为它们生成相同的特征描述子。

其实在找到特征点之后,事情就变得很简单了。还记得我们在5.5.3介绍的local distribution方法吗?我们只要以检测到的特征点为锚(说人话就是把以特征点为中心)并在其邻域内统计一些特征就可以了!讲完了特征点检测,让我们再看一眼 5.5.3 中出现的这幅图:

“x”就是我们找到的特征点,我们只需要在其邻域抽取特征,建立特征描述子

再感叹一下这幅图不可不谓精髓啊!几乎一图就涵盖了特征点检测、特征描述子建立和匹配所有的流程。

下面我们先大体介绍一下在描述子建立阶段获取各种不变性的方法,然后再讲讲不同特征描述子的具体实现。

5.5.4.2.1. 如何保证各种不变性?
  • 尺度不变性

对于建立了尺度空间的特征点检测算法,检出的特征点已经被分配到最合适的尺度,因此我们只要在对应尺度上去提取特征即可。对于不同尺度的特征,应该选取不同大小的邻域半径。这个道理也是显而易见的,特征提取窗口的大小应该和尺度等比例缩放。对于小尺度的特征,就用小窗口提取其内部精细的纹理和细节;大尺度的特征将会在对应尺度上用大窗口抽取,因为在大尺度图像上,小尺度的特征已经被高斯函数模糊,因此在对应尺度上使用大窗口只会提取对应尺度的特征。

而对于使用图像金字塔的方法,和检测特征点时一样只,不论在图像金字塔的哪一层,提取特征时都对锚点使用相同大小的邻域。可以对比一下神经网络中使用的anchor,我们希望用不同大小的anchor来匹配不同大小的目标,这里随检测目标大小变化的是anchor的大小;而图像金字塔,随特征点尺度变化的则是图像的尺寸。大和小都是相对的概念,因此缩小图像就等同于放大anchor(保持另外一个参数不变),而且缩小图像还有一个好处,就是可以降低计算量!

  • 旋转不变性

再看一眼 5.5.4.2 中的那幅图,我们是无法事先预料到检测的目标旋转的角度的。在检测阶段由于使用了尺度不变的算子,不管图像怎么转相同的特征点都会被检验出来;那么在建立特征描述子阶段,我们只要知道特征点发生了何种程度的旋转(到底转了多少),就能够在对其进行描述的时候,将特征旋转对应的角度(换一种思路就是,先转对应的角度再提取特征),从而实现特征描述子对旋转的不变性了。

  • 光照/对比度/...不变性

在介绍自适应阈值时和HOG行人检测的时候就已经提到了,对抗这些变化最有效的方法就是将特征进行标准化和归一化,光照的改变就是亮度统一增减,均值化以后共模信号统统被干掉;对比度的变化实际上改变了方差,在方差标准化后也消失得无影无踪。

5.5.4.2.2. 梯度直方图

HOG算是我们的老朋友了,在前面的5.5.2.3我们就介绍了这种特征描述子。梯度直方图尤为适合那些在选取特征点阶段就已经计算过梯度的特征点检测方法,不需要再为建立描述子单独计算梯度。

在尺度空间中检出的特征点,直接到对应尺度上,圈出一块地(根据尺度设定的邻域大小,即上图中的蓝色圆),然后划分格点一通算就完事了。为了让不同尺度生成的特征描述子拥有相同的长度,格点的划分应该采取相同的策略,如统一取8x8、16x16。注意在大尺度的时候,一个格点可能会覆盖多个像素,为了提高精度可以使用插值的方式进行计算。

后续步骤和 5.5.2.3 完全一致,不再赘述,这里着重说说如何确定特征点的方向。确定特征点的方向和提取特征基本大同小异,只不过不用划分子区域。以SIFT为例,首先在选择和尺度对应的邻域大小(如尺度为3,那么邻域半径就选3),然后为这个邻域建立梯度直方图,将属于各个方向的梯度幅值累加到对应bin中(同样,如果想提高精度可以对角度进行插值或分解到两个不同方向):

分配方向示意图,忽略更粗的绿线,我们只需要在整个邻域范围内统计直方图

另外,在分配方向的时候,SIFT选择给梯度幅值加上一个以特征点为中心的高斯窗,减小边缘位置的梯度对直方图的贡献。这也是一个很直觉的想法,越靠近中心贡献越大嘛。对于大尺度时的方向分配,和计算梯度特征的时候一样,可以选用插值的方式提高计算精度(随之而来的是速度的下降)。在统计完直方图以后,选取幅值最高的那个bin的方向作为该特征点的方向。

接下来要怎么做就很清楚了,在建立特征描述子之前,我们首先要将特征点旋转到其对应的方向,再进行特征抽取。这样不管图像或物体怎么转,只要在提取特征的时候,先旋转至特征点的方向完成对齐,就不用担心得到的特征因旋转而改变了!如果不理解,再看看5.5.4.2的那张猫咪的图片。

SIFT在具体实现的时候,还有一些细节:在计算特征点的方向时,如果有其他的bin幅值超过了主方向的80%(最大值的80%),就将其作为辅助方向以提高特征点匹配的成功率。具体实现为把该特征点复制几份,并分别赋予这些辅助方向作为它们的主方向,这可以提高匹配阶段的成功率和算法的稳定新(虽然会牺牲速度)。对比CNN的目标检测来说,就是增加检测框(降低置信度)以提高召回率!

5.5.4.2.3. wavelet(SURF-liked)

wavelet即小波特征,在5.5.3.2介绍GIST的时候我们曾提到小波变换,它是一种“局部”的频率特征(对比Furrier变换)。

熟悉OpenCV或人脸检测的同学肯定知道,Haar人脸检测器就是使用简单的小波特征建立Ada-boost分类器实现coarse-to-fine的级联检测机制。Haar也是用了积分图,一会看看Haar特征长什么样,你就知道为什么要SURF会选用Haar特征了。

小波的理论基础过于复杂多说无益,直接看Haar特征长什么样:

常用的Haar特征模板,白色取1黑色取-1

实际上也不用管什么小波特征,下面这个看起来是不是和以前见过的微分模板/边缘检测模板很像?直观上理解,就是在计算近似的梯度嘛。

有了积分图,算这玩意不是有手就行?斜着的特征还需要插值,方形的特征要拆成5个部分分别算,麻烦!我们就用最简单的水平和竖直的edge feature:

还是以特征点为锚点,选取对应尺度大小的邻域,对下图所示的扇形区域内的所有点计算上面的小波特征,按角度求和(实际上和直方图一样,把方向分配一下)。当然了,小波模板的大小也要跟尺度一起变化。算完之后转一个角度(SURF论文中取扇形为60度,每次旋转60度),再算。最后,取小波响应和最大的扇形方向为该特征点的主方向。SIFT用过的trick这里也可以用,比如给小波响应加一个高斯窗啊。不过插值可就用不上了,想想为什么?这就是积分图的好处!

这里实际上忽略了一些细节,还需要根据水平方向总响应和数值方向总响应在60°扇形内确定更精细的角度,感兴趣的参照SURF论文(前文已经给出链接)。

有了方向剩下的也没什么好说的了,先把窗口转到主方向,然后在邻域里划分格点和种子点,对每个种子点都建立一个Haar特征的直方图,统计完Haar特征后再把特征归一化,一个特征描述子就建立好了!

以上图为例,选用了8x8的格点,以4个格点为一个直方图统计单元(即一共16个种子点)

旋转之后Haar特征怎么算?积分图可只能计算横平竖直的响应啊!问题不大,上图右侧也给出了答案,就还是算水平和竖直的Haar特征,然后根据旋转的角度将刚刚算出的特征在这个角度上的分量累加在一起就行了。

SURF在建立特征描述子的时候,还会加上额外的一个“标志位”,即特征点处拉普拉斯响应的正负号。这样在进行匹配的时候可以减少一半的工作量,只尝试匹配那些符号相同的特征点。回忆一下多元微积分,拉普拉斯响应的正负号代表什么?

5.5.4.2.4. 标定质心法确定主方向

前面的特征点确定主方向一般都需要再次提取特征或计算模板响应(相关性计算),然后选取特征最多/模板响应最大。这类方法最大的缺点就是速度太慢,而质心标定法就是为速度而生的,它的做法非常直觉:计算特征点为中心的patch的“像素强度质心”,即对于单通道图像,每个像素的强度就是它的weight,然后对该patch计算质心即可。以5x5的patch为例,此处的x和y坐标都以特征点为中心零均值化:

组合得到的向量的方向,就是该特征点的主方向。

当然我们计算主方向是为了保证特征描述子的旋转不变性,但若你希望引入一些额外的特征,你也可以将质心法算出的质心向量归一化之后,将其模长作为一个预筛选特征用于稍后的特征点匹配,质心距离patch中心的距离一定程度反应了某种特征。

5.5.4.2.5. LBP

全称为local binary pattern(局部二值模式),看名字也不像是用了梯度特征,不过可以看得出来由LBP编码得到的特征是二值特征。它的建立方式和FAST角点检测一样,是基于像素值的比较。因此,对于在特征点检测阶段使用图像金字塔的方法,LBP是一个不错的选择。

二值特征相比于直方图特征拥有无可比拟的特性,在计算两特征描述符的匹配程度时,可以直接使用异或计算,而异或恰恰是计算机在底层完美支持的最基础的指令之一;并且二值描述占用的空间特别小,相比直方图特征的每一个bin至少为int32,二值特征只需要对应位数的内存就能够被保存。

LBP最早还不叫这个名字的时候,是研究texture spectrum的一种方法,看到spectrum就知道,这种研究范式应该是在尝试对图像特征进行某种线性/非线性的分解(像傅里叶变换那样),有兴趣的同学可以参考Wikipedia中的相关词条。基于像素值比较生成LBP特征描述子的方法首先在这篇论文中提出:Performance evaluation of texture measures with classification based on Kullback discrimination of distributions。随后,An eXtended Center-Symmetric Local Binary Pattern for Background Modeling and Subtraction in Videos这篇论文对过往的各种LBP改进方法进行了总结,并提出了一种更强大的方法以提高稳定性并为描述子提供了旋转不变性。我们按照发展顺序对LBP进行介绍。

最早的LBP描述方式非常简单,以锚点为中心建立半径为1的邻域,然后依次比较周围像素和锚点的像素值,像素值大于等于参考点者被置1,小于者置零;然后从左上角开始顺时针将比较后的二值数据序列化成8维的特征向量,就得到了LBP编码:

左侧图像生成的LBP编码为10001011

这种编码方式最多只会产生256种不同的特征,显然在实际使用中会出现大量的编码重复,并且最致命的是,它不具有旋转不变性,一旦特征发生旋转,生成的编码就会出现“循环”。那我们首先解决第一个问题,重复率过高。问题的原因就出在原始版LBP的编码方式太粗暴了,比锚点大的像素就全部变成1,小的全部变成0,它们之间没有任何区分度。FAST在找特征点的时候好歹还设置了一个可以调整的阈值。因此我们要想办法凸显出不同像素值在编码后的不同,但又不能增加太多的计算量,上文引用中的第一篇论文给出了解决方案:

邻域中的每个位置被赋予了位置相关的权值

在计算得到LBP编码后,将LBP编码乘以位置相关的权值,就会得到(d)所示的编码,将(d)中所有位置的编码求和,得到一个介于0~255之间的数,作为附加特征添加到二值编码中。这种方法被称作two-level LBP,在匹配的时候,首先检验二值编码是否相同,通过后再比较两者的附加特征。这样的改进也是非常巧妙的,在没有增加什么计算量的情况下,就轻松的扩展出了共 6561 种不同的编码(感兴趣的同学可以用排列组合的知识计算一下),使得稳定性和不同描述子之间的差异性大大提升。

接下来就要解决旋转不变性的问题。SIFT和SURF都是采用为特征点计算主方向的方法来实现旋转不变性,它们在获取主方向的时候都用到了梯度特征来统计直方图。LBP作为像素点比较的方法,为了得到方向而把每个位置的梯度求出来肯定得不偿失,它追求的就是速度。我们可以直接使用刚刚在5.5.4.2.4介绍的标定质心法来确定特征点的主方向,将邻域旋转到主方向后再提取LBP特征。

不过这里还有另一种方法能够让LBP编码本身支持旋转不变性。我们前面提到,在特征点发生旋转的时候,编码会出现循环的现象,以上图为例,若特征点顺时针旋转45度,第一个level的特征就会从10001011变成11000101,即发生了一位的循环右移,这会让编码冗余度不足的情况进一步恶化。不过在经过上面的two-level改进之后,我们已经拥有了6561种编码,建立描述子时重复的概率已经大大下降。那我们干脆就让编码循环起来,使得level-1的特征从256减少到36种(并不是,圆形是首位相连的),通过level-2的特征进行进一步区分。

所有旋转LBP模式

注意level-2的特征要和level-1一起旋转,否则固定level-2旋转level-1会导致发生转动后level-2特征不一致。自己动手画一下就能明白。

这番操作让重复率上升是肯定的,不过它以几乎为零的代价让描述子具有了旋转不变性,这自然是非常值得的。不过上述方法中给固定位置赋予权值的模式有点死板了,这只能凸显比锚点大或锚点小的像素出现的位置却还是不能比较它们之间的大小。因此另一种LBP描述子使用的方法是,对邻域内所有特征点根据像素值进行排序,按照从大到小的顺序赋予128、64、... 、1的权值(如果比锚点小就全部赋0),这样依赖Level-2的特征就不必和level-1一起旋转,还能区分邻域内像素值的不同。

为了进一步提高特征的丰富性,降低重复性,还可以为LBP选择不同的邻域大小,将多个尺寸生成的描述子拼接在一起,形成更稳定的特征描述符:

注意LBP一般在图像金字塔中提取,因此这里选择多个邻域是为了丰富特征,而不是像尺度空间那样选择对应尺度

对于图像金字塔中的每一层,LBP都要选择多种邻域进行提取,因此对于每种尺寸的图片实际上还是使用了一种模板(区分尺度空间中的自动尺度选择!)。看看上图就知道,你可以通过插值的方式让采样点处的像素值更精确。最后,以上图为例,生成的特征描述子一共有39维(8+12+16: binary以及1+1+1: int)。

最后检测一下LBP的不变性:尺度(图像金字塔),旋转(循环编码/标定质心),光照(强者衡强,只比较大小),对比度(只比较大小,不在乎差了多少),all checked!

5.5.4.2.6. BRIEF

全称为binary robust independent elementary feature(二值鲁棒独立基础特征,论文戳这里),和LBP一样也是一种二值描述符,不过它建立描述子的方式可谓独树一帜:以特征点为锚,选取31x31大小的邻域并在其内部选取256对随机采样的点对比较大小,从而生成256维的特征描述子。如下方公式所示,当选取的第一个像素大于第二个像素,该位特征置1,反之置零:

生成的就是256维的描述子了。论文给出了几种随机分布抽样的方法:

  1. 都呈均匀分布
  2. 都呈高斯分布
  3. 服从高斯分布 , 服从高斯分布,采样分两步进行:首先以原点为中心为 进行高斯采样,再以 为中心为 进行高斯采样
  4. 在极坐标下的离散位置处进行随机采样
  5. 在空间量化极坐标下的离散位置处进行随机采样,这种方法其实和LBP很像,只不过第二个点的选取是随机的
五种不同方法抽样得到的点对示例

在采样之前,还需要对全图进行高斯模糊以避免噪声带来的影响,对于采样到分数坐标的像素可以通过插值以提高精度。那么匹配的时候怎么做呢?显然要计算汉明距离,不过由于采样点对时使用了随机策略,我们就可以假定所有特征描述子中的1的出现概率一定是1/2,那么在匹配的时候,直接将那些汉明距离大于128的描述子筛除(思考一下,就是一个概率问题!),然后将剩下的描述子两两计算汉明距离,距离最小的视为配对成功。

经过实验发现第二种采样方法能够得到最高的匹配正确率,而且不论采用哪种方法,作者都发现匹配成功的点呈现近似的高斯分布,而未匹配成功的点、被直接筛除的点则自然而然地呈现出以匹配时的汉明距离大小为128为中心的非常漂亮的高斯分布(注意是匹配的时候计算得到的汉明距离,从采样方式中显然可以推断得到这样的结果):

第二种方法的实验结果,蓝色为匹配成功者,红色反之

要拥有尺度不变性只需要建立图像金子塔然后在每一层都进行检测(固定窗口31x31);而由于采样的随机性,想要获取旋转不变性从描述子本身也不太好下手,因此一般通过标定特征点质心的方法获取特征主方向,将邻域旋转到对应主方向后再进行采样。同样,对于光照和对比度的变化,BRIEF只关注两者的大小而不会对两者的变化有响应。

至此,特征点检测和建立特征描述符的相关内容就介绍完毕了。再次提醒一下,我们并没有按照SIFT、SURF、ORB等论文中的顺序进行介绍,而是将它们拆分成3个部分,即检测、描述和匹配,抽丝剥茧一一分析;同时,在讲解每个方法的时候都是按照这些技术的发展和动机进行的,很多细节和上述论文不尽相同。但是,在经过 5.5.4 的学习之后,相信你看这几篇论文已经是小菜一碟。请务必在看完论文后到OpenCV的源码中(大部分特征检测都在feature2d和xfeatures2d这两个库中,xfeatures2d在contrib内)了解这些算法的具体实现,或自己尝试复现这些算法以加深理解,纸上得来终觉浅!

当然还有一些其他的特征点检测和描述方法,如BRISK、FREAK、LDB等,我们介绍的都是一些2D的方法。特征点检测和特征描述符建立是计算机视觉中源远流长且仍然具有很大的活力的领域,尤其是对于视觉SLAM的方法,双目相机/深度相机/激光雷达的普及、算力的爆发式增长使得3D特征点检测和描述子建立有了长足而迅速的进步,由于深度信息的加入,使得新的不变性如视角不变性也成为了需要考虑的因素,点云(point cloud)处理可以算是这方面新的代名词。进来也有一些将深度学习、卷积神经网络等引入该领域的尝试,感兴趣的同学尤其是有志于今后从事SLAM等相关工作的,可以自行搜索了解上述内容。

这里推荐三篇关于图像特征匹配的综述,链接戳这里:图像匹配综述:特征提取、描述、匹配,点集配准,图像配准,立体匹配,图匹配,该作者的综述已经发表在IJCV2020上。


5.5.5. 特征点匹配方法

得到了特征描述子之后,自然要将两帧中的不同特征点进行匹配以方便进一步的操作,如计算两帧图片的相对位姿或进行图像拼接等。对于直方图描述子,一般计算特征之间的欧几里得距离,即范数;也可以采用其他度量以凸显/缩小特征描述子之间的差距(此处可以联系一下5.2.6.5处介绍神经网络损失函数的相关内容),给予差距较大的维度以更大的损失等。由于直方图也可以看作对特征分布的一种描述,因此还可以采用KL散度、Wasserstein距离、JS散度、MMD等计算两个分布之间的差异。对于LBP、BRIEF等二值描述子,大部分时候采用汉明距离来衡量描述子之间的相似性,由于使用了二值编码,在计算损失时的多样性就没有直方图那么丰富(玩不了什么花活)。笔者能想到的一个就是,当两个特征描述子在多个维度连续出现失配时提高匹配的损失(如果聪明的你动动脑筋说不定又是一篇paper!)。

brute-force暴力枚举是最简单的匹配方法,将所有的特征点两两进行check,损失最低的那一组视为匹配成功。此法对于两幅分别检测出m和n个特征点的图像,使用暴力匹配的复杂度是m*n,图像分辨率较高、选用的尺度较多的时候,这样的开销对于一些算力孱弱的设备来说显然是吃不消的。因此,一系列提高匹配效率的方法纷纷涌现。

时下最火热的方法是基于图的优化和深度学习方法,在SLAM中尤其是点云处理用的最多。可以将每个特征点建模为图的一个顶点,用随机游走、A*、D*等各种经典优化方法解决;大部分深度学习方法也是基于图的,如GNN、GCN、GAN(graph attention network,而非生成对抗网络)等,感兴趣的同学自行搜索。

而对于我们介绍的2D匹配,暂时不需要那么复杂的方法。下面主要介绍两个方向的优化方法,分别是machine-learning-based(加速)和align-based(提高精度)。


5.5.5.1. KNN/Cross Validation

K-Nearest Neighborhood即K-近邻,是机器学习的一种经典算法,这里假定读者已经了解此算法。和暴搜一样,在第一轮计算中,此法仍然会将所有特征点两两拉出来计算距离,不过,我们会保存距离该特征点最近的K个结果(一般设置K=2方便后续交叉验证,此处的“特征点”指的是第一重循环指定的特征点,来自图像A;“K个”指的是第二重循环指定的特征点,来自图像B);由于最近的K个中仍不乏距离远者,要设置一个阈值将相距过远者筛除。然后进行cross validation的第二轮计算:计算这K个特征点(来自图像B)各自的K个近邻(来自图像A),如果在交叉验证中,第一轮选定的点出现了在了第二轮计算中的K近邻里,说明这两个点很可能是一对匹配点。对于K>=2的情况,在进行交叉验证时一般选取两轮距离之和最小的点对形成匹配。

  • 看起来有点复杂,实际上就是正向匹配一次(确定A,找B中K个),反向匹配一次(B中K个,找A),如果正反都能找出相同的匹配点,就说明应该配对成功了。若反向计算的时候,第一轮的找出的K个点都无法匹配到第一轮的特征点,说明此特征点是out-lier(离群点、外点),即该特征点只在一幅图上检出或建立描述符时得到的特征不稳定,直接将其从候选点中筛除,后续的点进行匹配的时候就不会再和该点计算距离,从而降低一些时间复杂度。不过显然匹配复杂度仍然是m*n,并不会有数量级的变化。

  • 另一种常用的方法则是固定K=2,记测试点与这两则的距离为,其中。当满足如下两个条件时,视为配对成功:

即最近点小于阈值,且要两个点相对测试点的距离有足够大的差距;否则,将当前测试点从候选库中删除,后续不再匹配此点。此法可以通过特征之间的相似性差别筛除一些错误匹配点。

橙色来自A图,蓝色来自B图,橙色点只看得到蓝色点而看不到其他橙色点;反向验证也是如此,只需关注另一幅图像上的特征点

OpenCV将特征点匹配算法封装成了matcher类,其中就提供了KNN匹配的方法,详情戳这里


5.5.5.2. FLANN

在高维空间的大量数据中查找两个匹配的值是机器学习中的一个常见问题,实际上搜索引擎使用的技术也与此有关联,这里我们简要介绍一下FLANN。

FLANN是Fast Library for Approximate Nearest Neighbors的缩写,实际上它是一个用于快速匹配和最邻近搜索的库而非一种特定的算法,里面包含了以k-d tree(k-dimension)、k-means tree、HC tree(hierarchy clustering)等决策树为主多种经典的机器学习方法。特征点匹配实际上同样可以看作分类问题(single object for single class),因此非常适合通过决策树来解决。如果你有一定的机器学习基础,相信你在脑海里已经大致建立起了FLANN库工作的规则:将一幅图像(也可以是两幅图像,方便交叉验证)的所有特征点按照各种指标(类间方差、类内方差、Gini系数、互信息、信息增益......)建立一颗决策树,对于另一幅图像中的每一个点,都利用决策树来判断其类别;对于叶子结点有多个特征点的情况,可以采用取最小距离、交叉验证等方法获得最佳匹配。

当然,由于噪声的存在,通过FLANN得到的配对点有可能是“虚空配对”(一张图有一张图没有)或错配。但是由于树形结构的建立,筛选的时间复杂度一般都可以呈log级下降,特别是大型数据集下的匹配,FLANN能够获得1~2个数量级的速度优势,是一种典型的空间换时间策略。自然,OpenCV提供了对FLANN的支持,集成了部分FLANN算法,封装在FlannBasedMatcher类中,详情见此处。

在数据库检索中常用的一种技术LSH(local sensitive hash)也可以为匹配查找提供一些见解,感兴趣自行搜索。

5.5.5.1 和 5.5.5.2 都是对匹配速度的优化,必然会导致错误率有所上升。对于出现的错配点,可以通过接下来的介绍的办法消除错误匹配,提高精度。

不论哪里,都少不了深度学习。既然有特征描述子,那为什么不能给特征描述子也提取一下特征,得到描述子的embedding再对这些embedding进行匹配?


5.5.5.3. transformation/projection

上述的匹配方法已经在描述子的特征距离上做到极致,两个不同的特征点要是拥有相近的特征,似乎确实没有什么更好的度量办法将其区分。回想图像特征匹配的应用场景,我们一般是想要找到一幅图像中的某个物体在另一幅图像中的位置(物体识别),或是建立两幅图像的空间位置关系。如此看来,两幅图像应该具有某种变换关系,那么就由此入手来解决错配的问题。

这种方法被称作align-based,即基于对齐的方法。此类方法假设用于提取特征点的两幅图像是从不同位置/不同视角拍摄得到的(视角变化不太大,因为我们需要保证提取出来的特征不能有过大的变化这个前提条件),那么两张图片实际上可以用通过一个投影变换进行转化:

两张图像被不同位置的相机记录下来

而构成图像的基本单位,像素点(此例就是特征点),自然也服从上述的变化关系。于是问题就转变为,找到找到一个变换,使得该变换能够让一幅图像上的所有特征点和另一幅图像的特征点实现对齐。为了计算这个投影变换的参数,我们需要4个点对(投影变换矩阵中有8个参数,4个点对可以形成8个线性方程组从而求解)进行计算。因此在上述匹配结束之后,我们就选取一些点对通过投影变化,检查剩余特征点在坐标上能否对应,以去除错配点。那么这些点对应该如何选取?下一节揭晓答案。

在6.1.1.3,我们会用到投影变换,将所有角度的装甲板进行“视角归一化”,图像的in-plane、out-plane变换以及这些变换的性质将在那里介绍,可以先过去瞅一瞅。

从利用投影变换来检测物体的位置这个思路来看,也是很直觉的,以5.5.4 开篇的那张图为例:

a book to be found

我们希望在一些候选图片中找到上述物体,那么就可以为这些图片应用特征点检测算法并建立描述子,然后和上述图片中的特征点进行匹配:

随后,通过某些方法选取匹配点对,对剩下的点运行投影变换,若能对应上大部分特征点,则说明此物体在该图像中出现了。且倘若我们知道模板图片中特征点之间的真实3D尺寸,我们还能够通过反投影将候选图片中找到的物体的真实尺寸标出。实际上,一些经典的以图搜图算法以及图像索引库的建立(就和字典的字典序索引一样)就是通过这种方法完成的。此法也被称作基于特征点/描述符的模板匹配。


5.5.5.4. RANSAC & PROSAC
  • RANdom SAmple Consensus

即随机采样一致性算法,最早被提出用于解决从一组包含异常点(out-lier)和噪声的数据中正确拟合/计算出产生这组数据的数学模型或概率分布中的参数。而我们要解决的问题正是要将匹配点对中的错配点排除以找到正确的投影变换矩阵,因此恰好可以用RANSAC来解决。在此例中,图像B中的每个像素点都是由A中的像素点经过投影变换后得到的,投影变换矩阵就是我们要找到的数学模型,而两幅图中的特征点,就可以看作是从图像中经过某种特殊的采样得到的。

为了帮助理解,我们先看一个用RANSAC解决线性回归的例子。对于一组有较多立群点的数据(但它们实际上又不是那么离谱,无法通过阈值筛选的方法将其删除),倘若用常规方法如最小二乘可能导致解被外点带偏,迭代方法如梯度下降则容易引起优化过程中的不稳定甚至发散。而从概率角度出发的RANSAC则可以较好的解决这个问题(在某些方面有点像BRIEF描述子)。

左侧是待拟合数据,右侧是RANSAC得到的结果

对于线性回归的例子,数学模型是直线的斜率和截距,确定两个参数至少需要两个方程,作者在论文中特意提到,我们只选择最少的能够确定模型的样本个数。RANSAC通过如下方法进行拟合:

  1. 数据点中随机选择2个,并由此计算出直线的两个参数
  2. 由第一步得到的数学模型,用剩下的点对其进行检验,并通过拟合优度检验(MSE、MAE等,也可以通过启发式的方法设置一个容忍度并统计和模型相差以内的点的个数)来评估本次采样的准确性(相当于计算loss)
  3. 若拥有最高的拟合程度,则更新模型参数
  4. 若未达到精度要求或迭代轮数上限,回到1

实际上我们可以发现RANSAC根本没有在拟合数学模型,而是力图通过不断地“抽奖”,博取采样点都落在真实模型上的可能性,作者真是颇有赌徒心理啊。可就是这样简单的方法,却分外的好用。不过RANSAC未必可以得到最优的效果/正确的模型,从概率上来说,我们需要在迭代轮数和精度之间权衡(如果模型过于复杂,或者你的运气就是有丶背,很可能经过N多轮迭代也得不到正确的结果)

了解了RANSAC的原理,我们只需要把他运用在特征点匹配上就可以了,按照上面的流程,自己对号入座吧!

一种匪夷所思却异常有效的模型拟合精度判别方法是LMEDS,least median squares。RANSAC一般的判别标准是大部分点落入容忍度限制内,而LMEDS的准则则是选择一组使得这一组参数让模型在所有点上的误差的中值最小,LMEDS不像RANSAC一样有参数化的判别方法,其优化目标可以描述如下:

  • PROgressive SAmple Consensus

即渐进采样一致性,相比RANSAC对所有样本点都一视同仁不偏不倚的随机采样策略,PROSAC则对“好学生”有偏袒之心。如果我们在迭代之前能够掌握一些有关数据点的质量的信息,那么就可以对那些“成绩好”的学生多多关照一下。以特征点匹配为例,点对的相似性(点对的距离)就是度量样本点质量的一个很好的标准,点对的距离小一般就说明匹配得较准确,就比较不可能是错配点。因此PROSAC在处理前还要进行预排序,然后根据某种策略赋予不同排位的样本以不同的采样权重,之后的过程就和前述的四部曲一模一样了。

PROSAC依照的原理就是:具有更高相似性的数据更可能是内点,若用根据相似性进行排序的样本进行半随机采样(semi-random),得到的效果一定不会比完全随机采样的效果更差(下界)。

对于前面举的线性回归例子,怎么样判断一个数据是否更有可能是真实数据呢?聚类、决策树都是很好的方法!不过实际上在一些比较难定量计算相似性的时候,处理会更复杂一些,详见PROSAC的论文:Matching with PROSAC - progressive sample consensus

笔者认为还可以对PROSAC进行改进,得到D-PROSAC(dynamic,我自己取的名字),即在迭代的过程中,若某次采样的数据拟合出的模型已经比较好但是还没有达到要求(如目标是有90%的点落在邻域内,此次迭代有70%了),那么我们可以增加这一轮采样到的点或此轮迭代出的模型覆盖到的点的权重,让其在之后的采样中变得更重要。这只是一个直观的猜想,概率论学的好的同学可以帮我论证一下这个观点是否成立(笑)。

OpenCV在ccalib3d中提供了cv::findHomography以寻找两组特征点之间的投影矩阵,通过传递参数来选择RANSAC、PROSAC和最小二乘。

一篇比较新的论文指出,传统的特征匹配和projection trial往往被割裂开来,即先找到所有匹配,再在这些点对中尝试可能的projection,常用的方法就是一致性采样方法。然而实际上我们不需要那么多点,例如对极几何中我们只需要8对匹配点就可以估计两张图片的旋转和位移的方向。论文提出了“我们可不可以匹配与变换这两步结合在一起或者至少是分别迭代进行”的问题并给出了一种可行的方案。

这个工作的匹配和姿态估计使用的都是Transformer,这下连cv的基石都要被神经网络换掉🌶~

  • more about robust estimation

ransac和prosac这类一致性采样方法都属于鲁棒估计问题的一种求解方法,即数据中含有一些离群点,以及本例中完全错误的匹配点。鲁棒估计另一类求解方法被称作带权回归,本质上是加权优化问题(一般是带权最小二乘)。

有些地方也称第二种方法为m-estimator,但笔者查找资料显示m-estimator是统计学中常用的概念,表明当统计量趋向无穷时估计量会收敛到真值,这似乎和鲁棒估计没没有关联,感觉是讹误。若你有参考资料请联系我。

通过为给离群值施加额外的乘法,如Huber核、Thi-Sen核,以及各种惩罚加权函数、损失截断(给单个实例loss设置上限),使得离群值在该优化问题中所占的权重减小,从而让估计整体仍然保持鲁棒,不受离群值影响。这和我们在神经网络中为损失函数设计的优化非常类似:那里,我们给难样本的梯度以更大的权重,使得难样本产生的梯度压过大量简单样本,“强迫”网络偏向学习难样本;在这里则是降低离群值带来的影响。

一致性采样由于非常好用适用面非常广,凡是遇到模型的参数估计问题基本都可以作为SAC problem进行求解。因此在ransac之后也衍生出各种随机采样一致性的变种,如MLESAC、LO-RANSAC等,在速度、效率、inlier提纯度上都有改进。最终,所有的一致性采样性问题被流程化,推出了USAC(univeral SAC)。


5.5.6. 视觉词袋模型:基于特征点的物体分类/目标检测

在介绍特征点之前,我们提到了许多描述一张图片的特征的方法,这些方法一般都会对整个图片/ROI进行特征提取,得到某个对象的描述,它们的缺点在之前已经分析过。而特征点作为一种局部特征描述,可以获得目标对象更精简的信息,通过特征点检测除去哪些冗余的没有特征的区域,专注于有用的、有区分度的区域并为它们建立特征描述子,这可以看作一个两阶段的方法。

有了特征描述子之后,以图像检索为例,我们希望找到两张类似或相同的图片,那么应该如何匹配两张图片上的特征点呢?首先这与普通的特征点匹配问题不同,后者是在两个特征描述子之间查找配对,而这里则是对两个特征描述子的集合进行匹配。由于光照、曝光、尺度以及图像内容发生的变化,虽然我们的描述子有一定的不变性,但两个类似的场景生成的特征描述子及其数量肯定不会完全相同。更严峻的问题是,图像检索的对象是一个庞大的图像集,每张图像上都有大量的特征子,使用前述的两两特征描述子匹配的方法显然是行不通的。

为了降低匹配的复杂度,一种直觉的方法是将特征描述子分类。即为描述子的特征向量提出更高级的特征 ,或从某个角度来说,获得其的语义 描述,把相似的特征子归为一类(可以看作是一种结构化特征)。然而我们很难为特征描述子人为分类,怎么样的分类方法能够最大程度区分不同特征?怎么样的方法可以分出恰到好处的类别数?也就是说我们很难找到分类标签进行有监督训练。这时候无监督方法就派的上用场了,你一定已经想到一些最经典的方法,如K-means,mean shift,GMM等。

脸可以用一些特征集合描述,不同的脸上的五官相似但肯定不完全相同

当然,为了提高推理速度和训练速度,你还可以在聚类开始前运行一些降维算法以获得更精简的单个描述子特征。在训练集上训练好类别中心之后,我们根据距离最小原则将提取的特征描述子归入各个类别,再统计每个类别的描述子出现的次数,形成了另一个直方图特征——这就是基于特征点的视觉词袋模型。这样,我们就成功将原本的复杂度降低到二次方级别(假设都是用暴力匹配)。

统计每个类的特征出现的次数

词袋模型最早用于文本分类、摘要生成和文本检索任务,词袋的基本单位是单词或更高级一些的token。和图像相比,语言本就是信息密度较高的语义表示,因此一般不用提取额外特征。而图像中大部分地区都是无意义的背景像素或重复的单元,因此需要更高级的特征,防止有用的部分被无意义的部分吞没。这里,视觉词袋模型的基本单元是特征描述子的类别,在相同的元素在统一场景反复出现的时候,可以把描述子->描述子类别当作一种特殊的降维算法。

词袋向量的生成不要求两个输入有相同的描述子数量,也对图像的大小没有要求,这充分利用了特征描述子的各种不变性。其缺点是特征的统计不具有位置相关信息,即没有考虑上下文。在文本词袋中即使把句子打乱,生成的仍然是相同的描述向量。聪明的你可能会想到,也许我们可以为词袋直方图设置“主方向”......

既然可以用特征描述子生成词袋,自然也有深度词袋。在深度学习的时代,我们甚至可以直接将整张图片丢给神经网络让其输出一个描述向量,这在某种程度上和自编码器类似。除了这种端到端的方法,还有多阶段的整合了图像分割和目标检测的方法,通过生成图像中存在的物体分类、物体数量等具有结构化描述的特征,达到更好的分类效果。

5.5.7. 一个完整的特征点检测/匹配算法:以ORB为例

ORB(Oriented FAST and Rotated BRIEF)系列是当下在实时性要求高的场景中应用最多的算法。原始的ORB的特征点检测采用有向FAST,特征描述子选用旋转BRIEF,通过最邻近搜索完成匹配(论文中并没有说明具体是哪一种)。感觉对图像特征匹配已经比较了解的同学,可以尝试直接阅读原文:ORB: An efficient alternative to SIFT or SURF.在大型数据集上,实际上原始的ORB于精度和速度都已经超越了经典的SIFT、SURF等,后续一系列对ORB的改进更是让其如虎添翼,基本上是2D特征点领域的首选。

  • 特征点选取:FAST

即5.5.4.1.4中介绍的特征点检测算法,由于只需要进行像素值对比,速度是检测中处于第一梯度的。ORB中选用的是FAST-9,即对比像素在距测试点半径为9的圆上,因此具体实现起来和前述的FAST-4有所区别。另外,为了保证能够从任意尺度中搜索出特征点,ORB选择建立图像金字塔,不过没有说用了几层,下采样率是多少(一般是5层,1/2的下采样率,使用高斯模糊)。

  • 描述符的旋转不变性:oFAST

因为BRIEF无法提供旋转不变性,因此要为每个特征点确定主方向。ORB采用的是5.5.4.2.4中介绍的寻找图像patch质心的方法,也没说计算质心的时候使用的patch大小是多少......不过作者进行了消融实验,对比了质心标定、特征直方图和MAX(应该指的是patch中哪一个区域出现了最大的特征相应,如梯度、小波特征或者就是像素值,就将其于测试点的连线方向作为主方向),发现使用质心标定能够获取最好的效果,并且对于图像的噪声十分稳定,不同的特征旋转之后仍然有很好的区分度。

  • 建立描述子:steered-BRIEF

即5.5.4.2.6中介绍的特征描述方法。ORB在测试中加入了额外的细节,在建立描述符之前,先对图像进行高斯平滑;然后选取特征点附近大小为31x31的邻域,采样256对点通过比较建立描述子,每个采样点的值是其5x5范围内的高斯加权平均数。通过高斯窗进行平均可以使得特征更加稳定,降低噪声的影响。至于为什么是31x31,window大小为什么是5x5,作者表示我们在Pascal 2006上做过实验了!(狠狠堵住审稿人的嘴)

当然,为了提供旋转不变性,在提取特征之前,要先将图像采样patch旋转到主方向的位置。ORB加入了旋转预处理的操作,将oFAST确定的主方向离散到以30度为一个单位的12个方向,提前计算好961个点在经过度的旋转后会到达的位置形成查找表(look-up table)。这样,在建立描述子的时候,就通过LUT寻找离和主方向最接近的那个位置(实现为取模),常数时间就可以算出旋转后的位置,不需要每次都计算矩阵乘法(2x2)。

30°为一个单位在速度和精度之间取得权衡,而且由于对比像素值时高斯窗的加入,可以进一步提高对近似带来的噪声的抗性
  • 通过ML方法改进描述子:rBRIEF

rotate-BRIEF即rBRIEF是作者改进后的描述子,在下方的几组实验中也一起出现作为对照

BRIEF一个令人满意的特性是由它生成的特征描述子每个维度拥有接近0.5的均值(意味着所有生成的特征点中,在每一位上0和1的数量几乎持平,是对所有描述子的某个维度的纵向分析),这说明了BRIEF描述子的方差很大,不同的描述子之前有很好的区分性。但通过oFAST引入旋转不变性后,有些维度的均值会偏向0或1,这就说明此维度对匹配的贡献下降了,两个不同的描述子很可能会在这些位上有相同的值:

三种不同方法产生的均值分布,其中X轴表示的是距离均值为0.5的距离(绝对值)

除了让不同特征描述子间的位均值接近0.5,我们还希望建立一个特征描述子时,使用不同的测试对(即生成一个描述子时,在31x31patch上不同点对产生的特征,1位,这个测试要进行256次以形成256维特征)之间的关联性越小越好(我们肯定不希望两次采样都碰到相近的区域,可想而知这样会导致两次测试生成相同的结果)。为了分析单个BRIEF特征(就是256维向量,要找不同维度之间的相关性,对一个描述子的横向分析),作者对100k个keypoints运行了PCA并将不同分量的特征值大小进行排序,得到的结果如下:

对三种方法的产生的描述子进行PCA分解得到的特征值分布

可以发现三者的前几个特征值都很大,而BRIEF和Steered BRIEF的特征值在第五个分量之后迅速衰减,后续的特征值都非常小。回想PCA之后得到的特征值的含义,倘若特征值分布得越均匀说明有效的分量个数越多,真正有利于分类的特征越多。经过改进后的rBRIEF显现了较为平缓地下降,后续特征分量对于区分特征点的贡献要稍大于前两者。

于是作者从增加特征描述子内的不同维度差异(让产生一个描述子的不同维度拥有最大区分度)和同一维度的特征有最大方差(均值为0.5,这时候方差为0.25)两方面入手,通过统计的方法找出满足上述两个条件的描述子建立方法:

  1. 建立了一个包含大约 30 万个关键点的训练集,这些关键点来自 PASCAL 2006中的图像

  2. 列举了31 x 31邻域中提取的所有可能的二进制测试,即。每个测试都是邻域内的一对 5 x 5 patch,因此实际的数量是。所以对于31x31的窗口,最终得到205590 个可能的采样方式

  3. 针对所有提取出的所有特征点上述所有可能采样方式的测试

  4. 对于所有采样方式得到的结果(每种采样方式都要运行30万次),按照该采样方式对所有关键点进行测试得到的均值和0.5的差值的绝对值,进行排序,得到一个矩阵T(每种采样方法对应的是四个值,表明采样的两个点的位置)。这一步解决的是sBRIEF会减小某个维度特征的方差的问题,我们就选出那些均值接近0.5的选点策略

  5. 贪心搜索:

(a) 将排序后的第一种采样方式,放入结果 R 并将其从 T 中移除

(b) 从 T 中获取下一个测试,并将其与所有 R 中的测试进行相关性计算(即和剩下的结果都进行异或)。如果其绝对相关性大于阈值(零的个数太多),则丢弃它;否则将其添加到 R。这一步解决的是不同特征之间的差异性问题,选取那些相关性最小的选点策略

(c) 重复上一步直到 R 中有 256 个测试(256种选点策略)。如果少于 256,则提高阈值并再试一次

改进之后,显然是很有效的,作者还对比了三者在进行匹配时计算得到的汉明距离的分布:

中间的三个高斯分布是没匹配上的点,左侧是匹配成功的点

可以发现,sBRIEF的那些没匹配上的点有明显的右移趋势(毕竟用了LUT近似角度,旋转之后算可能还需要对像素进行插值,精度下降是必然的),导致内点和外点出现重叠而无法区分;rBRIEF就大大缓解了这种趋势,只稍稍右移了一点点。自然,改进之后在建立描述子的时候就不是随机选点了,而是按照机器学习方法得到的256种固定的方式进行采样,一方面提升了性能,另一方面还剩下了随机采样的计算开销,何乐不为?

本来还想说作者自己立靶子自己打,在相同的数据集上训练采样策略又在上面测试,颇有一种”过拟合“的味道。为了凸显自己方法的优势,作者还对将Pascal 2006数据集上学到的采样策略应用到其他数据集上,发现效果仍然很好。可见充分的消融实验是多么的重要。作者还对三种BRIEF进行了各种其他测试和可视化,画出来的图也是非常精美,论文确实值得一看。

还有一件有趣的轶事,ORB是OpenCV Lab发布的算法,据说其开发团队对于SIFT申请专利的做法表示很不满,因此决定不仅要开源一种有效的图像特征匹配算法,速度、精度上还要远超SIFT!

OpenCV自家的东西肯定是整得明明白白的,直接把检测和描述两个阶段合二为一,封装为了cv::orb,例程见OpenCV: ORB (Oriented FAST and Rotated BRIEF)

一些结构光相机就是装配了一个feature pattern投射装置和一个相机(一般来说是红外的),相机端通过检测pattern完成“匹配”(原理上和双目非常相似),从而计算“视差”,得到投射点的深度。


5.5.8. handcraft vs Deep feature:总结

由于使用深度特征是大势所趋,而且CNN在很大程度上似乎并不需要传统计算机视觉的相关知识,我们把基于CNN的目标检测和神经网络的介绍放在了传统的手工特征前面。不过经过 5.5.4 和 5.5.5 的学习,相信你已经发现了现代目标检测和图像分类中一些trick和hand-craft特征有着千丝万缕的联系。虽然LeNet-5在90年代就已经被提出,但CNN真正接过手工特征的交接棒,还是在AlexNet发布之后。处于两者间暧昧不清的时候,DPM(Deformable Part Model)可以算是时代的眼泪,它在手工特征残存的最后岁月里,将专家知识发挥到极致(当然也可以说是一款”学术“过度的产品,为了工程部署增加了大量复杂的细节和实现trick,多少有点现在的ConvNetX、YOLOV6那味了...),其可视化效果和基于CNN提取的深度特征非常相似;而且它的可形变模块的设计,应该就是形变卷积的前身。有兴趣的同学可以自行搜索DPM的相关内容。

描述子特征的归一化变成了现在的各种norm;图像金字塔变成了FPN;光照/对比度/旋转/平移/尺度不变性在某种程度上变成了数据增强;直方图特征进化成了softmax;小波特征、傅里叶变换成了注意力机制...... 如果花一些时间,你应该能找到它们之间更多的联系。即使是现在,传统方法仍在不同的细分领域、不同的角度启发网络的设计。当然,也有一些方法被神经网络弃之不用,如深度特征已经替换了几乎所有手工特征,描述子建立甚至可以直接通过基于神经网络的编码器进行(直接把邻域的整个patch作为输入);神经网络网络的端到端特性使得特征的提取和特征的匹配、识别融为一体难以拆分。传统方法并不能被称作老旧或过时,就像学习数理知识,若对按照科学史的顺序进行,能够很好地帮助你理清不同方法和理论发展的联系;手工特征的可解释性也是神经网络值得借鉴的地方,目前已经有大量的科研人员投入对神经网络内部的动力学研究。

个人认为,在神经网络的黑箱疑云被揭开之前、在层出不穷的启发式进步尚未停止之时,两者仍然将长期共存。况且Transformer这批黑马已经以迅雷不及掩耳之势异军突起,和前两者形成三足鼎立之势,颇有吞并CV和NLP的意味,甚至完成多模态的统一大业。因此,按照学科发展的顺序学习计算机视觉是非常有必要的,跟紧学术的最前沿同样是必须的。最重要的是拥有坚实的数理基础,并在如大山一般的书海中找到你所需要的珍宝,和它们之间的联系。

在8.1,笔者会就此问题进行更深入的探讨,从个人水平出发谈谈对计算机视觉的理解。



5.6.三维视觉基础

现实世界在空间上有三个维度,而照片却是二维的,图片在成像过程中损失了一个维度:深度。从仿生和类比的角度来说,人眼也能观察到三维信息,这是因为它是一个双目视觉系统。再加上人在基因中、成长过程中习得的大量先验信息,在大部分时候我们都可以轻松辨别物体的远近。为了更好地为下游任务服务,我们就要想方设法让计算机也能理解图像中的深度信息。

5.6.1. 坐标系变换

这里只是介绍如何描述三维空间中的刚体变换但是并不涉及它们为什么是这样的形式的解释,要进一步学习请自行查阅相关资料。

这里我们简要介绍一下三维空间中的两种刚体变换和它们的组合:平移+旋转=位姿变换。三维空间内任意两个刚体系都可以通过如下两步变换实现。首先是平移,令初始坐标原点为:

平移之后得到:
把旋转矩阵拆成绕三个轴的旋转:
将旋转施加到平移后的坐标系,得到变换结果:
由于平移变换不是三维向量空间中的线性变换因此无法用矩阵表示,为了方便起见我们增加一个额外的维度形成齐次坐标系,用四维矩阵描述三维空间中的任意刚体变换:
是一个3x1的向量,则是3x3的旋转矩阵,自己动手写一下,按照矩阵乘法进行展开,能够加深理解。

简要直观地讲解一下为什么三维空间中的平移不是线性变换。首先你需要理解线性变换的概念:线性变换是向量空间到其自身的线性映射。线性映射需要满足两个条件:

其中为向量空间中的任意两个元素,是任意大小的矢量。一种最简单也是最”定义“的角度就是,平移无法用三维方阵描述,因为常数矩阵是描述线性变换的工具,所有线性变换都可以表示为矩阵的形式(矩阵就是因为描述线性变换很方便,才被发明出来)。不开玩笑,我们从线性映射的定义出发,理解平移。为了方便起见同时又不失一般性,我们采用二维空间中的平移帮助理解。为了满足第一个条件,首先对其上的两个向量分别施以平移变换(记为),得到等式右边;等式的左侧为,等式右边的变换项却是!用二维笛卡尔坐标系可视化一下:

怎么画出来就不对了呢?这是因为虚线是仍然在原坐标系下进行矢量相加得到的结果,而实际上在经过变换之后,原点已经跑到处了。实际上,平移变换导致向量空间中的零元发生改变,而线性变换需要保证零向量在施加变换之后仍然是零向量,但在这里,变换后的坐标系原点已经跑到原坐标系的处!为了保证线性空间的性质,同时又能利用矩阵这个工具进行计算,我们加入额外的一个维度,让平移变换在新增的维度上进行,从而移除对原点的影响。


5.6.2. 相机成像模型

有了坐标系变换的基础,就可以学习相机成像模型了。物体投影到相机的光线传感器的过程,仅仅是在刚体变换的基础上增加了一些缩放。4.1.1已经介绍了相机的小孔成像模型,对于我们的应用场景来说这个近似已经足够,因此下面在进行变换的时候都把镜头等效为一个小孔。另外,在小孔模型下,像距和焦距相等。

像平面(相机的感光单元平面)上成倒像,将其竖直镜像后放置在光心前方一个焦距的位置方便计算

现在用数学化的语言描述物体上每一个点在像平面上的对应关系:

相机主点即相机光心/镜头的光学中心

利用中学的几何知识就可以知道,这里有两个相似三角形和,显然实际物体上的每一点和像平面上的物体都符合这个关系,用等式描述,即:

在这里,我们假定两个方向的焦距相同,用矩阵来描述这种关系,也就是:
这时的坐标原点是相机的光心, 是像平面上点的坐标,是物体在相机系下的坐标,在投影后将被压扁而丢弃。由于使用惯例和方便起见,图像在计算机中保存时原点为左上角,而刚刚图像坐标系的原点则是在CMOS的正中心。另外,CMOS上的感光单元都是离散的,因此这一步转换涉及坐标系的平移和数值的离散化:

O'就是CMOS中心的位置

一般都为CMOS长宽的一半,令为一个像素在水平位置和数值位置上占据的单位长度(就是一个感光单元的尺寸),写成矩阵乘法形式:

得到的就是最终在图像坐标系下的位置了,以OpenCV为例,访问该图像Mat的(u,v)位置,得到的就是上述变换后的像素。

最后,因为相机常常不是固定的,而是在空间中旋转、移动,我们希望将所有信息的处理都转换到绝对参考系下,因此还需要一个相机本体到绝对系的位姿变换矩阵:

所有的变换都已经介绍完毕,串在一起,看看一个物体从世界坐标系(绝对系)到像素坐标系的转换过程:
红色是物体从世界坐标系变换到相机坐标系的位姿变换矩阵;蓝色是根据小孔成像模型将物体从三维”压扁“成二维的伸缩变换矩阵(没有旋转,没有平移,但在x和y上都乘上了缩放系数f,即焦距),即相机坐标系到图像坐标系;绿色是图像坐标系转到像素坐标系的平移(图像原点在左上角)+缩放矩阵(离散化);紫色是在投影过程中丢失的维度,也是比例缩放系数(离得越远的物体越小,越近越大)。

红色矩阵内的参数也称作相机外参,蓝色和绿色矩阵则是相机内部的变换信息,称作内参,一般通过标定得到。标准化的相机(啥也不能动,固定好的)则由厂家提供内参(当然自己要再标定也可以,一般没有厂家给的准)。

在4.1.1也介绍了,实际上在成像的过程中,镜头由于种种原因会产生畸变,不能得到蓝色矩阵这般简单的变换,因此需要通过标定来确定畸变的类型和畸变参数以最大程度消除其影响。

如果你已经学习过线性代数,可以发现相机的小孔成像模型就是将世界坐标系下的点转换到相机坐标系下,与线性空间的映射和变换的概念类似,即在不同的基下表示相同的对象。然而投影变换是一个不可逆的由高位空间到低维空间的变换(检查投影矩阵的秩),因此在该过程中丢失了一个自由度的信息。


5.6.3. 相机标定

相机标定一方面是为了确定相机内参(主要是焦距和光心偏移大小),另一方面就是为了最大程度减小成像畸变带来的影响。另外,由于镜头安装的时候无法保证机械位置完全对齐,这都是标定需要考虑的问题。

对于小孔成像模型,影响最显著的两种畸变是桶形畸变和切向畸变。直接看图便知:

由于透镜制作工艺不能保证边缘的光线平行进光轴进入后仍然射向焦点,因此产生桶形畸变;来源未知

径向畸变以某一个中心往外延伸,距离此位置越远,畸变越大。两者呈一复杂的非线性关系,一般对其进行泰勒展开并取前数项进行近似。使用最多的是下面这种近似:

与之对应的是凹透镜成像的时候出现的枕形畸变,三种图像拍摄棋盘格得到的图像如下:

完美透镜、桶形畸变、枕形畸变;图源Wikipedia

患有近视的同学在验光或选购眼镜时常会听到销售人员向你推销一种”非球面镜“,这种技术就是通过改变镜片边缘材料的折射率或通过特殊的光学造型避免离光心较远的位置的畸变,对于凹透镜就是上图中的枕形畸变。

切向畸变则指的是下面这种情况:

由于贴装工艺问题,感光芯片在封装时可能无法和镜头平行放置,导致成像畸变的发生;来源未知

这种畸变就导致图像上的不同处与真实物体无法满足 5.6.2 中介绍那种简单的投影关系,而将与CMOS贴装的倾斜程度有关,一般用下式描述这种畸变关系:

需要通过标定确定两个参数。

关于相机成像过程中可能产生畸变的详细推导和该领域的发展,有兴趣的同学可以参考这篇综述:Camera Calibration - an overview。

最后,由于安装误差,可能导致镜头的光心和CMOS的中心点没有对齐,因此还需要确定两者偏移的距离。相机安装时的歪斜统一计入前述的切向畸变中。

至于标定的具体操作,这里提供一个OpenCV的例程讲解,参见:OpenCV相机标定完全指南(有手就行)。Matlab中也提供了视觉工具箱,通过GUI界面可以更方便地设置要标定的畸变类型,操作起来很容易。

鱼眼相机和360°相机的成像模型则不如小孔那么简单,在全向感知模块上可能会使用这些相机。当然现在也有大量的投影模型被提出用于标定,感兴趣的同学可以自行搜索。


5.6.4. 重投影和反向投影

重投影是用于检测成像模型和相机位姿估计是否准确的工具,在相机标定中常用于衡量得到的结果内参是否准确,而SLAM中常用于机器人本体当前位姿的估计和解算,重投影还可以用于可视化三维解算的结果;反向投影则是进行单目视觉解算的关键之一,经典的单目PnP算法就是基于反向投影实现的。

5.6.4.1. 重投影:re-projection

以标定相机为例,我们通过下面的流程验证标定结果的准确性:对于一真实物体中的某个点,我们已知其相对相机的位置如何,而相机内参已知,即我们用5.6.2中叙述的成像模型加以修正之后对该点进行变换,得到结果点;而实际在拍摄的时候由于小孔成像的近似和畸变的不完全矫正,的位置肯定无法和重合,两者在像平面上的距离误差就是重投影误差。(模型是我们给相机成像过程建立的一种描述,可以看作一种近似,必然和真实光路有差别)

OpenCV的例程在每次完成标定的时候,都会输出标定结果在用于标定的图像上的重投影误差的MSE(以mm计):

一般认为此值小于0.1为可以接受,若要进一步提高精度,应该使用更多标定图片并从不同角度拍摄标定板,还可以尝试提高对畸变拟合的阶数等。

真实世界中的值是怎么知道的?在进行标定的时候,你需要设置标定板的参数,如棋盘格尺寸、大小、每个格点之间的距离等。因此实际距离已经给定。

重投影误差的构建是视觉SLAM估计帧间姿态、滑窗和回环优化的重要环节,常常作为姿态估计问题的优化目标。

可视化就很简单了,假设你通过某些算法得到了三维坐标系下一些点的位置(比如3维目标检测、3维测量等),想要在图像上找出相应点,或者画出具有立体感的轮廓,就可以利用标定得到的内参对这些点进行投影,从而其获取在当前图像上的位置。


5.6.4.2. 反向投影:back-projection

反向投影是视觉测量的基础。顾名思义,反向投影即相机成像的逆过程,将一幅图像还原成世界坐标系下的物体点。显然,由于在正投影的过程中丢失了深度信息,我们在反向投影时应该添加约束条件以得到唯一解。

远小近大——这是最符合人类认知的一种测量。假设我们能够保证某个物体始终在像平面上形成正向投影(中心线和光轴对齐或平行),并且其真实尺寸已知,那么根据其上某可以被定位的两点间在像平面上的距离和真实距离之间的比值,由小孔成像时两者以焦距的比例就可以得知当前物体距离相机的远近。这时候选用的约束条件是物体始终正对相机不发生out-plane的变换。

另一种方法是添加等式约束,即用多个已知真实相对位置的点在像平面上的投影关系(一个真实点和一个像平面上的投影点形成一对约束,一个等式)确定相机和这若干个点之间的距离,这就是经典的PnP法(Perspective-n-Point)。实际上每添加一对点都会减少解的可能数量,如果不添加约束,我们一共需要6组匹配点12个方程来确定变换的所有参数。

  • 直观理解P3P的求解

我们这里就先从添加约束而不是代数分析的角度,分析P3P方法。P3P要求已知共面三点之间的位置关系(不能有点共线),从而得出物体到相机的位姿变换(即旋转矩阵和平移向量)。

当只有一对投影关系约束的时候,反向投影的结果可以处在以像素点为起点、和真实点连线构成的射线上的任意位置:

加入第二对投影点,反向投影的结果会被限制在一个范围内,但仍然有无数组解。约束即真实点之间的距离(图中绿色圆的半径),确定下方紫色点为第一个解后,绿色圆与上方射线的交点都是第二个点可能解的位置,因为只要保证两点的距离为半径R:

加入第三个投影,反向投影的结果被继续约束,这时候最多只有四组可能的解。首先在一条射线上确定一个点,然后作半径为该点到第二个点真实点的距离的球,并在该球与第二条射线的交点(1个或2个,取决于圆的半径和的位置)上建立第二个球(半径为第二个点到第三个点的真实距离),会与点第三条射线产生交点,最后以为中间建立一个半径为第三个点到第一个点真实距离的球,将和第一条射线产生交点。不断移动在第一条射线上的位置,直到和重合,这样就形成了一个面,并找到一组可能的解。

加入第三个投影后,实在画不来图,如果你有画3维动图的好方法,请告诉笔者。倘若你熟悉一些3维CAD软件如Solidworks的操作,可以作好后发给笔者,感激不尽。

abc是像素点;ABC是真实点,其边长就代表真实距离

实际上也可以把这个问题看作如何将三角形ABC不留缝隙地放入以三条射线构成的无限四面体柱子内。将这个柱子的顶点朝下放置,直观看来应该有4中情况满足上述条件:1. A处于稍下的位置,BC靠上;2. B处于稍下的位置,AC靠上;3.C 处于靠下的位置,AB靠上;4. ABC处于同一高度。

  • 代数方法求解P3P

有了直观的理解之后,现在用代数的方式构建约束方程组进行求解。还是上方的那张图,我们给每条边分别标注参数:

真实点和投影点的关系;图源https://jishuin.proginn.com/p/763bfbd69afc

根据余弦定理,我们可以得到如下关系:

在这里,我们有3个二次方程3个未知数,为了方便求解,这里采用替换的技巧。我们假设,再定义,方程转变为如下形式:
这样就可以消去,再用三个真实点之间的关系消除:
求解这两个二次方程,就可以得到P3P的解了。两个二元二次方程最多有四组解(排除无解的情况)。有兴趣的同学可以自行学习二元二次方程组的求解方法。得到四组解之后,一般再加入一组对应点用于检测,即在5.5.5.4介绍通过随机采样一致性验证投影矩阵那样,分别求出四组解对应的变换矩阵,然后对额外的那个点施以四种变换,求得变换结果与真实对应点相差最小的那个,即是我们要找的点。其实这就算是一种刚刚讲过的重投影方法。

  • 编程接口

OpenCV中提供了SolveP3P()和SolvePnP()方法,前者会返回四个结果(四个变换矩阵)让你自己进行后处理,后者则支持一系列的不同求解方法并返回一个单独的结果,必须要传入四个以上的点。至于如何通过传入不同的参数选择求解方法,参见:OpenCV: Perspective-n-Point (PnP) pose computation

而其他的PnP方法包括P4P、EPnP等,以及对点和点之间关系的不同如共线共面等都会产生不同的解法。它们通过近似、迭代、试错等方法利用前述的点之间的相对关系行求解。更多PnP的信息,可以参考这篇综述:A Review of Solutions for Perspective-n-Point Problem in Camera Pose Estimation

对于三维空间中的位姿解算问题,这里有一篇综述详尽介绍了包括单目、双目等在内的各种方法:Pose Estimation for Augmented Reality: A Hands-On Survey


下面三节内容是可选的。

5.6.5. 旋转描述和四元数

在 5.6.1 中介绍的旋转描述是旋转矩阵,由于其特性所有旋转矩阵的集合被称为特殊正交群。常用的旋转变换描述方法还有欧拉角(euler angle)和四元数(Quaternion),以及轴角表示(angle axis)。使用欧拉角描述旋转非常方便,不像旋转矩阵那样需要这么多的参数,但是其存在一个严重的问题,相信大家都有所耳闻——万向节死锁(gimbal lock)。关于此问题的解释,笔者认为Ele实验室的这个视频:无伤理解欧拉角中的“万向死锁”现象,应该是初学者比较容易接受的一种,没有复杂的公式推导,直观且严谨。之所以会出现这种情况,本质上是其定义导致的,我们要求按照一定的顺序对原始坐标系进行变换(众所周知变换不是可交换的),这就添加了某种约束,最终体现为死锁的出现。

另外,在学习姿态变换的时候需要注意矩阵左乘和右乘的区别,这可以从基变换的角度进行理解(一定要好好学线性代数!)。欧拉角变换会让坐标轴也跟着变换,旋转变换的定义将基于新的坐标系;而旋转矩阵则是固定坐标系,旋转变换的定义是基于原坐标系的。这也是变换矩阵乘法顺序的来由之一。

而四元数从定义上避免了死锁,从名字也可以看出它有四个维度(虽然仍然按照一定的顺序进行旋转,也有约束,但是增加了一个额外的维度,就像齐次坐标变换为平移增加了一个自由度使它变成其下的线性变换)。一种自然的理解方式是通过绕任意轴旋转的变换,即罗德里格旋转公式(Rodrigues' rotation formula)。四元数利用一些定义使得其运算规则满足前者,并且避免了三角函数的计算,这也是四元数描述旋转的另一个优点:运算量小,非线性函数在程序中的开销是比较大的。这篇教程:A brief introduction to the quaternions and its applications in 3D geometry是笔者认为写得最好的,强烈推荐学习。

5.6.2 就讲到相机相对世界坐标系会有一个位姿变换,其中的姿态变换(旋转矩阵)常常是我们要得到的信息,这里就会用上各种旋转描述。以SLAM为例,机器人搭载的相机将会随着机器人的运动而发生姿态变换,我们就要通过帧间的特征点位置变化求解相机位姿的变化;对于搭载IMU的机器人,还可以通过IMU得到的数据获取机器人/相机的位姿变换。再如我们希望在不同帧中确定一个移动物体在世界系下的位置,而相机为了跟踪该物体会随动,两者的位姿都在发生变化,这时候为了确定移动物体的轨迹,就需要通过前面介绍的相机成像模型和相机位姿变换以完成解算。

5.6.7 中我们会引入更多的旋转描述和姿态描述方式,并且剖析它们之间的差异,以及使用不同描述方式的动机。


5.6.6. 点云处理和PCL

点云(point clould)是一种描述物体三维信息的方式,在三维空间中的一大片点距离相近,所以被称为云。相对于二维图像它没有那么稠密(否则描述三维空间中的每一个点增加的空间是指数级的),但却拥有的额外的第三维信息:深度。产生点云的方式有很多种:单目相机(需要移动)、双目相机、深度相机、激光雷达,甚至声波探头。点云中的每个点除了包含三个描述位置的信息,还可以涵括RGB值(一般由可见光相机产生)、xx反射率(一般由主动式传感器获取,如激光/超声波反射强度)、采集时间戳或标号等数据。

采集得到的点云常常也包含噪声和外点,因此和二维图像处理一样,点云滤波也是必不可少的操作,只不过增加了一个额外的维度,基本算法如高斯平滑、双边滤波等仍然可用;由于三维点稀疏、不连续的特性,也加入了RANSAC、格点滤波等方法。同样,在点云中我们同样可以执行特征点检测的任务,由于点云数量相比二维图像像素少得多,我们常常可以用数个三维特征点的描述符构建一个三维物体描述符,这可以很好地对原始点云进行描述而不引入过多的噪声和背景信息。找到了特征点自然可以进行点云配准,这和二维图像特征匹配时一致的。剩下的很多任务都可以从二维图像迁移到点云上:点云分类、点云目标检测、点云分割...... SLAM中的点云特征处理如平面和线拟合更是其流程中核心的一环。当然,点云的一个非常重要的应用就是三维重建。只要你对二维的图像处理和视觉算法有大概的了解,学习三维处理也不会有太大的门槛。

和OpenCV一样,点云处理自然有开源的大规模算法库:PCL。PCL发布于2013年,基于C++编写意味着其拥有极高的性能,SLAM和3D视觉的火热更是促进了其快速地发展,其功能基本涵盖了所有点云处理的基本操作,内部包含了大量的点云处理算法,包括Features,Filtering,Key-points,Tree,Range Images,Recognition,Sample Consensus,Segmentation,Surface,Visualization等功能模块。官方文档参见:Introduction — Point Cloud Library

可以直接看官方文档入门,内部提供了大量Tutorial。笔者推荐观看哈工大竞技机器人-视觉竞培营-PCL作为点云处理的入门。

在下面的SLAM章节我们会再谈点云处理,尤其是激光雷达生成的点云。



增补:笔者也成为研究生,目前关注的方向是机器人中的状态估计问题,因此增加此章节浅谈自己这段时间的经验和体会,主要是overview的角度。SLAM系统构成复杂各个方面都需要基础知识,本科的高数三件套尤为重要,还外加一些机器人学和微分流形(主要是工具),以及非线性优化的内容。初入RM的同学,特别是三件套还没学完的同学建议跳过这部分。若阅读完《视觉SLAM14讲》等,已经对slam有了大致的了解,可以来这里看看笔者的碎碎念,说不定对你有一定的启发。

5.7.初探SLAM系统

根据slam的全称就可以看出其设计目的。我们在第二章介绍哨兵的时候其实就已经大致分析了机器人为什么需要知道自己在空间中的位置,以及自己周围有哪些障碍。生活中和我们最密切的使用了SLAM系统的产品包括扫地机器人、AR眼镜和自动驾驶汽车(包括辅助驾驶技术)。

更多关于SLAM的用途和研究动机,请参考《视觉SLAM14讲》的前言和概论部分,这里不再赘述。

5.7.1. 状态估计问题

为了简化问题描述,这里我们提到的状态估计主要是机器人的位姿估计以及地图点的位置(landmark)估计,暂时不考虑高级结构化地图如mesh和volume,还有语义地图等。

SLAM的核心问题和名字相同:

  1. 根据传感器的数据判断机器人的位置和运动状态
  2. 根据传感器的数据构建局部地图

第一个问题是对机器人的位姿的估计,第二个问题则是对机器人周围环境信息的估计。

状态估计最早来源于控制理论,指的是根据可以获取的量测信息,对系统的状态进行测算的问题。比如我们之前接触过的卡尔曼滤波器,就是一种状态估计器(观测器)。在slam问题中,可以把两帧传感器数据之前机器人发生的位姿变换作为待求的状态,通过传感器的观测我们要还原出这些状态。而地图点的估计问题与之同根同源。控制理论出身的人研究SLAM问题,常常考虑状态的能观性和估计值的收敛速度。

从概率的角度分析,则一般从贝叶斯估计的观点出发,在给定一些前提(先验)的情况下,找到能够使获得的观测被观测到的概率最大的那个概率分布(有点绕噢),即最大后验估计。这种方法还能够估计出传感器观测误差(或协方差)的大小以及当前状态的不确定性。

另一种理解则是,传感器在两帧和多帧之间的数据,可以通过处理后产生关联(常说的data association),根据这些关联,我们从理论上可以构造许多约束等式或不等式。显然实际数据存在误差,我们希望根据观测数据构建的这些式子能够最大程度地满足约束。为了定义程度 的高低,我们需要某种度量或判断准则,这就自然而然与所谓的 最优 关联到一起。在引入一些先验的情况下,该带约束的优化问题很多时候会和最大后验问题殊途同归。

我们发现,控制理论中的状态观测问题需要一个模型,但是我们并不知道SLAM的传感器系统是如何运动的,所以一般还会给运动系统一个匀速运动与转动的运动学假设,同时估计机体的转动和平动速度。如果已知slam系统是安装在某个特定的平台如AGV小车或足式机器人,那么便可以引入机器人的动力学模型,通过该模型推断机器人的状态先验,再用上面说的额外传感器观测进行融合、进一步优化,求取精确状态达到更好的效果。

5.7.2. SLAM系统的一般构成

第一个问题被称作里程计(odometry),或者tracking问题,大部分工作在slam系统的前端完成。一般只利用相邻两帧(也有维护一个动态窗口)的传感器数据,计算出机器人当前的姿态。由于机器人需要时刻知道自己的位置(在slam系统中一般是相对位置),所以前端对实时性的要求较高,需要一个高频的里程计输出,考虑太多数据则没法达成这一目标。为了构建两帧之间的优化问题,tracking需要在两帧传感器数据中得到数据的关联、对应关系,从而构建约束优化问题以求解位姿。

第二个构建局部地图以及更大规模的优化的程序则被称为后端,后端同时还会考虑更多帧的传感器数据,对之前的轨迹和地图点进行更精细的优化。后端一般会构建额外的线程运行,不要求实时,而是对精度由更高的要求。一些算法还会评估当前前端的负载,在cpu有较多空余时执行后端优化,确保前端的实时性。

为了提升大范围场景的SLAM效果,完整的SLAM系统还包括回环检测(loop closure)模块,让机器人知道自己回到了之前经过的位置,根据这个信息构建额外的约束,对系统进行优化以缩小长时间运动带来的轨迹漂移误差。为了不影响前端的实时性,回环检测的执行频率较低。实际上我们也没有必要时刻执行回环检测:1. 移动范围较小时不会产生很大的误差 2. 即使长距离移动触发了回环,也只需要在这附近触发一次即可。也就是说没有必要连续触发回环优化,原因同1,即经过一次回环优化后,可以认为误差已经降低许多。

注意回环检测和重定位是不同的技术。loop closure是在未知场景中确认机器人两次到达了同一位置;重定位则是已经有了一个先验的/全局地图,根据当前传感器给出的信息,确定自己在全局地图中的位置。有一些地方说回环检测是在SLAM实时构建的局部地图中确定自己在局部地图的位置,这是完全错误的。截至目前,回环检测大部分情况下不依赖于先验地图,一般是通过提取关键帧的一些信息形成的特征描述(比如之前介绍过的词袋模型或干脆就是给当前获取的传感器信息生成了一个特征向量)进行匹配完成的。

5.7.3. 为什么使用李群\&李代数

这部分内容可能有助于你建立使用李群和李代数进行优化的直觉。

位姿(pos&rot)是SLAM中最重要的需要被优化、被估计的变量。通常的优化问题存在一些约束,对于向量空间中的变量的约束,我们很容易通过拉格朗日乘子法或线性规划将其转化为无约束的优化问题,进而用各种优化方法求解。然而用于描述位姿尤其是姿态的变量是旋转矩阵、四元数这种光滑 流形上的变量,流形是高维空间中的低维嵌入(这句话隐式的描述了变量的约束,例如球面是三位空间的二维嵌入,只要经纬度就可以描述球面上的任意一点),不满足向量空间的定义,且这种结构上的加法是没有定义的,比如球面上任意2个点若执行“相加”操作则结果就不在球面上了,会跑到集合外面。以旋转矩阵为例,两个旋转矩阵相加,结果不再是旋转矩。要对两个旋转矩阵进行复合操作,唯一的方法是矩阵乘法,旋转矩阵之间通过唯一的复合操作构成了一个群,这就是所谓的特殊正交群 ,旋转矩阵的9个元素不是自由的,它们必须满足的约束。这一切使得常规的需要计算一阶或高阶导数的优化方法无法展开;即使直接计算解析解,也会存在相应的困难。

群:一个集合,定义了单位元,以及一种复合操作,集合中的两个元素在经过复合操作之后仍然在集合中,并且每个元素都可以找到逆元,使得两个元素的复合为单位元。

但是,和李群这种光滑的流形一同出现的,满足一些规则,拥有一些特定运算的数学结构还包括李代数。李代数是光滑流形上变量的切空间,可以看作某点的切平面,此且平面与同维的向量空间同构。李群和李代数之间刚好有指数映射可以完成两个集合的元素的转换,任意一个李代数都可以映射到李群上。这样,我们可以把两个群元素的复合操作变为一个群元素和一个李代数转化为群(通过指数映射)之后的复合,从而定义所谓的广义加法 。这样,我们就定义微小增量,从而可以将狭义的导数推广到李群上,通过广义加法定义导数,从而计算优化时需要计算的雅可比矩阵等,从而得到迭代更新的增量,并将其应用到李群上完成状态更新。

光滑表示这个群在几何结构上是平滑的,不存在凸起,因此处处都存在切平面,是“广义”可微分的。

同时,我们在进行估计状态的时候常常认为状态存在一定的不确定性,最常见的做法是认为我们获得的观测数据是真值和一个加性噪声的结合,比如高斯噪声。但恰如刚刚所说,旋转矩阵的集合这个特殊的群不存在加法,因此没有办法在这个李群上通过加性噪声来表示不确定性。还好李代数和线性空间同构,一切在向量空间中能够使用的我们熟知的那些技巧,都可以拿来主义。我们定义一个李群上的状态真值和噪声的复合: 。在李群状态上使用扩展卡尔曼滤波称为误差状态卡尔曼滤波ESKF(error state KF),或者叫MKF(manifold KF)。

为了进一步理解李群和李代数的物理意义及其应用,推荐你阅读这篇文章:A micro Lie theory for state estimation in robotics (arxiv.org)。文中给出了关于李群和李代数非常直观的可视化,同时介绍了李群和李代数的一些重要运算和变换。当然,请不要害怕数学,虽然作为一名工程师(也许只是现在)你只需要将其当作工具使用,但我还是建议你了解背后的原理,这可以帮你更好地实践。

5.7.4. 常用传感器配置及处理方法

简单介绍不同传感器配置下的经典前后端算法,以及它们的优缺点。

5.7.4.1. 单目SLAM

这是最廉价的SLAM方案,仅需要一颗摄像头就能实现。前端的tracking部分为了找到两张图片的关联,一般通过特征点法或光流法追踪相邻图片的对应点,然后根据点对构成的约束,执行2D-2D的姿态估计。这些姿态估计方法统称为对极几何,包括:

  • 单应矩阵homography估计:当点都落在同一个平面时使用此法。
  • 本质矩阵essential和基本矩阵fundamental估计:大部分2d-2d的情况使用。因为点落在一个平面上时约束会退化,所有才使用单应矩阵。这两者的区别在于本质矩阵估计需要知道相机内参而基本矩阵估计不需要相机内参。

不同的对极几何问题使用的约束不同。相关内容网络上有大量资源这里不再赘述,推荐阅读vslam14讲。

而直接法则不考虑像素关联,直接构建一个优化问题,目标是求解一个位姿使得上一帧的patch通过位姿变换投影到当前帧时的光度误差最小。为了使得patch之间的区分度更好,一般使用特征点检测的方法初始化patch的中心位置。

由于单目相机缺失了尺度这一信息(没有办法获取绝对尺度,实际上就是因为像素深度信息在成像过程中丢失),这些2d-2d方法能估计出帧间的旋转,但只能获得平移向量的方向。

在获取了两帧图像的相对位姿之后,通过对两帧图像上相同点的三角化即它们在世界坐标系下的对应点和它们自己的两两连线会构成一个三角形平面这个约束,计算出该像素在世界坐标系下的位置。同样出于上面的原因,这个位置的尺度是不定的(三角形的其中一边是通过对极几何估计出来的,这个向量只有方向没有大小)。一般情况下我们将第一帧和第二帧初始化估计得到的平移向量设置为单位长度,后续的尺度均以此为基准。

在有了之前三角化的3d地图点(假设尺度已经固定),后续的点若和地图点匹配,便可以执行2d-3d估计,也就是常说的PnP算法。PnP的求解也可以看作优化问题,优化目标是找到使已知3d点在成像平面上的2d投影点的重投影误差最小的相机坐标系和世界坐标系的位姿变换。通用的方法是先找到数个匹配,计算得到解析解,然后再通过非线性优化的方法进一步减小投影误差。

为了提高位姿和建图精度,单目SLAM会在后端保存最新几帧或关键帧的窗口,并持续追踪窗口内帧的特征点,当多帧图像中出现相同的追踪点时便可以构建多帧的优化问题:确定每一帧的位姿以及它们看到的相同(世界坐标系下的)点的位置,以最小化该点在所有图像上的重投影误差。MSCKF是经典的多帧窗口滤波方法之一,它只维护多帧位姿而不会将特征点位置加入状态当中,仅当特征点跟丢时才对其进行三角化,然后再用三角化之后的特征点计算重投影误差,对多帧位姿进行优化。

若没有执行多帧窗口优化,由于相机两帧之间移动距离小、分辨率不足、特征点位置不准确等原因,直接三角化得到的地图点误差可能较大,还有一类用于解决该问题的方法被称作深度滤波器:将地图点的深度信息建模为带有不确定度的概率分布,每当有新的帧观测到该点,就再进行三角化,并更新该概率分布。最常用的假设是深度服从高斯分布,用类似卡尔曼滤波的方法更新地图点的深度。

这里的深度一般指该地图点在 自身(3d)和第一帧观测到该像素点的图像上的位置(2d)的连线上与第一帧相机光心的距离。地图点与第一帧图像观测自身的2d点的连线(会穿过光心),地图点再之后的观测中不断被更新,但时钟在刚刚提到的那条线上。线上和光心的距离就是我们所谓要估计的深度,因为2d像素已经可以确定深度,唯一不确定的就是在这个像素射线上的深度,所以叫深度滤波器。

对于回环检测,一般使用特征点视觉词袋,每隔几帧取一张图片提取词袋向量以及该帧位姿并保存,同时检索当前词袋向量和已经保存的向量是否有相似程度较大的。若超过设定阈值则认为触发回环,会构建上一次到达此地的位姿和当前里程计位姿之间的约束,优化所有关键帧的位姿使得整体轨迹误差减小。当然也可以使用learning-based的方法,end2end或目标检测都是可用的。

当特征点法遇到纹理缺失的场景会遇到无法检出的问题,导致帧间位姿无法估计。光流法和直接法受光照和相机参数影响较大。同时,由于对极几何的约束条件,单目SLAM无法应对帧间发生纯旋转的情况,这会导致约束退化。

最经典的单目SLAM算法是PTAM,这是最早提出将前端跟踪和建图过程并行化的算法框架,后续的工作大部分继承了这种范式。Mono-SLAM是最早使用EKF的filter-based vslam,解决了一些早期对地图点进行滤波带来的维数灾难问题。ORB-SLAM是使用ORB角点作为帧间跟踪手段的单目slam方法,使用ORB生成的rBRIEF描述子构建视觉词袋进行回环检测,后续还推出了第二个和第三个版本,还兼容了其他传感器配置。

5.7.4.2. 双目SLAM

双目相机的配置也非常亲民,和单目相比需要的额外步骤是两个相机的外参标定。根据前面介绍过的视差法,只要得到两个相机同一时刻采集的图像上在世界坐标系下的相同点,通过三角测量便可以求解得到该点的深度。视差法实际上和单目相机的三角化非常类似,不同的是,双目相机中两个相机的外参是已知的,平移向量的大小和尺度都已经确定,从而可以获取绝对的深度。

因此,在双目slam的前端tracking中一般使用PnP算法求取当前帧和上一帧的位姿变换,而为当前帧生成新的地图点时,使用两个相机的视差计算,为了找到对应关系,则是在两个相机拍摄的图像上执行特征点法或光流法。后端的滑窗优化或使用关键帧构建多帧观测约束的方法单目类似,也是求解最小化重投影误差的优化问题。

如果在双目拍摄的两张图片上执行直接法,还可以构建半稠密的地图。由于已知两个相机的外参,通常只需要一次观测就可以得到比较好的深度估计:在左相机的一条极线于右相机成像平面的投影上寻找对应点。极线搜索通常也可以建模为光度误差最小的优化问题。若要进行稠密的建图,则需要额外的硬件支持如GPU,极线搜索可以高度并行化。

ToF和结构光深度相机的深度测算原理和双目类似(一个发射端一个接收端,两者的外参已知),可以归入此类。

双目相机能够观测绝对尺度且不用担心纯旋转导致的退化(通常的PnP不要求这种约束),但是同样受光照和纹理的影响较大。

经典的双目SLAM算法包括orb-slam2/3,vins-fusion和open-vins(这两套框架支持双目相机,在不搭载IMU的时候也可以运作)。

5.7.4.3. 2D激光雷达

2d激光雷达也叫单线雷达,在仓储AGV、餐饮服务机器人以及扫地机器人上的使用尤其广泛,大家在购物平台上常见的所谓智能slam自动驾驶小车,其上一般就搭载这种激光雷达。通常的结构为一对激光发射接收头,和一个承载该传感器的带编码器的电机。工作时,电机以恒定速度旋转,激光发射器则在固定的频率下发射探测光束,由接收头记录装置在特定角度下的回波从而计算距离。

和图像数据不同,雷达的数据较为稀疏,而且返回的数据只有反射点距离激光雷达的位置信息。2d雷达得到的数据可以视为在一个平面上,所以图像处理的方法常常可以派上用场,将场景中有障碍物(激光反射点位置)的栅格/像素设为1,而在激光途径点的像素均为0,我们就得到了一张二值图像。

2D雷达的前端匹配使用得最多的方法是迭代最近点匹配iterative closest point(ICP),不像vslam一样通过光流或特征点匹配去寻找correspondence,ICP直接指定当前雷达扫描帧中和用于配准的地图点最近的点作为匹配,并以点对的距离范数作为误差,通过svd获得解析解或使用非线性优化求解位姿。这实际上是一种期望最大化算法。

除了point2point的匹配,还有point2plane的匹配,即认为当前扫描点和之前地图点构成的平面应该充分接近,因此其优化目标是最小化当前扫描中的所有点到地图点形成的和扫描点最近的平面上的距离。除了点对匹配和平面特征之外还有线特征,以及你可以定义各式各样的pattern,只要你能够用合理的方式去提取地图点云的局部特征,然后在current scan中将点对应到这些特征上,计算误差即可。

既然特征可以是各种各样的,为什么不能选择最好的特征去描述点云信息?所以我们可以使用机器学习的方法而不是手工设计特征,这就导出了所谓的正态分布变换(Normal Distribution Transformation,NDT)。我们可以根据已有的点把点云建模为概率分布,即把地图点视作点云分布上的采样,通过最大似然的方法拟合一个点云分布(先验),然后利用current scan作为观测,得到位姿的最大后验估计。这样,不需要手工设计特征,我们就把已有的地图转化成拥有不同方差和均值的正态分布集合(这里的μ和Σ就是我们的“特征”)。

这和高斯混合模型(GMM)非常相似!只不过SLAM中为了减小计算量,我们会把空间分割为很多体素,然后在体素内对点计算均值和方差,得到每个体素内的高斯分布。

对比vslam,Lidar的前端基本很少出现寻找特征匹配的步骤(当然也有使用点云特征的方法,比如LOAM、LeGO-LOAM等),更多的是直接最小化point2plane残差。因此,多帧优化对Lidar来说也较难展开,因为我们无法像camera一样找到多帧约束关系(雷达很难在几帧扫描中都获得一个相同点回波)。不过激光雷达本就可以获得绝对的距离信息,因此雷达里程计的漂移相对视觉里程计较小,一般只进行回环检测和优化即可。

上面提到,ICP的变体都需要计算current scan点到配准目标平面或线等特征的距离。为了得到这些平面和线特征(而且是离当前点最近的),我们需要在配准目标点云中寻找距离当前点最近的一些点,然后将他们拟合成平面或直线。如何从庞大的候选点中找到最近的用于拟合的点?这就是经典的k-nearest问题。这其实和特征点匹配非常相似,只不过特征点匹配中对比的是相似的特征描述子(特征向量),这里要匹配的是点的xyz坐标(也可以看作某种特征)。所以,我们额外维护一个支持快速查找临近点的数据结构,包括kd-tree以及其改进ikd-tree、hash-point、iVox和Dynamic Octree等。

为了减小匹配计算量同时方便回环检测,Lidar slam常常会把地图点分割成局部地图保存。即把连续几帧scan的结果存在一张子地图上,而不是把所有的点都配准到全局地图中。当机器人移动的距离超过阈值,就创建一个新的局部地图,并复制之前地图中的几个关键帧的scan为后续配准使用,配准的结果输出的是current scan到局部地图的位姿(显然每个局部地图都保存了自己到全局坐标系的位姿)。这样,前端在进行ICP的时候就不需要考虑所有的点云,只是在当前的局部地图上进行。创建新的局部地图时,之前的局部地图会被保留下来用于回环检测等。而用于k临近查找的数据结构则始终只需要维护一个局部的地图(更远的点用不到了)。

2d雷达的回环最简单的方法是匹配测试,即尝试将current scan和历史上保存的局部地图或关键帧进行ICP匹配,若成功且残差小则认为匹配成功,触发回环优化。为了提高匹配的成功率的准确度,常常使用corase2fine的金字塔匹配方法,先尝试分辨率较低的地图上匹配,若匹配成功则结果作为下一级分辨率更高尺度更小的地图ICP的初值,直至匹配到最高分辨率的地图(或关键帧)。

如果使用了点云特征的方法,也可以像vslam一样使用词袋模型来进行回环检测。由于点云的特征不如图像丰富,为了获取更好的scan描述,一般直接对关键帧或局部地图的所有点云直接提取全局描述子。

得益于2d雷达能够为图像所表示,我们通常使用占栅格地图来表达2d雷达的建图结果。占栅格地图地图不仅方便图像处理算法的应用,还有一个优点是可以直接用于导航:地图已经指示了可通行区域和障碍物。和vslam一样,为了提高精度,我们常常会构建概率栅格地图,地图上的每个像素并非0/1的二值描述,而是一个0-1的像素点,表示该格点存在障碍物的概率。每当有扫描点落在格点内(说明有反射)就增加被占有的概率,反之则减小。使用ICP的方法一般会使用占栅格地图。

地图的描述还可以使用增量式NDT,由于高斯分布优良的性质,其共轭分布仍然是高斯分布,那么在current scan在配准之后,就可以作为新的观测融合到之前的概率分布中,得到后验概率分布。显然,使用NDT匹配,对应的增量式NDT地图是不二之选。

需要注意,不论是3d还是2d雷达,只要不是纯固态的单次扫描方案,激光雷达都会出现运动畸变,因为不同角度的激光不是同一时刻发射的(可以类比卷帘快门)。因此一般我们会根据上一帧得到的机体速度、角速度对采样点进行运动补偿以减小畸变。如果有IMU可用,则使用与每个点采样时间最近的imu数据进行补偿以尽可能减小畸变。

5.7.4.4. 3D激光雷达

3D激光雷达一般分为固态雷达、混合式雷达(半固态?)和机械式雷达。固态雷达和相机有些类似,只不过每个“像素”都具备主动发射的能力;混合式雷达则是对机械式雷达进行了一些改进,尽可能减小机械磨损和震动给测量带来的影响,比如不让发射接收模块旋转而是在其上增加一个可以旋转的棱镜,livox系列的雷达大多使用这种结构;机械式雷达也是最经典的设计,3d的机械式雷达式对2D雷达进行了额外的自由度设计,在2d雷达的yaw轴基础上增加了pitch使得发射器和接收器可以俯仰运动。也有在yaw电机上于pitch的不同角度安装多对传感器实现立体扫描的方案。主要讲解数据关联方法和优化方法,以及和2D LIDAR的异同。

3D lidar的前端配准方法和2D类似,经典的pts2pts和pts2plane,以及NDT都可以使用。由于增加了额外的维度,地图描述一般使用体素(Voxel,其实就是很多小立方体,就是2维栅格增加了额外的z轴)。占栅格地图则较少使用,一方面是因为3D lidar的应用场景一般为大范围高精度定位和自动驾驶,机器人运动场景复杂且空间跨度大,若要增加网格精细度,则存储3维栅格地图的开销很大,而空间中的大部分区域又都是空的(没有点云),存在较大的浪费,这不允许我们使用细粒度的网格。

这也是kd-tree描述地图的优势之一。包括前面介绍2d lidar slam时提到的kdtree的各种改进(ikdtree、iVox、Octree),都是在保证较小空间开销的同时支持k临近查找以方便特征计算。

由于3D点云的数量远超2D点云,因此其回环检测一般不会采用2D的匹配测试。常见的做法是为关键帧或局部地图提取全局描述子,比如经典的Scan-Context,以点云的重心为重心,计算了360度范围内不同距离的点数量构成直方图(我们之前已经介绍过图像的特征描述子,这里只是换到极坐标下面再来一次!)。当然也有结合局部点云特征和全局特征的方法,以及deep learning的深度点云特征提取方法,这里不再赘述。

5.7.4.5. IMU和轮式里程计
  • IMU

我们已经在硬件部分介绍了IMU。单独使用IMU进行惯性导航的误差很大,因为加速度计和陀螺仪都有不可避免的零飘和噪声,但是我们可以使用IMU给出系统更精确的运动学方程(注意是kinectics而不是dynamics,slam处理的时候默认不知道系统的动力学,一般先验给定系统作匀速旋转+平动,不过也有一些工作如足式或轮式机器人将SLAM纳入控制系统中的观测器框架,把动力学模型加入系统先验以得到更好的估计效果),再和其他传感器结合以达到更好的效果。通过IMU数据的航位推算就可以获得机器人在短时间内的位姿变化,如果将此状态作为相机对极几何估计或雷达ICP的初值,通常可以获得更好的效果。

由于IMU数据相对camera和lidar来说频率非常高,我们不希望每次imu采样数据到来都进行一次位姿计算,而是考虑两个其他传感器的帧之间所有的imu数据——将他们累积在一起一次性处理位姿变换,这就是预积分。很显然,预积分量只和机器人在这段时间内的运动有关,而和除了零飘的估计值之外的始末状态无关。

使用预积分的另一个好处是,我们可以采用更高阶的运动模型(一般的模型都是匀速或匀加速):可以用多项式拟合一段时间内的加速度数据以获得真实运动的高阶近似(相比简单的零阶保持其和一阶梯形积分)。

  • 轮式里程计

即通过轮式机器人安装在驱动轮上的编码器估计机器人的瞬时速度,尤其是移动机器人的电机上一般都装有光电编码器、码盘式编码器或霍尔磁编码器(汽车上估计当前速度的里程计也有)。此瞬时速度可以作为一个额外的量测值加入系统中,作为图优化中和速度状态相连的一个变量边以及滤波框架中的一个量测方程;也可以和IMU一样作用,对机器人的短时间运动进行航位推算,获得位置和yaw的变化量。

  • 其他相对式测量

只要能够得到机器人当前运动学状态量相关的传感器,都有相同的作用。

5.7.4.6. GNSS及其他绝对式测量

GNSS即Global navigation satellite System,包括耳熟能详的GPS、北斗,还有比较不知名的格洛纳斯和伽利略。他们的定位原理都是基于TOF的距离测量:三颗卫星同时和GNSS收发机通信(三圆交点之多2个,test排除一个天上的点剩下的就是你的位置)。当然,也可以加入更多测量以提高精度。如果机器人搭载单个收发机,则可以实现经纬度和高度的测量,若增加额外一个收发机,则构成了所谓的RTK,确定空间中任意2个点的位置便可以确定除绕此2点连线的轴的旋转之外的其他自由度(如果再增加一个则可以确定机器人在空间中的位姿,只要和前2点不共线)。

移动基站定位、UWB、声纳等其他绝对式测量用途和GNSS类似。

camera或lidar前端可以给出一个帧间位姿的估计值,而GNSS也可以给出一个带噪的位姿,因此,GNSS的测量可以作为图优化中和位姿状态连接的边或滤波框架中的一个位姿量测方程。

由于这种绝对式测量给出的量测值不会累积(不像imu和里程计得到的误差会累积,这里没有积分或累加的过程,即使有零飘也是真值附近抖动),它们就成了绝佳的状态量测手段,可以很好地帮助矫正其他传感器的零飘并修正累积误差。

5.7.5. 多传感器融合

单一传感器的缺点在上面都介绍了,camera直接法面对白墙或光照剧烈变化的场景会失效,特征点法在缺少纹理和细节的场景会失效,lidar在长走廊、隧道等存在对称性的场景会发生退化,imu的累积误差大,GNSS的数据存在较大幅抖动且室内无法使用,磁力计受周围磁场影响大,轮速计在打滑时不可靠......

所以我们博采众长,组合不同的传感器搭建异构感知系统是最佳的选择。出于成本和可用性考虑,最常见的组合包括VI、LI、LVI,它们在大部分场景都能够工作。而完整的感知系统甚至会包含每一种传感器,比如自动驾驶汽车上一般就有前后感知雷达+BEV全景相机+毫米波雷达+超声波雷达+IMU+GNSS-RTK。

VIO系统使用imu的预积分观测作为两个图像帧间的位姿先验,其余部分和vslam类似。若要进行图优化则不同时刻的位姿间增加了预积分约束(两个位姿作差的结果和预积分结果应该尽可能接近)。

LIO系统同样将预积分作为先验,同时还会利用imu的高频数据对扫描点进行运动畸变补偿。

在以上两个系统中,若V或L遇到退化场景,IMU在短时间内的积分结果可以保证里程计不发散。

LVI系统的组合则较为丰富,当相机遇到白墙或光照变化剧烈的场景时,雷达仍然可以工作;雷达遇到走廊或平地时,相机仍然可以工作,即使是极限情况也能靠imu顶上一阵子。IMU还是作为其他传感器帧间的位姿先验,但如何融合lio与vio的输出则是一个重要议题。是独立运行两个里程计然后使用滤波的方法将它们的结果融合得到最终的输出,还是以某种方式将相机和雷达的观测联系/合并起来,共同维护一个里程计和地图?前者的方案包括LVI-SAM,LVI-fusion,R2LIVE的框架和前者类似,而R3LIVE、Fast-LVIO则采用了后者的形式。

关于多传感器融合SLAM,可以关注笔者的知乎专栏:LVIO SLAM - 知乎 (zhihu.com)

5.7.6. 滤波和优化\&传感器耦合程度

之后更新,将会提供统一的视角。

5.7.7. 可能的进展

笔者认为的重要研究方向和未来,以及近来较为热门且有可能突破的方向。当然不如综述文章来得细致周到,感兴趣可以自行搜索相关内容。

  • 高精度SLAM

精度体现在两个方面:轨迹精度和始末端漂移。对于一些行业级应用如勘探测绘、高精地图、精密仪器制造等,对于里程计的精度有更高要求。可能的方法包括考虑机器人本身的动力学模型以获得额外观测、获得传感器更精确的量测模型以减小系统误差、对状态误差进行更精确的建模使其尽可能与实际情况一致而不是使用简单的分布假设。

  • 语义SLAM

通过目标检测、点云分割的方式识别出场景中的语义对象如车、人等物体并用结构化数据或特征向量描述,而不是使用点云或线面特征的方式保存地图。这对动态SLAM以及大范围SLAM来说有重要意义。

  • 稠密地图\&Nerf-SLAM

激光雷达生成地图的稠密程度取决于线数,视觉地图的直接法或深度相机也只能做到半稠密,双目相机计算稠密地图的开销则非常大。一种可能的方式是结合多种传感器进行融合,另一种方式则是图形处理和3维重建非常火爆的神经辐射场Nerf。与SLAM对地图和位姿的状态估计建模方式不同,Nerf直接学习位姿+图像到3d点的颜色和透明度的映射(给神经网络找到一个优化目标,然后拟合!)。

这就意味着使用训练好的Nerf模型进行推理时,只需要输入图像数据和位姿数据,就可以输出图像上点对应的深度,从而完成建图。反之,给定一些点的深度和图像,可以直接获得位姿数据。

  • 动态场景和life-long SLAM

通过点云分割/目标检测的方式筛除场景中的动态部分,或者将动态物体的运动学也进行建模,使用动态物体反而提升slam的性能,和语义slam的研究有很大的重合。

life-long SLAM则要求算法可以适应变化的地图,如场景中的物体和形状出现的增减。



5.8.路径规划和轨迹生成与跟踪

当我们使用SfM或SLAM技术构建了地图并且定位自身在地图中的位置,下一步就是实现地图上可通行区域间两点的路径规划。首先我们需要将先前创建的地图转化为可导航的地图,最经典的导航地图就是栅格地图。如果使用2d slam则可以直接将点云地图处理后转化为占栅格地图,3d slam生成的地图通常将z方向的点进行压缩:对于人造场景首先确定地面,然后将z过大的点(比如高于机器人1.5倍)和低于地面的点删除,最后将所有点投影到地面。这时候,就可以使用图像处理的方法对2d地图进行处理,进一步划分出可通行区域和障碍物区域。

对于地图中出现的动态障碍,一般通过点云识别(直接进行点云分割)或结合相机(联合标定后将点云投影到图像平面上,识别出哪些点是属于动态物体的)的方法,把障碍动态地更新到导航地图上。动态场景下的建图和规划都是各自领域内的重难点。

一张2D栅格地图也可以看作一个很大的邻接图,每个栅格与附近的四邻域或八邻域是直接相通的。而图算法中经典的遍历算法DFS和BFS相信大家都不陌生。DFS和BFS把每个边的权重都定义为1,若存在带权边,则使用最小路径的生成算法Dijkstra。

以上算法都有收敛和最优的保证,但存在求解速度较慢时间复杂度较高的问题。一些改进的方法则考虑给予额外的“指引”,比如应用最多的A*算法会增加先验方向上的探索权重,还有大量启发式的探索规则。而规划领域的最新进展则是将路径规划建模为优化问题,引入各种结构的损失,然后使用通用优化求解器计算最优解。

笔者对这个领域的了解仅限于此,基于优化的planning没有涉足,欢迎补充。

得到了运行路径之后,我们就可以驱动机器人沿着这条预定路径移动,到达指定目标了。但我们发现,这时候的路径实际上是一系列的点(栅格地图不论2d还是3d都是离散化的)。当然,我们可以使用位置闭环的方式控制机器人,每次给机器人设置一个waypoint(即路径规划生成的路径点)为闭环目标,当机器人到达该点再选择下一个waypoint为目标,如此往复直到到达最终目的地。

这样做虽然简单但也存在很多缺陷,比如栅格精度低时很可能两点间的连线会碰到障碍物、机器人在到达waypoint时参考输入改变导致阶跃使得系统产生震荡、机器人的运行轨迹不平滑,速度不连续不利于控制等。我们自然而然的想到使用这些途径点拟合一条平滑的曲线。

这个曲线可以是多项式曲线,可以是弧段,可以是任意样条曲线。为了保证连续性,我们要求机器人在每两条曲线连接处(即路径点处)的速度相同或加速度相同,就可以保证机器人在曲线间切换时运动的平滑过渡。最常用的是三次多项式和五次多项式插值,并要求在路径点处两条曲线的中后者的起始与前者末端的位置的n阶导数相同(n可以自己指定),这样就可以生成一个方程组,求解之后便可以得到机器人的参考轨迹(以及参考速度、加速度等高阶量,取决于你的约束)。这个步骤称作轨迹规划,这也是机械臂控制中至关重要的环节之一。

得到了位置、速度等参考量之后,我们只需要让机器人去跟踪这些参考输入,最后一步就是轨迹跟踪。这是控制领域经典的问题,最简单的方法是使用无模型PID调参,进可对机器人的动力学进行建模使用model-based PID或线性反馈控制器;当然也可以使用LQR或MPC这些优化控制方法。

可以看到这些模块之间是层层递进的关系,而且彼此之间存在一定的重合,若你无法理解三者的关系,在网络上搜索一下它们的图示也许有帮助。

最新的一些工作也尝试设计端到端的路径规划-轨迹生成-轨迹跟踪算法,通过优化的方式直接求解出控制器的输出,一步到位。若在轨迹跟踪时使用MPC,实际上多少有一些浪费,因为MPC的问题建模就是考虑了状态和输入输出约束的,而在路径规划和轨迹生成时,先前本就已经使用了一些约束条件。



5.9.自主决策

这是机器人与环境交互最上层的算法,可以看作用于生成机器人该如何与环境互动行为的策略,通过传感器获取信息(可能还会结合历史的环境信息和之前的行为),分析之后给出机器人在当前或接下来的行动。决策的结果取决于我们的设计,一般是较为抽象的,需要由下层算法作进一步的解释。

最简单的决策方法是rule-based策略,这是专家工程的一种,需要我们手动设计机器人的行为以及触发这些行为的条件。比如,当哨兵机器人观察到相机视野中存在对方英雄机器人,就去追击它;当前哨站告破,就自动返回巡逻区等。

既然hand-craft策略不好确定,而且设计起来又很复杂,自然而然的想法是使用数据驱动的学习方法。若给定当前的环境信息需要采取的动作则可以使用监督学习,比如决策树、分类器(类别就是要采取的行动)。也可以采用聚类的方式,前提是当需要合理地为环境信息提取特征,提取的可能方法是降维,如自编码器、PCA,得到聚类中心后为每一个类别设计合适的行动。

ML的方法虽好,但也存在一定局限性,比如结构化信息较难建模、收集数据以及标定需要耗费大量人力物力。强化学习的方法对于机器人和环境的互动是一个很好的选择。给定不同行动以及行动结果(通过传感器收集)给予行为一定的奖励,从而让机器人学习到能使奖励最大化的策略。关于强化学习的入门介绍,推荐李宏毅教授2022学期的RL部分。

最后就是大一统框架下的优化方法。首先人为或通过数据驱动的方法得到一些准则,在准则的前提下(准则可以是优化的约束,也可以是代价和误差的定义方式)求解优化问题,得到特定意义下的最优策略。不过优化的方法相对RL方法较为“短视”,而且前者常需要建立模型,而RL可以是无模型的。

对于比赛中的应用,较好的方法是封装一些基础的动作集合使得机器人最终的行动的粒度合适,这个目标再交由下层的路径规划执行。如果小到每个电机的输出是多少,这已经不太能算是决策。可以类比人的决策,在你想去餐桌上拿茶杯的时候,你并不会“告诉”每一块肌肉应该在合适的时候输出特定大小的力,而是在较高层级上给运动中枢发送指令。

对于信息的采集,也需要提取出有意义的语义信息,如30°方向2m处有一台对方的步兵机器人、正后方有一台我方的英雄机器人、本机的速度是vx=2m/s,vy=0.3m/s等,而不是直接输入图像、激光雷达等的信息。

当然有所谓的end2end方法,但也许并不是正确的解决方式。在比赛中也没有这么大的数据量能够喂饱大参数的模型。

云台手目前仍然可以给哨兵机器人发送指令(虽然频率有限制),因此一般只需要简单的rule-based策略配合即可。




6.各算法具体应用

在第五部分中介绍了一些简单的OpenCV函数和他们能实现的功能,同时也大致浏览几乎所有我们可能会用到的算法的原理。根据视觉主要的开发任务,接下来将分为7个不同的兵种/方向进行介绍,每个部分其实都有交叠,但也有各自面临的难题。现在就让我们把这些算法具象化,应用到实际当中去吧!

P.S. 2021和2023的大疆青年工程师大会(bilibili搜索RoboMaster机甲大师和RoboMaster赛务君,有合集),部分队伍的赛题任务解决方案分享非常值得观看,有些学校的方法有很好的借鉴意义,参考价值很高。自行判断是否值得观看。

主要介绍的方向包括装甲板识别、能量机关识别、相对位姿解算和目标运动学建模。还有对

6.1.装甲板识别

6.1.1. 传统的灯条匹配+数字识别方法

这是以一种手工设计特征为主的专家方法。仔细观察装甲板,显然装甲板上并没有什么丰富的纹理和细节,不适合使用梯度特征检测;我们会发现其最显著的特征就是两侧发红/蓝色光的灯条和装甲板中央的数字贴纸(哨兵/前哨战/基地则是图案贴纸),表现为大块的色块,非常适合形态学操作。因此考虑检测成对的灯条+这对灯条中的数字。大体的流程如下:

预处理的过程,目的是使灯条特征凸显出来便于之后的轮廓筛选

下面分别介绍预处理的每一个步骤。

6.1.1.1. 预处理
  1. 通道分离相减

假设我们现在处在红方,则要识别蓝色装甲板。从RGB空间思考,显现蓝色的像素有什么特点?显然它的蓝色分量要比绿色分量和红色分量大。如果用拆分通道后的蓝色通道去其他颜色通道还有比较大的余量,说明该物体在图像中足够“蓝”。白色的物体三个通道分量都很大,显然相减之后会得到接近零的值。再看看场地的等效和光线环境,除了白色的炽光灯就是场地红蓝色灯条。

RoboMaster2021技术交流的场地光线条件

因此一般选择蓝色通道减去红色通道(识别红色则相反),更保险的做法同时考虑绿色通道,确保B的value是最大的,从而让提取出的灯条足够红/蓝。

经过此番操作后,会发现原图中的蓝色区域仍然有着较高的亮度,而红色和白色区域几乎已经是灰黑色的了(相减后Mat是一个通道,以灰度图的形式展现)。

一张英雄的图片,右边为蓝色通道减去红色通道后的效果
  1. 二值化并取交集

第1步中得到通道相减灰度图必然还要进行处理并得到更明显的特征。因此需要设置一个合理的阈值对通道相减后的图片进行二值化。可以根据不同的阈值得到不同效果的图片进行不断地调整最后得到一个最优的阈值。官方已经提供了装甲板发光的RGB参考值(但是请注意不同的相机和镜头拍摄得到的RGB显然不同,我们一般通过在场地矫正参数或使用设定自适应阈值,最好是实现自适应阈值,以防光线条件剧变对处理效果产生影响。我们之前在 5.5 也介绍了阈值自适应的基本方法,可以回去看看)。

由于RGB空间中每个通道既包含了亮度信息又包含了颜色信息,无法将其解耦,所以我们一般将蓝色通道减去红色通道并二值化后得到地二值图于蓝减绿并二值化后得到的二值图相与从而在一定程度上降低耦合的影响。

当然,更好的办法是将图片转换到HSV空间,从而解耦颜色和亮度,就不必进行流程图中右边分支的操作了。从处理的效果来看,HSV方法一般要优于RGB;但是从速度上来看,将RGB转换到HSV空间有额外的消耗,这取决于你使用的运算平台是否支持矩阵并行计算的加速操作(颜色空间转换要对图像中的所有像素点进行遍历)。

对第一步生成的图片进行阈值二值化得到的图片

吉林大学TARS-Go战队在他们开源的代码中提到可以将 split() 和 subtract() 合并为一个利用指针遍历整个矩阵的操作。在没有GPU或并行加速优化的情况下,他们重新编写地函数确实提升了性能。但是笔者又找来了一台配置使用ippicv加速地intel NUC,分别运行两种方法发现使用opencv封装好的函数能够利用加速库带来的优化,取得更好的性能。

细心的同学会发现,为什么灯条明明是蓝色的,但是通道相减后中间却变成黑色的了呢?(说明红蓝分量接近)这是由于拍摄这张图片使用的相机的曝光调节的过高,感光单元达到饱和(蓝色分量一下就饱和,红、绿分量虽然小,但是曝光时间长使得这两者颜色的光在采集时也逐渐达到饱和),灯条就呈现白色了(图1)。所以在使用传统方法的时候,我们都会降低相机的曝光,一般拍摄得到的图像如图2。同时,降低曝光还能减少杂光的影响,进一步减少预处理的时间和复杂度。

p1:放大图像查看细节,发现灯条确实呈现白色

p2:降低相机曝光后拍摄得到的图像,装甲板灯条显现纯粹的红色,杂光也大大降低了 我们在下面的步骤中就以这张图片为例简化过程分析,同时也不失一般性

当然,你也可以思考一下,如果我就要使用较高的曝光来拍摄(希望中间的数字特征更为明显),有没有什么方法能够提取装甲板的特征呢?

提示1:灯条外面有一圈明显的背景光晕,包围了灯条,形成了一个闭合的轮廓

提示2:颜色光晕包围高亮度灰度轮廓

提示3:为高亮的长条形区域内部统计颜色直方图

  1. 形态学操作

在5.1部分我们已经稍微介绍了一些形态学操作。最常用的是dilate()和erode(),此外还有open()、close()、morphgrade()、tophat()、blackhat()等。后面五个操作都是膨胀和腐蚀的组合操作。希望详细学习这些函数请参考《学习OpenCV3》的形态学操作部分。我们最常用的是dilate(),erode(),open(),close()这四个能够很方便地修改二值图轮廓的操作。

在第1步和第2步的处理结束后,由于拍摄角度、杂光干扰地原因有些灯条可能会发生“断裂”或变得“破碎”,我们肯定希望在查找轮廓地时候能够把完整的灯条筛选出来,因此这一步我们使用dilate来扩张轮廓的边缘使断裂处闭合。同样可以使用闭操作:先膨胀再腐蚀(为什么不会又断裂?想想形态学操作的原理)。使用闭操作也可以防止灯条在直接膨胀后的面积扩张,这可能会导致之后的解算不够准确。注意结构元素的选择,机器人上的装甲板必然是横向的,即使是上坡中的机器人其装甲板倾斜角度也不会大于30度,因此我们要着重注意灯条竖直方向可能出现的断裂(细长的特点致其更容易受光线条件等的影响),一般会选择纵向放置、椭圆形或矩形的结构元素。

膨胀前后对比,因为各种原因(不合适的阈值、突变的曝光等)断裂的灯条恢复了原状

我们也可以在这一步之前对就直接图像进行滤波以去除噪点和杂光(把之后的滤波移到这一步),这也能减小形态学操作的运算量。(同样在未进行二值化的时候就可以进行滤波平滑图像去除噪点,但是这样滤波的运算量也会增大,处处都是trade-off啊~)

不过最好的方法是不要使用形态学操作,由于其逐像素运算的特点导致其典型单线程时间开销可以达到0.6~1ms。并且其对于不同尺度目标产生的误差不同(远处的一个灯条所占像素数本来就少),这将导致后续算法中其他用于筛选灯条的参数设定有比较大的影响。

如果颜色阈值分离和相机参数(当然和相机本身的性能有关)配合得很好,在很多时候可以省去形态学操作,大大降低时间开销,毕竟形态学操作和滤波一样是逐像素的算子。

​

  1. 与灰度(亮度)二值化图取交集

前面讲到RGB空间没有解耦颜色和亮度信息,因此我们在这一步筛选掉那些虽然呈现蓝色但是总体亮度低的物体(如场地等效、反光等)。得到最后图像后,我们再根据情况使用滤波器,得到最终的预处理图片。

预处理得到的最终结果

​ 如果你在前面的预处理中已经利用了亮度信息,有可能不需要执行这一步。


6.1.1.2. 寻找轮廓并筛选拟合出的对象
  1. 轮廓

利用OpenCV封装好的函数 findContour() 来寻找二值图中的封闭轮廓,注意若使用 findContour() 的重载函数即加入hierarchy参数,则可以将子轮廓以树的结构存储在一个vector当中(这也是较高曝光时灯条在预处理时变成黑色的这种情况的处理方法,寻找有一个子轮廓的轮廓即为灯条)。若预处理能得到较干净的二值图,可以大大降低这一步的时间。其原理大概是遍历所有像素并寻找周围8个像素是否是当前像素的同色像素,和 floodFill() 有异曲同工之处。一般选择使用CHAIN_APPROX参数来降低空间开销和之后拟合对象的时间开销(轮廓中的直线会直接使用两个端点存储而不是保存直线上的所有点),并且使用这个参数后,我们还可以对轮廓进行筛选,显然长条形接近梯形/矩形的灯条在使用直线近似时不需要很多点就能够表示,若轮廓点数超过10(此参数需要不断调试),那么几乎可以断定这不是我们需要寻找的灯条。

通过筛选点数可以去除其他光斑和发光物体的干扰

  1. 拟合对象与配对筛选

虽然得到了描述轮廓的一组像素点,但还是将它们转化成更容易操作的对象方便一些。OpenCV提供了三个函数用于拟合轮廓点:minAreaRect(),fitEcllicpse(),minEnclosingCircle()。显然灯条是长条形的,我们一般使用前两个函数。以 minAreaRect() 为例,将上一步留下的所有轮廓全部用最小面积拟合并存入一个数组。

此处需要注意OpenCV中RotatedRect的角度系统,在5.1我们已经提过了这一点。

拟合出旋转矩形对象后,通过两个嵌套的循环将旋转矩形两两匹配,利用一连串地ifelse筛选长宽比、面积比、高度差、中心距离、倾斜角度等信息,或是给这些信息赋予不同的权重计算得分最后设置分数阈值(这种方法大大提高了筛选的连续性,降低了筛选函数的非线性特性,会比硬性的截断有更好的效果,但是有额外的开销,trade-off again!)。和前一步一样,如果轮廓筛选的干净,匹配的时间会成阶乘级别地降低!

特别注意,筛选的时候应该尽量选择尺度不变的特征,这样对于不同距离灯条的筛选才能尽量减少出现漏判和误判的情况,这点在 5.5 已经介绍过了。

到这里,我们已经筛选出了可能属于同一块装甲板的、两两配对好的矩形。然而,还是有很大的可能会出现这两个灯条其实并不属于同一块装甲板:

从侧面或斜45~135的角度观察机器人的时候,我们可以看到3-4个的的灯条,但实际上视野范围内只有一个装甲板

所以,我们还要利用其他信息来把“假”装甲板过滤掉。

这一步其实就是在给灯条实体构建特征描述子,我们也可以将灯条的角度、长宽比、像素大小等信息抽象为一个特征向量或某种结构化的信息,然后利用kd-tree或无监督聚类对灯条进行分类,以方便查找和匹配,最大程度减小遍历开销。这在灯条量大(你的预处理或后续的一些算法可能要求较高的曝光或增益)时尤其有效。


6.1.1.3. 数字识别

在模板匹配、svm、神经网络方法出现之前就已经有队伍想到直接利用最简单的信息来筛选这些假装甲板。例如深圳大学开源的传统方案是根据ROI内灯条的大小和角度特点来筛选真正的装甲板:

思路如图,截取自深圳大学的自瞄算法开源仓库的README

当然,装甲板的两个灯条之间也不应该有其他的灯条出现。虽然这是一种非常巧妙的方法(奥卡姆剃刀原理,能用简单的就不用复杂的),但是也只是降低了误识别的概率而无法完全避免该情况,如上坡的机器人和在平地上的机器人、相机拍摄时产生的畸变导致边缘的物体弯曲、两辆机器人在非常靠近的情况下、小陀螺运动的机器人在正好斜45度正对相机时等都有概率出现误识别。因此我们需要更好的办法提高算法的准确度。我们在前面提到,装甲板另一个非常显著的特征就是中央的数字或图标贴纸,那就从这里入手吧。

首先我们需要找到一个处理这个数字/图标的方法。随着装甲板的旋转和平移,显然装甲板大部分时候都不会正对相机,这样我们的搜索空间就会变得很大,问题的非线性程度也会上升,因此要设法让它变成正对我们的一个平面(kind of normalization!)。有没有什么方法可以把装甲板变换成一个正对我们的矩形?有!前文提到,OpenCV贴心地提供了获得投影变换矩阵的函数和通过变换矩阵进行投影变换的函数:getPerspectiveTransform()和 warpPerspective() 。前文也稍微涉及了一下仿射变换和投影变换的不同,其实我们从相机成像模型(小孔模型)也可以理解,物体都是通过投影的方式在感光单元上成像的。那么要将一个相对相机来说歪斜的装甲板转到垂直,就需要使用投影变换。

再次通过图示感受一下两者的差别吧,图中还列出了平移、旋转+平移和旋转+相似操作

两点确定一条直线,三点确定一个平面,而投影变换的维度又比同维度的线性变换要多1(垂直于那个平面)因此我们至少需要找到一个平面上的四个点和变换后的四个点组成的四组匹配点才能确定整个平面的投影变换。线性代数好的同学可以尝试从线性空间的角度理解,4个点可以连结成三条不相干的直线,3个线性无关的向量张成R3空间。当然从3D数学利用坐标矩阵变换也可以直接推导得到结果,网上的资料很多笔者这里就不列举了。随后,我们在得到两两匹配的灯条后,分别选取左灯条的左上、左下角点,右灯条的右上、右下角点,并与一个正方形的四个角点匹配组成点对,将他们投入 getPerspectiveTransform()函数中,得到投影矩阵。随后就可以利用 warpPerspective() 来将整个装甲板投影成一个正方形了。由于拉伸或压缩,变换需要对图形进行插值或抽样,我们同样可以传递不同的参数来选择不同的方法,在此不赘述。不过我们会发现,如果以装甲板的两个灯条为矩形的两条边,是无复覆盖整个数字/图案的。因此我们通常会在灯条所在的直线上往两头各延伸2/3~1个灯条的长度,以达到覆盖装甲板的目的。

截取自吉林大学开源仓库,JLU的同学的做法是在两个方向上各延伸了2/3个灯条的长度

这里额外提一下,使用广角相机或是装甲板处于视野边缘的话灯条可能会畸变得很严重,长宽比例和面积等筛选条件会受到较大的影响。如果想要在此时也取得较好的投影效果则需要利用undistort() 函数先对ROI进行校正,再进行投影变换,否则由于畸变带来的非线性因素会让投影得到的图像变形。不过在大多数时候这个影响并不是很明显,也可以选择使用畸变系数小的镜头来从根本上解决这个问题(钞能力是最强的)。

在得到可能装甲板中间的ROI后,我们就可以开始识别上面的数字/图案了。以下介绍几种常用的方法。注意,为了降低计算复杂度,一般在得到ROI的正投影之后,都会对其进行下采样以减小计算开销。

  • 模板匹配

首先需要制作各种数字和图标(哨兵、前哨站)的图像模板。模板匹配实现原理就是将模板图片在待检测图像上滑动,检测、计算覆盖区域的每个像素的相关性,常用的方法有平方误差匹配、标准差匹配、乘法相关匹配、标准相关匹配等。这些都可以通过设置 matchTemplate() 函数的不同参数而实现。因为数字和图案在经过标准化之后,几乎和匹配的标准模板一致的特征,并不会出现新增或减少,因此模板匹配可以取得相当的效果。

因为我们直接将ROI投影成一个标准的正方形,因此只要准备大小相同的正方形模板即可,且在运行时因为ROI和模板刚好重合这样也不需要我们在窗口上滑动检测。注意小装甲板产生的ROI经过投影后应该会刚好符合模板,因其正面基本就是一个正方形;而大装甲板是长方形的,注意在制作模板的时候要对其进行resize()操作。

综上,又根据比赛规则,总共需要制作7种模板:1号英雄(需要resize),3、4、5号步兵;哨兵机器人、前哨站、基地(图案)。将得到的ROI分别使用不同的模板进行匹配,得分最高的即为对应的兵种。若所有模板的匹配得分都没有超过设置的阈值则说明此装甲板是一个假装甲板(我一眼就看出你不是人!)。根据匹配得到的结果,操作手还能通过按键来选择打击哪一种机器人。

模板匹配虽然使用起来最简单,不过也正是由于其简单的机制导致它的精度和准确度不是那么的高,而且数字和图案都是只有黑白两种颜色,在同一个区域往往也有很多重叠,导致不同模板之间的差异不够大,若受到干扰可能难以区分。

实际运行时变换后的数字和标准模板的对比,可以看到由于投影变换的误差导致数字并没有”正立“
  • 手动特征分类

这是一个非常朴素的想法,我们将可能的灯条匹配对进行投影变换后,在得到的ROI上划分一组合适大小的网格,然后分别对每个网格内是否有像素块,如数字1在中间的这一排竖直方向上的网格都有像素块。这样我们就能针对不同的数字和图案得到不同的匹配标准,在检测时分别验证这些标准,从而得到分类。要是不符合任何一个分类,则认为是一个伪装甲板。

利用handcraft特征对数字和图案进行筛选,绿色网格为一种合适的划分策略

这种方法实际上是一种专家规则的分类器,和我们在筛选灯条时的筛选条件非常相似,也可以看作一种低配的、非线性的SVM。其分类准确率和划分的网格密度成正相关,速度则反之。不过这种情况比较容易受到干扰,而且正确率会受到灯条角点定位精度的影响(角点抖动可能会导致投影变换不准确,无法得到正立、居中的数字/图案)因此建议不仅要在数字有像素块的地方进行检测,还要在周围的空白处进行判断并给予像素块的有无以一定的宽容度(如果全部是有/无,则可以使用二值编码,计算模板和所得特征的汉明距离),这是因为背景类(非装甲板)的像素块往往是随机分布的,有数字的情况仅仅是庞大的搜索空间内非常非常小的一个子集。

  • SVM(support vector machine)

支持向量机在线性可分问题上表现的非常好并且对于线性不可分的问题也可以通过核方法来实现向高维空间的非线性变换,从而让问题变为线性可分。网络上关于svm的原理和教程也非常丰富故不在此赘述,高等数学的多元微积分部分也有对拉格朗日乘子法的介绍,因此推导起来也并不算太复杂(对偶方法的掌握非必须)。

对于有多个分类的情况,可以选择训练多个简单SVM或一个多分类SVM,关于这两种方法的性能,笔者暂时没有进行测试,如果有同学进行了性能和精度的比较,可以联系我补充这一部分哦。不过从直观角度理解,多个简单的SVM训练起来肯定比一个多分类SVM简单,不论是从原理还是实践角度。SVM要求输入一个特征向量,但是我们得到的ROI是一个nxn二维数组,因此要做的第一件事就是将此ROI Mat 通过 resize() 变成一个一维的1xn^2向量,随后投入SVM进行预测。

OpenCV中提供了封装好的二分类和多分类SVM,其训练和预测可以参照此教程:在OpenCV中使用SVM。上交在19年的视觉开源代码中就使用了SVM进行数字标签分类,可以戳他们的开源仓库,吉林大学2020年则是使用了一个多分类SVM,并且提供了他们训练好的支持向量参数。

采用了核方法的多分类SVM,生成的分类超平面可视化
  • 使用CNN的数字分类

图像分类当然少不了CNN,CNN的开山之作LeNet-5就是通过一个拥有四层隐藏层和两个全连接层的数字识别网络。根据5.2中介绍的CNN基本知识我们可以得到如下的流程:首先制作数字和图标的数据集,给他们打上不同的标签,当然数量越多越好。可以直接使用传统算法对ROI进行裁切,只需要保证视野内只有一块装甲板就可以批量制作数据集了。

由于我们在前处理中将图像转为正视图,所以不需要采集那些倾斜、歪曲的装甲板作为训练数据,只需要使用正视图(当然你也可以尝试增加非正视的装甲板图片作为训练数据使用,这样可以省下前面的投影变换操作;或是使用数据增强,但这未必能提高效果,因为在这个问题里搜索空间其实并不大)。由于搜索空间和类的数目并不多,预计每个分类100张图片就能得到较好的结果了。同时我们还可以加入负样本或背景作为一个分类以提高分类精度。原理已经在前文介绍不在再赘述。

对于如此一个小规模的网络直接使用cpu推理即可,消耗的时间可以基本也是和前面的处理处于相同或更小数量级。opencv提供的dnn模块可以用于这部分的训练,文档的tutorial部分提供了相关教程(你甚至可以不用torch和tensorflow等框架,直接使用c++编写一个网络!这还能增加你对神经网络的理解和造轮子的能力)。

经典的分类网络LeNet5

经过预处理、轮廓寻找与筛选和数字识别,就能够得到最终的真·装甲板了!那么应该如何利用这个信息来控制云台转向装甲板呢?如果现在就想知道,请直接跳转到 6.3 看看吧。


6.1.2.基于CNN的目标检测方法

6.1.1 中介绍了最后数字识别的CNN方法,更进一步,我们可以将整个pipline都转移到神经网络上来。需要的前置知识已经在 5.2 比较详细地展开了,因此我们就从实践的角度以装甲板识别为例讲述大概的流程。

6.1.2.1. 网络选择

首先需要根据运算平台算力、精度与速度的平衡选择一个检测网络。关于计算平台的选型我们已经在 4.2 中介绍。因此,你可以根据前文的信息针对网络的参数量、这些网络的benchmark、你的运算平台的cpu算力和gpu算力来筛选你想要使用的网络。显然我们可用的运算平台是无法同时满足大分辨率输入和速度的要求的,当下的最优选择大概是416x416(追求速度)、512x512(权衡)或640x640(精度)。320x320的模型速度虽然快,但是对远处的小目标效果明显下降(除非使用变焦相机或者双相机的策略),mAP也要远低于前述的三个模型。相机的分辨率一般和网络的输入不同,因此在送入之前还要进行resize()下采样。

Model size mAPval 0.5:0.95 Params (M) FLOPs (G) weights
YOLOX-Nano 416 25.8 0.91 1.08 github
YOLOX-Tiny 416 32.8 5.06 6.45 github
Model size mAPval 0.5:0.95 mAPtest 0.5:0.95 Speed V100 (ms) Params (M) FLOPs (G) weights
YOLOX-s 640 40.5 40.5 9.8 9.0 26.8 github
YOLOX-m 640 46.9 47.2 12.3 25.3 73.8 github
YOLOX-l 640 49.7 50.1 14.5 54.2 155.6 github
YOLOX-x 640 51.1 51.5 17.3 99.1 281.9 github
YOLOX-Darknet-53 640 47.7 48.0 11.1 63.7 185.3 github

MegaVision旷视开源-YOLOX的benchmarks

YOLOX是截至本文完成前Objection Detection领域anchor-free+one-stage算法的state-of-the-art

更新:修订这个章节的时候,连YOLOV6、YOLOV7都已经跑出来了...... 而对于CPU设备友好的FastestDet也已经发布,当然还是推荐NanoDet,因为实在太方便了!

这里推荐使用NanoDet进行上手,对新手的友好度非常高,且官方的ROCO数据集(下面马上会介绍到)就使用了COCO标注格式,可以直接进行训练。并且作者提供了各种推理框架的部署demo,包括OpenVINO、Tensorrt、MNN、NCNN、Mega-engine等等。其他的网络包括YOLOv5、YOLO-Fastest、上述的YOLOX以及百度的PicoDet都是可以尝试的对象。两阶段网络不适合应用在装甲板检测上,大部分模型的推理时间太长无法满足实时性。

6.1.2.2. 准备数据集
  • RoboMaster组委会在2019年开源了一组赛场数据集,同时也提供了YOLO-v3利用该数据集的训练示例。不过官方提供的数据集有一个问题,这些图片都是从赛场直播相机中截取的,大部分是俯视视角(大概比较适合哨兵吧),并且亮度都较高。川大&沈航则是搭建了一个RMCV视觉开源数据站 ;云南大学&中国科学院深圳先进技术研究院也建立了自己的数据集并开源:RM-DATASET。希望最后能有一个统一的数据集平台,提供各种样例与分类的数据供大家选择,也方便大家向此社区贡献自己制作的数据集。
  • 常见的数据集规范有COCO和VOC,也有简单的txt(yolo格式)。打开数据集的标注简单浏览,就会看到类别信息、grond truth的中心坐标和宽高这几个数据,每条数据对应一张图片中的一个对象。有些数据集采用的是绝对坐标,有些采用的是归一化到[0,1]区间的相对坐标,他们各有各的优劣。
  • 官方的数据集还有对机器人的标注如工程车、步兵机器人等,如果我们只是识别装甲板可以把这些多余标注利用一个文本处理脚本去除,通过一些简单的正则表达式知识就可以完成辣。笔者这里有一个写好的脚本,也一起放在文末的链接中了。
  • 开源社区有大量的目标检测项目,想要白嫖它们就需要掌握一个最关键的技术:数据集处理。这需要一些序列化数据处理的知识,和对常见api使用的掌握。实际上是非常简单的,基本上是一个熟能生巧的过程。如COCO使用json来保存标注,VOC使用xml保存标注,YOLO系列则一般用txt文本进行保存。网络上也有大量的不同标注文件转换的脚本。不过为了提高可定制化程度,建议亲手熟悉一下python中的如dom、json和regular等库的使用,跟着处理COCO数据集或VOC数据集的范例脚本一起写,你很快就能学会标签处理了。在学习和实践过程中,你也可以将写好的标注处理和标注转换脚本保存下来,久而久之就形成一个属于你自己的常用标注工具箱,逐渐可以覆盖大部分常用的场景,复用起来非常方便。
6.1.2.3. 开始训练吧
  • 配置环境

建议使用anaconda来管理不同repo的环境防止出现各种软件包版本不兼容的情况。如果没有梯子,可以更换conda和pip的软件源(和Ubuntu的apt是一样的道理),可以让安装过程更为迅速。

由于英伟达引领的CUDA几乎已经成为了深度学习训练的事实标准,若在Linux下,配置深度学习环境你需要安装显卡驱动(默认驱动是nouveau),你可以选择在“软件与更新”中的“附加驱动”中选择需要的驱动版本。然后你还需要安装CUDA,强烈建议利用conda/pip安装cudatoolkit而不是从英伟达官网下载!因为你并不需要CUDA的所有功能,上述安装方法已经提供了深度学习训练所需的所有工具。还需要特别注意的是显卡、驱动版本、cuda版本和pytorch版本之间的对应关系。

Release Notes :: CUDA Toolkit Documentation提供了显卡驱动和cuda之间的对应关系,对于不同的cuda版本,切换到对应版本的release notes,你只需要关注表格中的最下方:

而pytorch和上述软件和驱动之间的对应关系,在pytorch官网已经帮你安排好了兼容性:Start Locally | PyTorch,若这里没有你要的版本,Previous PyTorch Versions中提供了历史版本和其他cuda版本的支持,建议按照网站上的关系进行安装。完成安装之后,在终端中通过以下命令确认安装:

python # 在终端中启动python,注意先conda activate对应的环境
import torch
torch.cuda.is_available() # 若输出true说明cuda安装好了
t=torch.Tensor([1,1])
t.to("cuda") 
print(t) # 输出tensor([1,1], device='cuda:0')说明cuda工作正常

剩下的依赖根据算法仓库中的readme走就可以了,其repo一般都会提供一个requirements.txt,直接用pip安装所有环境就完事了。也就是:

pip/pip3 install -r requirements.txt # 若在base环境安装或没有使用anaconda,提示permission denied,有可能需要管理员权限,加上sudo即可
  • 修改训练配置参数

此处以笔者队伍上个赛季使用的Nanodet-m为例,为大家介绍“炼丹”的过程。我们实验室用于训练的电脑是一台安装了1080ti的老电脑(3年前3599,现在能卖4599你敢信...来人吶把挖矿的都给我鲨了!!!萨日朗),显存12G(训练的显存就是要大!这样才能跑更大的batch并且减少访存次数,频率低慢一点没关系,宁要12G的3060也不要8G的3080)。不过不用担心,有独显的笔记本也可以用来训练,实在没办法就在tb上找服务器租用业务。

更新:30系显卡价格终于跌破发售价,为了清库存老黄又推迟40系发布,急不可耐的同学找好时机下手吧!千万甄别不要买到矿卡!另外,AMD为了追赶NV的深度学习生态,推出了ROCm对标Cuda,通过复用Cuda接口的方式实现了对Cuda的兼容(即给上层应用提供相同api,具体实现肯定是不同的),而A卡向来拥有比N卡更高的显存和性价比,不知道有没有猛士愿意弄张6500xt回来试一试(逝一逝)。

这部分内容其实是很早就已经写好了,只可惜时过境迁,在发文前nanodet已经更新了plus版本!快去github上点亮🌟吧,我敢保证nanodet是笔者用过的最容易上手部署最简单的检测网络了!这里还有笔者对NanoDet-Plus的完全解析,逐行注释非常详尽:NanoDet代码逐行精读与修改(零)Architecture_HNU跃鹿战队的博客-CSDN博客

若要继续使用之前的verison,在github页面左上方切换tag回之前的版本即可。请特别注意,本小节内容仅针对 nanodet v0.4.2,新版本不再适用,如有需要请参考上方的链接,内含NanoDet-Plus的训练细节和参数配置。

Nanodet的作者贴心地提供了config文件配置的详细说明。我们在此基础上再详细一点,配合前面所学的知识介绍一下各个参数的配置,使得大家对深度学习的部署能有更直观的理解。也建议大家在编写较大型的项目的时候,采用配置文件的方法来保存常用的、可能会修改的参数,避免程序中出现一些magic number,提高程序的可维护性和可读性。对于编译型语言,使用读取配置文件获得参数的方法还可以省掉重新编译的环节,提高开发效率。

  • saving path

这个不多说了就是log和模型文件、参数文件的保存地址,不填的话就选用默认值。

  • Model
model:
    arch:
        name: OneStageDetector
        backbone: xxx
        fpn: xxx
        head: xxx

当前几乎所有的检测模型不论是单阶段还是二阶段都可以把网络结构分为3个部分:骨干网络backbone、多尺度特征融合的FPN、检测头detect head。简单复习一下:骨干网络通过叠加卷积层、池化层、归一化层等得到特征图,然后FPN等特征融合neck会从特征图中选择若干并用某种方法来融合不同feature map的信息。单阶段网络直接在FPN处理的基础上回归出类别和bbox,双阶段网络则是先获取bbox提案再在提案得到的proposal区域运行分类和bbox回归从而得到更精确的位置和最后的分类。下面三个config是对backbone、fpn、head的具体配置。

  • Backbone
backbone:
  name: ShuffleNetV2
  model_size: 1.0x
  out_stages: [2,3,4]
  activation: LeakyReLU

nanodet提供了其他backbone如GhostNet、MobileNet-V2等轻量化骨干网络,可以直接更换name中的参数。model_size是模型的倍率,可以把他看成网络的“胖瘦”,增大这个值会扩大每一个stage的feature map大小/增加通道数。out_stage则是和之后的FPN联合使用,指明要用于特征提取的输出stage层数(请看下图),backbone一共四个stage,最上方的是原图的输入,nanodet取后三个stage输入PAN中。左侧的数字是下采样率,即特征图相对于原图的尺寸比例,一般每个stage都减小一半。

从backbone输出的每一层feature map中提取特征进行融合,选择默认参数即抽取第2、3、4层的输出

activation选择的是激活函数的类型,还提供了ReLU、RectifyReLU等激活函数。正确选择合适的激活函数也是一(疯)门(狂)学(炼)问(丹),不过据说使用LeakyReLU是目前最好的选择。在其他网络中,笔者找到如下的建议:

  • 用于分类器时,Sigmoid函数及其组合通常效果更好。
  • 由于梯度消失问题,有时要避免使用sigmoid和tanh函数(在前一层有较大激活值时变化不敏感)。
  • ReLU函数是一个通用的激活函数,目前在大多数情况下使用。
  • 如果神经网络中出现死神经元(未被激活,权重几乎为零),那么PReLU函数就是最好的选择。
  • 请记住,ReLU函数只能在隐藏层中使用。

  • FPN(Neck)

fpn:
  name: PAN
  in_channels: [116, 232, 464]
  out_channels: 96
  start_level: 0
  num_outs: 3

NamoDet使用PAN(Pyramid Attention Network)替代了最简单的FPN,不过这个模块的目的也是为了融合不同特征图的信息。你也可以尝试使用其他的FPN来测试他们的效果。这部分在网络设计中一般被称为Neck。

第二个参数设置了从backbone中输入特征图的对应通道数,注意维数和通道数要和backbone中的out_stage匹配。

第三个参数设置了输出特征图的通道数。

另外两个参数份分别是融合开始的层和输出的特征图数(看上方NanoDet的结构图就好理解了)。

  • Head
head:
  name: NanoDetHead
  num_classes: 2
  input_channel: 96
  feat_channels: 96
  stacked_convs: 2
  share_cls_reg: True
  octave_base_scale: 5
  scales_per_octave: 1
  strides: [8, 16, 32]
  reg_max: 7
  norm_cfg:
    type: BN
  loss:

改了检测头就不能叫nanodet了,直接从num_classes看起,这个参数设置了总的类别数。

input_channel是输入检测头的通道数,注意和FPN中的out_channels匹配(如果你想改的话)。

feat_channels是经过检测头部卷积后输出的特征通道数。

stacked_convs检测头部使用的卷积块数量。

如果把share_cls_reg设为True则表明分类和bbox回归共用conv block,否则会为两个任务分别启用不同的conv block,这就是精度和速度的trade-off了。NanoDet原结构是把他们作为两个分支,作为对比,yolov1就是复用了这部分特征。不过根据cnn的可视化研究,似乎回归和分类利用的是不同特征,分别训练肯定能得到更好的结果,当然性能必然没有复用ConvNet的快了。

octave_base_scale和scales_per_octave这两个参数可以不用管,是为了让配置文件兼容anchor-based的网络。

若你了解两阶段网络中的RPN或anchor-based网络,应该能较好地理解这两个参数。octave_base_scale为特征图每个位置上anchor的基础尺寸的数量,如这里为5即每个格点都有边长为5个像素的anchor。对于anchor-free方法,scales_per_octave只能设置成1,因为每个格点只会预测一组输出,nanodet作为anchor-free模型是不使用锚框的,类似于只有一组anchor。对于anchor-based的检测方法,此参数代表一个网格内有多少种不同尺寸的anchor,在设置此参数后还需要设置anchor的长宽比例)。nanodet使用的FCOS头直接把对应feature map上的每个位置作为训练样本,只要那个grid落入了任意一个ground truth范围就把他认为是正样本然后直接进行回归,而不是用生成的anchor去计算和ground truth的IOU再选取IOU大的几个anchor进行回归。

这部分不理解可以回到 5.2.6.2 介绍Faster R-CNN中的RPN部分看看。

strides是对特征图进行下采样的时候的步长,还是要和之前的参数设置对应(我们使用了三个经融合后的特征)。

reg_max是和distribution focal loss有关的参数,因为DFL要求回归出框的分布,该参数指定了离散的区间数目,5.2.6.5.7中介绍了通用离散损失。

norm_cfg则是关于归一化层的设置,type中可以选择不同的归一化方法,最长常用的是Batch Norm,也可以使用Group normlization和Layer normlization。

  • DataSet
data:
  train:
    name: coco
    img_path: YOUR_TRAIN_IMG_PATH
    ann_path: YOUR_TRAIN_ANN_PATH
    input_size: [320,320] #[w,h]
    keep_ratio: True
    pipeline:
      perspective: 0.0
      scale: [0.6, 1.4]
      stretch: [[1, 1], [1, 1]]
      rotation: 0
      shear: 0
      translate: 0.2
      flip: 0.5
      brightness: 0.2
      contrast: [0.8, 1.2]
      saturation: [0.8, 1.2]
      normalize: [[103.53, 116.28, 123.675], [57.375, 57.12, 58.395]]

这里就是训练集、测试集的路径设置了。默认支持coco、voc格式的训练集。当然你也可以使用其他自定格式的,只需要在nanodet/data/dataset中自己制作一个dataset类并提供__getitem()__和__len__方法即可。

img_path是训练图像的路径,ann_path是图像标签的路径,注意coco格式是以整个.json文件的形式保存的。inputsize设置输入的图片大小,也就是投入网络的分辨率,勾选keep_ratio则会在resize的时候保持图像的比例而不进行拉伸。

在pipeline中可以选择是否进行预处理和数据增强。

启动multi_scale的话将会对图像进行多尺度训练(不同程度的缩放,有利于识别小目标和特大目标。系数由参数决定,如这里就是0.6-1.4倍的缩放)。stretch则是水平方向或竖直方向的拉伸和收缩,rotation设置旋转的角度,shear设置裁切,translate是平移,flip镜像。最后三个参数分别可以对图像的亮度、对比度进行增强,范围可以自己设置。

normalize则是用于输入归一化的参数,不了解的同学可以参考 5.2.3 中的归一化和标准化。

测试集val的参数设置和data相同。

注意在windows训练时上的路径‘/’要改成‘\’或’\\'否则提示找不到文件。

  • Device
device:
  gpu_ids: [0]
  workers_per_gpu: 12
  batchsize_per_gpu: 64

这是设置你的设备的参数。

第一个参数是选用的显卡序号,单卡不需要更改,多卡则传入[0,1,2,3...]。

第二个参数是每个GPU并行的dataloader数,根据显存大小来选,多少G显存就填多大的值。在显存足够的情况下可以让训练速度变得更快,和batchsize不可兼得。

第三个是每个GPU一个batch运行的数据数,也就是 5.2 介绍地mini-batch数量,这可以根据你数据集的大小更改,同时也要注意不能设太大(显存限制,太大了也会影响网络参数收敛)。太小的话训练效率太低,并且由于样本数量少会引起梯度不稳定,BN的统计量太少也不能获得很好的效果甚至适得其反。

  • schedule
schedule:
#  resume:
#  load_model: YOUR_MODEL_PATH
  optimizer:
    name: SGD
    lr: 0.14
    momentum: 0.9
    weight_decay: 0.0001
  warmup:
    name: linear
    steps: 300
    ratio: 0.1
  total_epochs: 200
  lr_schedule:
    name: MultiStepLR
    milestones: [130,160,175,185]
    gamma: 0.1
  val_intervals: 10

这是用于训练过程以及对应优化的一些基本选项。

第一个参数resume若去除注释则可以继续训练权重文件,冒号后面可以不用加东西。

第二个参数需要和第一个参数配合启用,填入需要继续训练的模型。这两个参数搭配可以让你使用预训练模型(pretrained model)以减少训练开销并获得更好的收敛效果。

optimizer中的参数是pytorch提供的一个优化器模块。

第三个参数name是选择优化器,nanodet支持所有pytorch提供的优化器。SGD即随机梯度下降。

第四个参数是学习率learning rate。

第五个参数是梯度下降的加权超参数beta,我们在加速神经网络的训练部分介绍了它。

第六个是权重衰减率,即正则化中的超参数,同样在 5.2.3 介绍过。

warmup的参数分别有warmup的类型如linear、exp、constant,steps是warmup迭代的次数,ratio则是热身的学习率,可以看到要比lr小一些并且是不断衰减的(若使用linear和exp,这可以防止在初期出现振荡)。在训练初期以低的学习率进行“热身”可以稳定训练防止梯度出现异常,因为初始化的参数往往在搜索空间中处于Loss较高的位置。

total_epoaches就是总的训练次数(遍历整个数据集的次数,不是mini-batch),每个iteration会训练一个minibatch,每个epoch的iteration数为数据集的样本数➗minibatch大小。

lr_schedule可以参照torch的官方文档进行选择。name参数选择优化方法,如MultiStepLR就是选择在某个确定的结点进行学习率衰减,milestones是要选择的衰减时机,一旦到达milestone的那个epoch就按gamma来衰减学习率,前面我们也介绍了可以在学习的后期逐渐降低学习率。

最后一个参数val_intevals则是选择每多少各epoch评估一次当前模型的性能。评估得到的性能会被存储在训练log里。

如果你是第一次训练网络,这些参数请尽量不要修改,先看一看,然后直接开始训练一轮试试!

  • Evaluate
evaluator:
  name: CocoDetectionEvaluator
  save_key: mAP

用于评估结果的测试集配置。目前只支持coco evaluation。save_keys是用于评估的标准,还可以选择AP50,AP75(当然了这是标准多目标检测的评估集),组委会提供了评估集以测试mAP。当然也可以关闭这个选项,直接在实际中测试,不过这不能让你的训练变快多少。

  • class_names

这就不多说了,是用于可视化输出bbox的标签。

总之,了解这些 configuration有助于你更好的理解神经网络的框架,在模型训练完成后发现了缺陷也可以更有根据的修改这些超参。不过作者其实已经把这些参数调节的比较好了,很多参数也是当前业界从理论和实践角度检验后公认较优的。如果你是初次配置,尽量不要修改对你来说过于复杂的参数哦。

  • 开始训练并查看LOG

在配置完config.yml后就可以开始训练了。训练在小黑窗中进行,会实时展现当前的batch数、epoch数和当前的loss,还会将在每个interval得到的评估结果展现出来。训练结束后会根据每个interval的评估结果和loss的大小在我们之前指定的路径下生成最佳的参数。当然,训练得到的最终参数也会被保存下来。windows上的训练由于兼容性的问题有时候cmd会卡住不动休眠(训练yolov5也曾经出现这个情况),这时候按一下回车又能继续训练了。如果你的epoch比较大,建议关闭电脑的休眠和睡眠模式,防止训练到一半停止训练,独显直连屏幕也最好不要熄灭屏幕,如果熄屏windows有时候会认为显卡不需要工作了,直接就把显卡的供电降下去让他嗝p了(什么垃圾检测机制..)。不过即使停止训练也没关系,上面介绍过nanodet是能够恢复训练的。long也一并保存在相同的路径中,如果你发现生成的参数不对或者效果不好,你就可以track不同时刻中log的信息来修改你的模型和超参数。

NanoDet和大部分开源的检测网络都支持通过TensorBoard查看每种loss的下降趋势以及epoch间测试时的可视化结果。在自己构建网络的时候只需要引入tensorboard包并将每一项loss都加入一个用于可视化的字典当中即可。NanoDet在训练时会将checkpoints和训练日志以及tensorboard数据文件保存在work_dir下(你的训练工作目录),打开一个终端输入:

cd <YOUR_SAVE_DIR>
tensorboard --logdir ./

tensorboard的GUI会以网页的形式映射到一个本地端口,在浏览器中打开终端里弹出的链接即可。

笔者之前训练nanodet保留的截图

6.1.2.4. 评估效果

在训练过程中如果设置了中途评估,那么在log里就可以查看每个interval的结果。当然我们更希望看到直观的结果,那就直接把训练好的权重文件和模型文件替换demo中的相应文件,然后通过相机或者读取一个测试视频来实时查看结果吧。根据你使用的运算平台的不同,我们最后一般会将网络部署在ncnn、tensorrt、openvino等推理框架下。关于如何使用这些推理框架,NanoDet的仓库中提供了大部分的demo(再次强烈推荐NanoDet作为检测网络进阶的学习对象!代码编写的非常规范并且训练和部署都很方便),你也可以自己查阅相关的资料来学习对应的推理框架。一般来说我们需要将训练得到的权重文件和模型文件转换成onnx(通用的、标准化的网络描述格式),再进一步转化成其他框架下对应的参数文件。对于这些传播较为广泛的网络,github上一般都有开源的各种框架的部署教程。

装甲板识别demo的运行效果

6.1.2.5. 存在的问题
  • 已经了解过距离解算或是提前看过pnp解算的同学会发现,你这个框框没办法刚好框住装甲板鸭,而且bbox始终是一个标准的矩形,不会随着装甲板的倾斜而贴合灯条的四个角点,这可要怎么解算呢?别担心,我们还有特征点检测。前面介绍的网络在检测头部会回归bbox的四个点,那么我们通过修改检测头的结构和用于训练的数据集,让网络直接回归装甲板灯条的四个角点不就行了吗!这正是上交2021赛季开源的解算方案,他们修改了YOLOv5的检测头,并以最小化回归得到的角点和数据集中角点之间距离的MSE(mean square error)为优化目标对loss function进行修改,便得到了能检测四个角点的网络。沈航2022的赛季开源也使用了四点检测网络,而且他们还提供了基于YOLOX-nano修改的训练代码。当然,使用这种方法需要你自己制作数据集哦(RM视觉数据站上也有大量标注好的四点数据集)。一些现成的关键点检测网络也可以用于我们的检测任务,只要把关键点个数调整为4个,并把自己的数据集改成对应形式即可。

  • 不过,如果你不需要距离信息,只希望得到装甲板中心相对相机光心(图像中心)的偏角,那么到获取bbox这一步也足够了。或者你选择使用的是双目、深度相机、激光雷达。关于这些情况如何进一步处理,也请看 6.3 。

  • 笔者这里提供一个由NanoDet修改的范例以供参考:扩展NanoDet以实现任意四边形检测任务。整个模型的思想仍是FCOS式的,我们预测四个角点相对于prior中心的偏移量,即预测每个角点对中心的dx和dy。这样就不用考虑anchor,在进行匹配的时候不需要担心生成的点不是凸包使得IoU难以计算(即四个点)、出现相互关系和位置异常的情况。

  • 由于装甲板上的不同数字和图案(哨兵,前哨战,基地)在远处看起来非常相似,这对于广角镜头或需要远距离攻击的兵种可能带来不利影响,若网络出现欠拟合或模型容量不足,很可能出现误识别(特别是场地的一些灯效以及定位标签,在网络看来可是和装甲板非常相似的)。对于这种情况,可以考虑和传统算法一样的two-stage方法:先利用网络定位装甲板,再专门设计一个分类网络对装甲板中的数字进行检测确认是否是真实装甲板并分类。

这样的方法看起来完全是多此一举,然而,只检测一类可以大大提高网络参数的针对性和准确性,原本你让网络根据微小的差异判断不同分类,就相当于在它小小的脑子里塞进了它不该承受的知识。但若先将所有装甲板视为一类,这就大大减轻了它的负担。这也是小目标检测、难例区分等问题中的常用trick,先把所有小目标当作一个类别检测出来,然后再将ROI送入分类网络进行细分。

基于CNN的目标检测方法最显著的特点就是泛化性能好,对于复杂的光线条件能够通过扩充相应数据集的方法来解决,也可以加入负样本防止误识别。其部署也不是什么难事,对于新队伍来说更加友好(直接使用官方数据集然后找一个开源的算法也能取得不错的效果)。并且习得特征(训练得到的卷积核和全连接的权重)的表示能力要大大强于人类专家手工设计的特征。在当下的RoboMaster赛场上,神经网络算法已经和传统识别分庭抗礼,之后的事,大家应该也能猜想到结果。


6.1.3. 设计key-point detection方法的装甲板检测

阅读这本小节内容需要对目标检测有相当的认识,请根据自身情况酌情选择。这些观点部分来源于实验验证,部分来源于笔者的测试以及猜想,很多都是经验性的知识,不能保证正确性,有不同的看法很正常,欢迎讨论。若你已经对目标检测和pytorch编程较为熟悉,建议阅读。不过请注意,即使是使用一些开箱即用的算法进行修改(YOLOv5-face、PolarMask等),就已经可以达到很不错的效果了!如果你是第一次尝试,不宜在这些琐碎上花费太久的实践,尽快改好网络(工作量不是特别大),训练起来成功部署才是第一要务。为了设计一个适合RoboMaster比赛中的装甲板识别检测框架,我们可以从以下5个方面入手:backbone、neck、head、label assignment、loss,同时可以考虑data augmentation、pretrained model、knowledge distillation等其他可能提升网络性能的模块。

装甲板识别的任务实际上对于目标检测来说较为简单,分类数目并不多,而且特征明显(灯条+数字/图案),用一些小容量网络也可以做到95以上的mAP,所以除了雷达之外,one-stage的模型一定是我们的首选。因此,下面的讨论都是基于上述前提展开的。

6.1.3.1. Backbone

考虑到边缘平台的运算能力(大部分队伍没有用上AGX这种性能怪兽),我们不可能选用过于庞大的骨干网络。backbone作为整个网络中最耗时的模组,串行的特征提取流程限制了其并行优化能力,所以在它就很大程度上决定了网络的推理性能。相比自己设计模组,笔者建议使用现成的常用骨干并在其上进行删改。一方面可以减小工作量,另一方面可以利用在大型数据集上的预训练权重。

大厂用NAS(Neural network Architecture Search)搜出来的(实际上也只有大厂有能力搞)模型往往拥有比手工设计的模型更好的性能,如现在在ImageNet上霸榜的卷积模型都是基于EfficientNet加上各种训练trick得到的。选用NAS搜出来的轻量化模型是一个很好的选择。而常见的部署于边缘平台和移动端的MobileNet、ShuffleNet系列也是此任务的不二之选。GhostNet、魔改的CSPNet(YOLO最喜欢用的)系列对于速度和精度的权衡也还算不错。如果你打算使用Jetson的开发板,大可以尝试使用标准卷积而不是DW卷积的模型。

重参数化在RepVGG提出之后就非常的火热,在注意力当道的时代甚至出现了RepTransformer。它相当于用inception的结构进行训练,在部署的时候再从宽度上将模型”锤扁“,将几个分支通过运算等效成单个分支。对于分支结构特别多的网络,像DenseNet、CSPNet这种,利用重参数化方法可以大大提高运算速度。不过要想自己对网络进行re-parameterization还是有一定难度的,需要对网络的部署、归一化层的原理有相当的理解。建议使用现成的重参数化网络,或参考其结构,在上面进行改造。

对于Backbone的修改,主要集中在模型的输入部分。轻量化网络常常会给输入添加一个slice层,把原图切成4、9个小块再堆叠在一起,使其变得和网络中间一样拥有多个通道和更小的feature map。这种方法对于精度来说必然是有一定损失的,但是能够大大提高速度。还可以给slice前的图片添加位置编码使得在切片之后不同的小块仍然可以拥有对相对位置关系的感知。通过stem cell也可以迅速降低特征图的分辨率,即卷积使用更大的stride,这个结构适合加在2~3个block之后,否则会导致网络还没来得及提取纹理信息就直接被大步长忽略。

另一种减小计算量的方法是在浅层网络使用大核的DW卷积,别看核增大会导致这个层次运算量的增加,实际在引入1x1融合不同通道特征的时候运算量还是小于组卷积核标准卷积!并且大核还有一个好处就是更大的感受野,可以汇聚更大范围内的信息。其实在高层也可以使用大核,感受野的扩大有利于引入前景和背景信息的交互。在大部分辨率的输入下尤其适合使用。并且,大核对于CPU型设备还有一个好处,就是能够减少访存次数,可以一次性把一个卷积要算的参数和输入全部拉到高速cache里,不用反复切换一层中需要的参数。这也是”大参数量并不意味着高latency“的原因,网络结构的设计在很大程度上会影响推理速度!RepLKNet就是通过DW+重参数化+稀疏计算将大核卷积的计算效率提高了一个数量级,并且还拥有更高精度!但是目前对于大核的重参数化,除了旷视的两个模型,其他还不够普及。

对于CPU型的设备,可以将backbone后段的卷积全部替换成DW卷积,底层不适合是因为还没有形成有效的语义信息,特征太低级。GPU适合并行运算所以在除了在瓶颈层之外少用DW卷积。在backbone的中层,还可以多多使用bottleneck,用通道数换feature map的大小。后段也可以将一些1x1卷积替换成SE module来建模不同通道的关系以降低计算量。

当然我们不可能把上面的方法全用了,现有的模型是经过NAS暴搜了不知道几千个GPU时或者手工精心调整了N多的epoch才得到的,固然有它的合理性。花活整太多很可能达不到效果,速度甚至还要往下掉。

6.1.3.2. Neck

当前标准的设计是抽取backbone的后三个stage输入neck,最后一个stage经过一层卷积进行上采样之后形成四个初始的分支,每个分支都有往上和往下的卷积+concat操作以实现特征融合,最后分别接入各自的head里。笔者认为RM中的目标都是以较大尺度为主的,除非使用长焦(>8mm)的镜头,否则除了极少数的怼脸情况(怼脸也用不上自瞄了),不会出现那种占满屏幕三分之一以上的目标。因此,上采样得到的那个分支完全可以直接砍掉,我们用不着回归那么大的目标,甚至由于anchor的分配策略,在多尺度训练的scale vary开得比较小的时候(因为前述的原因也根本没必要开那么大),最上层的head几乎得不到什么训练,可能传回去的全是负样本。

要追求速度的话,可以参考YOLO-Fastest的设计,直接把不同尺度特征之间的融合通过上/下采样+element-wise add实现,精度的损失是肯定的。上下采样之后,利用一个SE module计算两个层次的特征的重要性然后加权融合也是ok的。同样,可以大量替换标准卷积和组卷积为DW卷积。

Backbone有神经网络架构搜索,Neck也有,NAS-PAN就是典型代表。不过对于追求速度的我们来说,NAS-PAN多少有点太冗余了。

6.1.3.3. Head

检测头的基本设计其实没什么好说的,一般轻量化的头就是2~4个block的卷积,极端的会做到1个block甚至只使用两个卷积核。最极端的那种,直接让3个卷积头共享参数,这种一般需要2个block不然啥特征也学不到。

接下来就是回归和分类是否解耦的问题。多篇paper指出,分类用到的特征和回归相比基本是不同的,可视化结果表明,回归特征一般集中在物体的边界上,即物体和外接矩形的角点附近;分类特征则一般在物体的多个部位有极值响应。所以使用解耦头的动机就是提高精度,让它们各学各的特征。ICCV 2021 oral TANet就深入分析了分类任务和回归任务的不同,指出即使使用解耦头也仍然是用一个anchor或一个参考点尝试得到两个任务的解,然而分类和回归对应的最优回归起点(其实就是anchor,anchor-free的模型则是中心点)都是不同的,所以作者整了一个Task Align Module并用Task Align Learning拉近两个任务的最优anchor。代码已经开源,有兴趣的可以去了解一下,精简一下它的TAHead应该不会损失太多速度。

特别注意的是,装甲板有三种颜色8种分类(12345哨兵前哨站基地),如果不解耦颜色和贴纸将有19种(只有建筑物和哨兵有紫色),加上熄灭的装甲板那更是不得了。首先要面对的就是样本分布不均匀的问题,很可能会带来长尾效应,少见的样本不仅学不好还会影响其他样本的学习。另外这么多的分类会让检测头的通道数增加,每多一个通道就是一组卷积的运算量,对于标准卷积来说更是复杂。所以颜色和类别是一定要解耦的,虽然解耦会增加额外的运算量,但是可以大大降低网络的学习难度。并且,在此情况下只要增加颜色分支所带来的额外运算量小于去掉分类数量带来的运算量,速度反而会提升(掐指一算,不是板上钉钉的事吗)。而且颜色特征属于特别好学的那种,颜色分支甚至只需要一组卷积即可,这样运算量又降了一截。分类数减少之后,Softmax的输出也会更显著,用于平均的分量更少了。

之前有一些模型会把最后的卷积换成全连接,以提高不同特征之间的交互,但有实验已经证明没有这个必要了,1x1卷积在很大程度上可以替代全连接,并且速度还要快一些,矩阵运算更少,CPU自带的向量指令可以直接计算1x1卷积。

一种想法:由于颜色只是某个通道的值,是不是根本不需要通过分类的方法来区分?为了帮助网络更好的学习,我直接在backbone后面还没进入neck的时候就加一组卷积让它输出颜色,使得后面的网络能够更专注于图案与数字的分类和角点的定位。有可能卷积都不需要,直接把特征丢进来就是一个sigmoid。或者加一个SE module,让它选出最能代表颜色的两个通道,我就认为一个代表红色一个是蓝色,硬train一发。或者干脆不让网络输出颜色信息,那灯条在他眼里就是发光/不发光两个状态,在后处理中通过判断角点处的像素值来确定颜色(为了稳定,可以加一个高斯窗)。

重点来了,我们应该如何设计检测头来回归装甲板的四个角点?(实际上是灯条的四个角点)笔者认为有这几种设计的思路:

  1. 基于矩形框或单个参考点,再回归出四个角点的offsets
  2. 视为特征点检测问题,不管框

区别一下这两种思路,它们的出发点是不同的。第一种思路还是经典的通用目标检测,先用一个外接矩形把装甲板框住,然后在框内再确定四个点的位置(可以是一阶段直接把框和点一起预测出来,也可以采用框得到ROI再加卷积得点)。第二种方法则不考虑IOU,使用的是人体关键点检测、面部关键点检测的思路,建模的是点和点之间的关系。装甲板不像通用目标检测,我们应该从它横平竖直的特点出发,考虑这两种思路;另外一个需要注意的问题是,四个点形成的外接四边形没有办法很好地包裹中中间的数字/图案。

  • 基于框或点的想法是很直觉的,把原来的方法当作两个互不相干的任务。并且这样设计对于后处理非常方便,直接使用预测得到的矩形框进行NMS,就不用任意四边形的IOU计算了(不过这似乎有一种训练和评估不一致的感觉,两者之间有一些gap)。注意到四点标注的数据集一般标注的是装甲板的四个角点,那我们会想到在读取数据的pipeline中加入一个计算装甲板的外接正矩形的流程,用这个矩形框作为预测矩形框的bbox。但是这个bbox有一个缺点,就是它并不能把整个数字图案包含进来(数字和图案几乎占满了装甲板的整个纵向,灯条只有1/2),虽然高层的卷积有更大的感受野,但多少有一些特征没有对齐的意味。想要解决这个问题可以在头部或只在最后一个block加入空间注意力,让特征之间引入更多交互。更适合此问题的方案是引入可形变卷积与池化,相信它们很容易学到要将关注点放在未被包含进bbox或角点四边形内的这部分图案特征。

我们前面也提到,用于预测的特征通常在bbox的边上有极大响应。再者,从我们刚刚生成GT的方法看来,最终预测的角点都应该在bbox的某一条边上,除非专门标注矩形框以包含整个数字/图案。有一种任意四边形的检测方法叫Gliding vertex,就是先回归出一个矩形,然后在四个顶点上各预测各预测一个offset值以得到任意四边形,相当于一种coarse-to-fine的检测思路。如果提供了额外的bbox标注(包含整个图案),那么应该比较适合使用两阶段的测试,先预测bbox然后再把bbox当ROI提取特征输出角点(和两阶段网络一样,做ROI-Align,SPP等都可以)。对于用外接矩形生成的bbox,似乎就没有必要预测了。我们直接预测四个点,然后计算出外接矩形用于NMS和标签分配,可以获得更高的效率。

当然,四个点的预测可以使用FCOS式的point-based,也可以从anchor box上回归offset。由于有倾斜的装甲板,我们可以布置倾斜的anchor,比如15°为步长,布置7组,即可,再斜就倒了。这样应该可以大大提高召回率,对小目标和倾斜目标也很友好,多预测出来的那些框就交给NMS吧。

对于offset的建模,不管是anchor-based还是point-based(anchor-free在某种程度上其实就是基于点)可以采用GFL的方法,回归出一个分布而不是单一的位置(单一的位置相当于狄拉克分布,冲激函数)。回归出点的分布对于后续的预测有很大的帮助。但是这里回归出的offset的分布,和基于关键点的heatmap有有些不同,因为此处是在每个回归起点上都预测一组分布,而heatmap则是预测每个位置作为角点的概率。另外,由于角点回归是否准确会大大影响之后目标距离的解算(大部分都是用SolvePnP),可以尝试把更层的特征引入,比如增加一个attention,或者直接用亮度二值化得到的mask输入回归分支,防止回归出的点不在灯条区域(很显然的思路嘛,角点肯定是在灯条的高亮度区域的末端)。

这个预测问题和旋转目标检测、任意四边形/多边形检测有很多相似之处,可以关注一个做旋转/通用目标检测的大佬:yangxue,应该能给到你很多启发。

  • 关键点检测也有两种流派,基于点回归和heat-map的,我们这里关注的是后一种。CenterNet其实算是回归和heat-map之间的过渡模型,它会预测目标中心位置的heatmap,然后选取top-k个响应点,分别以这些点为回归起点预测检测框的四个边界(我们可以直接预测四个点8个值)。再往heatmap的思路走就是纯热图,摒弃回归的方法,预测5张heatmap,分别代表物体中心和四个角点出现的概率,然后根据位置关系对响应点进行分类从而形成装甲板预测。一般每个点还要预测一组offset,因为heatmap的分辨率没有原图那么高,多少有点便宜。这时候,亮度值mask就尤其适合角点的定位细化,要是能直接把高亮度条状区域的端点也输入网络让它更关注这些区域,则可更好地预测热图。

对于纯概率分布图的方法,关键问题是如何将属于同一个装甲板的点归类到一起。从几何的角度出发,可以在后处理的时候直接把离中心点最近的k个响应留下来,选出符合距离阈值并且概率分数最高的那个作为其中一个角点。还可以把这些点都保留(数量不能太多),然后跑一个暴力搜索,筛选那些合理的四边形(其实就是手工特征,比如生成装甲板的两条边是否平行,角度差大不大,和前述传统检测内容基本一致,不赘述)。可以利用内积和外积以加速运算(角度和距离可以用这两种运算一起搞定,比单独筛选快得多,不用走分支结构)。基于框的方法一样可以在loss中加入几何约束,我们在loss部分再细说。

如果不想用手工特征,加入attention也是不错的选择。可以在数据处理的时候用四个角点算出装甲板中心点的位置,然后增加一个二值的mask。用这个mask引导注意力,使得装甲板的中心点和对应的四个角点拥有最高的注意力分数,后处理的时候就选这些响应最大的点。只要权衡速度和精度,注意力方法end-to-end的特性应该是更受欢迎的。

因为点数并不是很多,人体关键点面部关键点动辄几十的数量和我们的问题没法类比,点与点之间的关系也非常复杂,因此它们常用的嵌入向量、把点之间的关系建模成图的方法属实是没有必要。

6.1.3.4. Label Assignment

这部分就着重探讨一下前述的回归的思路应该怎么样进行标签分配。动态标签分配被OTA带火之后已经算是检测网络的标配,不管用在什么架构上都能涨点,对于我们的场景也同样重要。标签分配通过选出那些优质的预测结果,将GT分配给它们从而实现优者更优。不用担心会有anchor无法接受正样本的监督,神经网络超强的抗噪能力以及丰富的数据增强,加上前层的参数共享都能让各个anchor逐渐学习到最适合自己预测的目标。

基于heatmap的方法不进行需要标签分配,因为每个位置输出的就是该点存在特征点的概率。用的最多的方法就是让预测的分布和GT分布尽可能一致,选用一些度量分布的方法或者软标签(对于分布而言一般用GT,加一个高斯窗)。

暂时不关注框和角点怎么产生,只聚焦于哪一个anchor适合回归当前的目标。在标签分配中我们一般把匹配损失(称作cost以区分和loss的不同)分为IoU损失、类别损失和GT与anchor的中心距离损失。鉴于我们回归的是角点,还可以视情况加入角点的回归损失。

不管用什么方法得到了外接或包围bbox,我们都可以用bbox和GT bbox的IOU替代角点形成的任意四边形与GT的IOU计算cost。虽然bbox的IOU和四边形IOU之间存在gap,但这并不会影响我们追求更大IOU的优化目标。显而易见的是,随着外接矩形框的IOU逐渐增大,四边形IOU在总体趋势上也是跟着增大的,只不过我们很难衡量两者之间严格的相关关系,以及用这种替换方法产生的损失函数是否可导、损失和IOU在不同情况下的趋势。并且,相对于任意四边形IOU的计算,这非常非常方便。即使是旋转矩形要计算IOU(如现在常用的skew iou),过程都非常复杂(还好有人已经实现了),对于NMS和标签分配都有很大的效率损失。并且,旋转IOU有一个很大的缺点就是存在不可导的地方。至于任意四边形的IOU,计算时要考虑的情况和细节就更多了。哈工大威海的同学自己编写了任意四边形的IOU以实现标签分配,值得敬佩。由于IOU loss基本已经替代了以往通过L1 loss等优化bbox的方法(就是评测和训练不一致),使用IOU loss的合理性基本不用质疑;此外,采用IOU loss之后,和之后的角点距离损失之间的正交性恰好也会提升,防止出现重复优化的情况。使用IOU损失还有一个好处,就是它拥有尺度不变性。还要考虑一个问题,就是使用的IOU损失类型,这会影响其和角点距离损失之间的正交性以及评估的有效性,笔者没有定性分析过也不好妄下结论,建议用D-IOU或S-IOU即可(新出的一些精心设计的IOU loss可能反倒不适合,因为其设计目标是矩形框检测,而这个实际上并不是我们的优化目标,siou都怕是一些矫枉过正的嫌疑)。

类别损失则有两种,颜色和图案。若使用了解耦检测头的设计,则特征不一致的情况会得到很大的缓解。若没有解耦,则在计算cost的时候就需要特别注意平衡IOU损失和类别损失了:定位准确的anchor往往没有很高的分类分数,反之亦然。这就需要在计算cost的时候采用更精细的策略,最好是通过系数衡量类别和定位的”掌握程度“,从而动态调制两者所占的比重。

GT和anchor之间的距离显然是一种直观的先验知识,尤其是point-based的网络基本都提出过/依赖于这种损失范式,我们相信那些优质的预测框应该位于anchor的中心,这也符合卷积核和感受野两者的形状在空间上对称的直觉。加入这部分cost也可以防止出现开头所述的”坏学生不被关注”,导致此anchor直接摆烂。

在某种程度上,对于基于anchor的算法,IOU损失会包括GT和anchor的距离损失。

最后是由于引入角点需要增加的角点距离损失。虽然这看似(不是看似而是就是)和IOU损失会有冲突,但是由于使用了外接矩形替代,两者相关性会下降。另外,IOU损失和距离损失在学习的不同阶段比重也是不同的(即使你没有加入调制系数)。对于关键点检测,我们常常使用wing-loss以保证不同角点之间的优化程度能够相互平衡,而在装甲板这种四个角点都差不多的情况下,其实不太需要担心不同角点回归失衡的情况。因此,smooth-L1足堪使用。若想要增大表现差的预测的损失,可以考虑加入L2。另外,我们同样应该保证损失的尺度不变性,对于不同尺度的目标应该乘上尺度相关的系数以修正目标尺度变化带来的度量不一致的影响(大尺度的目标,即使角点偏离10个像素也说明回归的足够好了;反之小尺寸的目标即使改变一个像素也会引起巨大的变化)。

讲完以上四种cost,应该考虑它们在总cost中所占的权重了。从总的角度看来,类别损失和IOU+角点损失应该势均力敌,距离损失大概为IOU损失的一半。这种想法也是经验性,不过有一些论文专门研究loss比重的问题,可以参考。为了适应网络在不同学习阶段的掌握程度,在不同的轮次应当采用不同的系数:目标检测网络早期学习注重类别后期注重回归精度应该是毋庸置疑的结果了;另外,颜色损失于学习初期可以给予更大的权重,之后逐渐下降但也不能忽略否则会出现红蓝部分的情况。距离损失在回归任务已经较好习得时也可以降低,因为此时每个anchor已经逐渐找到了自己应该预测的角点和bbox。至于IOU和角点,遵循总体原则即可,可以尝试让角点cost占比稍微超过IOU,毕竟IOU对于微小的角点移动不是那么的敏感。

毕竟这么多组对照笔者也不可能一个个试,都是经验或猜想(吹水)。有时间的卡多的同学可以多跑点消融实验,整理一个表格。

另外,Dynamic SR-CNN提出选取的top-k个anchor中的K也应该是可变的,在初期应该让尽可能多的anchor接受正样本监督,随着学习的逐步提升,在减小k的数量以更精细地调整网络中的参数,这也是一种可以尝试的策略。

6.1.3.5. Loss

其实和标签分配差不多,只不过没有了GT和anchor的距离损失。最终的损失函数就是颜色+类别+IOU+角点。在这一部分,笔者推荐给角点的权重提高一些,大概为IOU损失的1~2倍。优化器推荐选用Adam,这是最稳妥、最不需要担心参数调试的方法了。如果不是自己编写的网络,是从其他算法中修改而来的,那么它原来应该就已经选用了默认优化器,拿来直接用就完事了。若你想要尝试一些新鲜的优化器,可以考虑使用Adam的数种改进如AdamW、NAdam和RAdam等,在pytorch中均有提供。如果你认为自己的炼丹技术以及炉火纯青,可以挑战一下SGD。

gradient clip和模型参数滑动平均也是推荐使用的训练策略,都能帮你在最大程度上稳定训练过程。训练初期sigmoid常常可能出现NAN,尝试使用tanh替换,或者加入一个梯度死区。激活函数尽量选择通用的RELU,可以尝试LeakyRELU。SiLU、Swish、ELU等在某些平台、设备或推理框架中可能尚未支持,检查确定可能之后再整花活。另外,不要忘记在log和所有用到除法的地方加上!

另外需要注意的是标注角点的顺序。如果数据集中的点都是按某种顺序标注的那没什么问题,若是随机标注的则需要在分配标签、计算loss之前先为标签的角点或预测的角点进行排序!推荐顺序为tl、tr、dr、dl,从左上角开始顺时针排序。

6.1.3.6. 其他
  • 数据增强

在装甲板的定位任务中,中间的数字/图案特征明显是起了非常重要的作用(尤其是解耦颜色和分类之后更甚,即使两者没有解耦,数字特征仍然是区分装甲板和其他类灯条的发光物体最重要的部分),而灯条则是区分数字/图案和其他类似物体(特别是场地中的定位标签、大块白色不明物体,都常常被识别成装甲板)。因此,针对这两种情况,推荐加入一些上述的负样本,如场地上的灯条、定位标签、没有贴贴纸的装甲板、熄灭的装甲板(可以作为其他类,颜色中加入熄灭)。

多尺度可以很好地增加多样性,根据数据集中的样本形状选择合适的缩放大小,和前述一样,除有特殊需求外最大尺度不宜超过画幅的1/4;小尺度的目标也要保证再resize后的图片里至少占有共40个以上的像素(COCO把长或宽小于32的目标视为小目标),除非有特殊需求。

旋转增强大概开到20~30,不排除数据集里本来就有一定角度的装甲板(爬坡、视角不同等),开太大容易学到倒的装甲板,甚至把一些场地光效和不知道是什么的玩意识别成装甲板。

伸展和收缩增强应该根据图像预处理的pipeline和相机的分辨率进行设置!这一点非常重要,此参数的设置取决于图像在被输入网络的时候是怎么被resize的。是直接不保留原图比例暴力resize成输入大小,还是保持比例加padding?抑或是保持一边的比例...... 伸缩的参数尽量和预处理设置的一致!

务必关闭镜像增强,因为数字和图案可是不会倒转的!

另外几个关于像素值的增强(对比度、色调、亮度、饱和度等),也要保持在可能出现的场景范围内。相机调参往往不会特别剧烈,而且至少要看清装甲板图案又不能让灯条过曝,因此这几种增强也要根据工况进行调整。饱和度注意不要调太低不然会编程灰色。尤其是色调,如果变化得太大可能会导致紫色装甲板被识别成红色或蓝色!(我们学校哨兵开局狂射对方前哨站的场景仍然历历在目...)

对于综合的数据增强如mixup、mosaic等,都可以尝试使用。

  • pretrained model

一般指的是在大型通用数据集上预训练的分类模型(对我们来说就是backbone)。浅层卷积核学习到的都是一些简单的纹理和边缘特征,预训练模型相当于帮我们把这部分特征学好了。不过对于装甲板检测来说,更高层的特征几乎是用不到了。笔者建议将backbone的前两个stage冻结,开放剩下的部分以进行训练。一般来说,数据量越大,解冻的层数就越多。在浅层特征已经学完的情况下,整个网络往往会收敛得更快。不过也有论文指出,通过一些特殊的方法和精心设计,从头开始(train from scratch)同样能达到相同甚至更高的精度,并且在更短的时间内收敛,有兴趣的可以查阅相关论文。一般来说,还是推荐使用预训练模型的。开源的算法中在配置文件里可以修改是否使用预训练的参数。

如果刚刚上手,直接用预训练模型就完事了,冻结整个backbone(这一般也不需要你设置)只开放neck和head。若你修改了backbone的结构,其实仍然可以使用预训练模型,只需要将那些未修改的地方对应即可。pytorch保存权重的文件是以字典的形式存储的,对此感兴趣的同学可以在pytorch文档中查阅相关内容。

不知道有没有卡多的同学愿意用现有的所有数据集帮大家整一个预训练模型?

  • knowledge distillation

相关的理论就不再介绍了。对于分类问题,相对于GT,教师模型可以学生模型带去更丰富的类别相关信息;对于回归问题,尤其适合那些回归分布的方法如GFL,同样可以提供更多的关于定位的信息。one-stage模型非常适合知识蒸馏,两阶段模型还要考虑RPN生成的anchor不一致的问题,不过DETR在提出之后,有很多两阶段模型借鉴了一对一匹配的范式,如Sparse R-CNN就学习一组自适应anchor,者可以很好的解决ROI对齐。

另一种可行的方法在有标注数据上,利用教师模型进行标签分配,GT仍然使用标注。NanoDet的AGM就是利用了这个方法。当然,蒸馏最显著的优势还是在大量无标签数据上的调整。理想情况下,利用大型高精度的教师模型你只需要开着相机采图小模型就能跟着学,爽就完事了。

将模型修改成知识蒸馏模式并不复杂,只需要将教师的预测当作GT即可。



6.2.能量机关激活

注意,2023赛季之后,能量机关的图样已经更改,请参考规则手册。但我们仍然可以通过形态学操作为主的方式进行识别,故下面的内容还是有一定的参考价值的。对CNN的方法来说没有很大区别。

6.2.1. 传统方法

和装甲板识别一样,传统方法也需要人类专家来设计特征。那么我们照葫芦画瓢,一起来看看能量机关的特点吧。能量机关的扇叶相比装甲板要复杂一些,未击打过的扇叶中间有一个箭头型流动的灯条,末端则是一个闭合的矩形边框。已经激活的装甲板的两侧灯条也会亮起,并且中间的灯条停止流动。到这里,两种状态的区别已经很明显了。能量机关的中心还有一个机甲大师的标志性icon:右下角有个小闪电的”R“。

待激活的扇叶(右下)和已经激活的扇叶(正上)

相比梯度特征,形态学特征更加显著。那么,我们可以通过区分扇叶两侧的灯条是否亮起来分辨已经激活的扇叶和未激活的扇叶。预处理的一种pipeline如下:

整个流程与装甲板识别类似。同样,若采用HSV空间进行通道分离,则不需要右侧的分支,直接利用Value进行亮度阈值二值化。下面还会介绍另一种更为巧妙的使用了漫水法 floodFill() 来区别扇叶,直接将滤波那一步替换为 开运算 + floodFill() 函数,这样不需要进行后续的形态学处理。


  • 预处理

和装甲板识别类似,我们首先要找到方法来描述这些灯条,因此在滤波之前的操作都相同。

二值化后的效果
  1. 使用方法一,则在滤波之后,我们考虑使用闭操作让未激活扇叶中间流动的箭头灯效和末端的矩形灯条框连接成一个整体(若闭操作还无法实现连接,则可能是结构元素设置的不好,可以尝试直接加大结构元素的大小或进行多次闭操作[注意运行闭操作之后要运行使用相同结构元素的开操作以保证轮廓不不扩大]),得到一个“锤子”状的图样。在能量机关激活点几乎是正对能量机关的,光线干扰会比较小,有利于预处理的进行。
常规方法预处理完成后得到的效果
  1. 使用方法二(漫水),则在二值化后先使用开元算使得图像中的各个部分分割开来,随后通过漫水法把只有一层轮廓的部分全部变成一种颜色,直接去除了流动的箭头灯效和能量机关外部的斑点等:
漫水法的效果,箭头灯效、R标和背景均被化为相同的颜色,截取自csdn:海人007的博客

显然漫水法得到的图样更有利于我们进行进一步的处理,但是其时间消耗要比方法一大的多(漫水法需要遍历每个像素),不过后处理会更简单,最后的速度还说不定(笔者还没测试过两种方法的速度,那就再挖个坑吧大笑)因此我们在之后会并行介绍这两种方法,分别标注为1和2。

注意,cv::threshold()可以选择是否反转像素值,因此才会出现上面这两幅图中区域颜色不同的情况。选择封闭区域还是打开区域为1对后续角点定位精度有较大的影响,这取决于你测量的真实世界下能量机关扇叶末端装甲板的大小和测量方式。


  • 选择未被打击的扇叶
  1. 方法一即根据常规方法得到的图案,使用 findcontours() 得到轮廓。因为已经激活的扇叶内有3个子轮廓,未激活的扇叶只有一个子轮廓,通过hierachy中的信息就可以筛选得到待打击的扇叶了。对只有一个子轮廓的那个父轮廓的子轮廓应用 minAreaRect() 拟合一个旋转矩形即可。笔者曾经看到使用模板匹配和svm的方法来筛选轮廓从而得到扇叶,那些额外操作都是根本没有必要的(这里也提醒大家要掌握算法的底层原理,去了解函数的每个参数,对这些常用的方法要有充分的理解)。
findcontour后的示意图,遍历hierachy中的子轮廓
  1. 方法二会更简单一些,这里又有两种思路:

  2. 1.先进行闭操作,把结构元素设置的足够大,已经激活扇叶的三块区域就会连成一块,随后使用 findcontours() ,并计算轮廓的面积,显然已经激活的扇叶的轮廓面积远大于未激活的扇叶。那么大小合适的便是待击打的扇叶末端的装甲板了。此时对那个面积合适的轮廓使用 minAreaRect() 得到一个旋转矩形,就得到了目标。

再次闭操作得到的效果,左下角的轮廓仍然是一个漂亮的矩形
  1. 2.直接使用 findcontours() 并对所有轮廓应用 minAreaRect() 得到一个旋转矩形数组。对此数组进行遍历寻找那些长宽比例接近2的矩形,这些是扇叶末端的装甲板。再将这些装甲板和扇叶上的长条形轮廓拟合出来的矩形一一匹配,检测装甲板中心到这些长条中心的距离,根据能量机关的特点我们会发现未被激活的扇叶末端的装甲板中心离其他长条中心的距离都要远大于已经激活的末端装甲。因此,根据这个关系我们就能找出待激活的那片扇叶了。
方法2.2的图示,未激活的扇叶末端装甲板到其他条带的距离至少是已经激活的装甲的2.5倍
  1. 3.不过此方法对预处理的准确度要求较高,若预处理后背景中仍有长宽比或面积和扇叶/装甲板相近的大块区域,将极大地影响此法的稳定性。为了有效地分离前景和背景,可以尝试第五部分介绍的自适应阈值方法等。

这样,我们就得到最终需要打击的装甲板了。不过,能量机关是处在不断运动的状态的,因此还需要对其运动轨迹进行预测和拟合。

  • 能量机关中心R标识别(可选)

为了获取能量机关的中心以拟合其装甲板转动产生的圆,可能需要通过识别中心的“R”来确定圆心。一般击打能量机关选用6/8mm镜头(12mm则R比较清晰),R基本呈现为一个发光的圆,并不能明显的区分其边界。同样用形态特征的思路检测,给出一个大小搜索范围,寻找那些形状接近圆,面积与先验值相近的轮廓。找到之后,检验其是否在扇叶的中心线上以筛选其他光斑。还可以利用扇叶末端装甲板的宽度和扇叶长度的比值,先推算出中心的位置,然后再划定ROI,寻找R标。如果R比较清晰,模板匹配/SVM等,或者直接利用神经网络定位R标。在出现两片扇叶以上的时候,就可以利用扇叶中心线的角点确定R标的位置了。

因为检测中心的耗时并不大,一般建议识别“R“标以提高圆拟合的精度,可以将通过不同方法确定的圆心进行加权以得到最优的结果。


6.2.2. 基于CNN的目标检测方法

前面我们已经介绍了基于检测网络的装甲板识别,能量机关的识别和前者的流程基本一致,我们首先根据扇叶的发光特点来选定要识别的范围并制作数据集,然后选择一个网络对其进行训练,得到最终的模型。

这里给出两种基于网络的方法:

  • 旋转矩形检测框

此方法首先识别整个扇叶,再对此ROI运行轮廓检测得到扇叶末端装甲板的信息。常见的目标检测网络只需要四个参数就可以确定一个矩形框,因为他们都是水平放置的,而旋转bbox则需要一个额外的参数angle来确定位置。定义angle一般有三种方式,分别为:

  1. OpenCV定义法,,x轴顺时针旋转碰到的第一条边为width。也就是说,width可能比height更长!这也是在做传统方法识别的时候,需要手动归一化角度表示的原因。

注意在4.5版本之前,定义为,x轴逆时针旋转碰到的第一条边为width。

  1. 长边定义法le,或。

  2. 长边定义法lePlus,,因为处于水平位置的框比较多,较不适合将90度和-90度作为回归边界。

关于这几种定义的优劣,参考这篇文章:关于旋转框定义的一些理解和感想。这个专栏也有很多关于旋转检测的理解。另外,使用MMRotate可以轻松构建各种旋转目标检测网络,还支持导出onnx,非常方便。当然也可以直接检测装甲板和扇叶,然后根据检测框之间的包含关键,确定待击打扇叶的位置。虽然可以将能量机关激活点处视角近似为正视,但是旋转框生成的四个点定位精度肯定是不如传统算法来得高的。

先运行检测得到ROI,再使用传统方法解算

直接检测旋转框的效果;图源大连交通大学TOE战队

  • 识别扇叶的特定角点

此方法和装甲板角点识别相同,得到需要击打的扇叶后直接对末端的四个角点进行解算。不过这里要注意最好不要只选择扇叶末端的装甲板的四个角点,因为未激活和已经激活的角点的特征差别并不是很大,区别仅在于内侧的两个角点旁是否有灯条。因此应该选择更特殊、特征更丰富的点作为用于回归的特征点,否则遇到场地光线和其他灯光的干扰很容易出现漏检或误检。具体网络的设计请参考6.1.2和6.1.3。

直接回归角点,不同颜色代表不同分类
上述选点策略的效果,画面旋转是因为云台正在跟随目标转动,稍显卡顿是gif帧率原因;图源哈工深南工骁鹰战队

6.2.3. 圆拟合

计算出能量机关的外接圆(一般来说使用末端装甲板的中心为边缘,方便预测后进行击打),目的是为稍后的运动预测和轨迹拟合计算角速度。下面介绍的方法都是在像平面中的圆拟合而非三维空间,这是因为在能量机关激活点处,机器人云台的高度和能量机关的中心高度相差无几,而且两者间7m左右的距离使得云台在旋转的时候,能量机关平面相对于相机来说基本仍然保持正视,因此直接假定相机的中心线始终垂直于能量机关平面。

若能在三维空间中对圆进行拟合,理论上得到的精度必然是高于前者的。不过,由于当前应用最多的测距方法仍然是单目的PnP算法,因此在7m距离上,解算出的平移矩阵(相机光心相对目标的距离)的误差在0.1米级时常用的事,第三个维度上的波动对于曲线拟合是十分不利的(对相机和IMU的标定又提出了更高的要求),更何况增加一个额外的维度对算力将提出很大的挑战(第三个维度的引入导致我们还需要确定能量机关转轴的方向)。若通过双目、深度相机等方法解决了这个问题,同时找到了较好的优化算法、手眼标定也达到了相当的精度,那么可以对其进行尝试。

基于此假定,我们不会保存装甲板中心在三维空间中点的位置,而是在为每一帧都确定一个圆,该圆在图像坐标系下的位置会随着云台的转动(即相机位姿的变化)而变化。

  • 追踪每一片扇叶

预处理后得到的结果并非只有待激活扇叶对我们有用,那些已经激活的扇叶同样可以在后续各个阶段提供信息,如角速度、位置等。因此,我们应该保存每一片扇叶的信息并持续对它们的位置进行追踪。最简单的方法是根据两帧之间的角度变化量或扇叶中心点(一般在预处理时得到)的位置变化量来确定。6.4.3中有更详细的介绍。

通常没有必要使用目标跟踪方法,处非你希望云台在切换瞄准目标的时候仍然能够让不同帧之间的扇叶保持继承关系。

简单起见,我们只在跟随目标装甲板的时候更新追踪器,一旦跟随目标发生切换(如激活了目标,或能量机关发生重置),就需要清空之前跟踪的装甲板,重新初始化跟踪器。

  • 使用扇叶和装甲板的比例关系确定圆心

能量机关的尺寸信息已经在规则手册中给出,因此,利用识别得到的装甲板尺寸和扇叶尺寸,我们可以计算出圆心的位置,再根据装甲板的中心从而确定整个圆:

蓝色长度由预处理得到的外接矩形或轮廓确定,距圆心距离从规则图纸上获取

当然,倘若在第一步中已经识别出圆心,可以用此法得到的圆心与前者进行加权,获取更准确的位置。另外,当有多片扇叶亮起的时候,我们可以通过数个扇叶中心线的角点确定圆心,多种结果都可以进行融合。

即使是将其当作正视图处理,通过数帧末端点拟合圆的方法也已经几乎被弃用(即在像平面上利用圆周上的数个点拟合)。原因之一是拟合需要采集一段时间的数据,在这个时候云台不能发生运动(除非利用IMU的信息将运动后产生的角度变换量记录下来,通过重投影以获取云台在初始位置时新产生的点在图像上的位置)。

  • 使用投影变换以获取能量机关正视图(optional)

本节开始就强调我们将在每一帧都重新计算圆。虽然由相机位姿的改变引起的能量机关视图变化量很小(不超过20度),但要提高精度我们还有一种方法:获取像平面上的图像到能量机关正视图的投影变换矩阵。制作一个能量机关正视图的模板,在其上取四个点(相距越远,结果越准确,受到噪声的影响越小);然后在预处理得到的结果于相同位置取点,计算两者的变换矩阵。可以取多组点分别计算结果进行平均以降低随机误差。经过投影变换后的能量机关视图就是真正的正视图了。而得到的投影变换矩阵,就包含了标准模板到当前视图的旋转变换信息,稍后计算角速度的时候直接就能用上。

把每一片扇叶都映射到标准位置,比如下方左侧正中央朝上的位置。如此投影产生的变换矩阵中的子矩阵(旋转矩阵)就包含了旋转变化信息,由此可以计算出帧差。

左侧为标准模板,为了凸显效果特意夸张了右侧图像的形变

6.2.4. 旋转函数拟合

小能量机关是匀速转动的,因此在得到拟合圆之后实际上就可以直接预测击打位置了。规则手册并没有给出小能量机关旋转速度的误差范围。如果你认为自己的算法能够给出更精确的结果,那么可以自行用角速度拟合出其匀速具体数值(官方参考值:10RPM)。而大能量机关的旋转策略如下:

能量机关转速按照三角函数呈周期性变化。速度目标函数为:,其中 spd 的单位为rad/s,t 的单位为 s,a 的取值范围:0.780~1.045,ω的取值范围为 1.884~2.000,b 始终满足 b=2.090-a。每次大能量机关进入可激活状态时,所有参数重置,其中 t 重置为 0,a 和ω重置为取值范围内任意值。能量机关的实际转速与速度目标函数的时间误差在 500ms 内。

而其旋转方向则在能量机关每次启动时是任意的。

由于一般难以确定旋转的初相,我们需要拟合四个值。下面先解决能量机关旋转方向和旋转速度的问题,然后介绍四种可能的求解方法。

6.2.4.1. 计算旋转角速度

大部分情况下使用帧差法即可。前一小节中拟合出的圆在这里可以派上用场。一种方法是根据帧差得到两帧间末端装甲板的位置差,根据采样时间,计算出速度,再由拟合出的圆的得到该速度在切向上的投影并除以圆半径得到角速度,稍显复杂。另一种方法则是求出两帧间装甲板的旋转矩阵,OpenCV中的cv::warpAffine()可以派上用场。之前在拟合圆的时候也说过,如果将扇叶变换到标准模板上,得到的投影变换矩阵中也包含了旋转信息。不管用什么方法,我们都可以通过相同扇叶/装甲板在两帧间的旋转关系得到角度差,同样根据采样时间得到角速度。

另外,我们还需要确定角速度的方向。最简单的方法是通过两帧间装甲板和能量机关中心这3点,通过叉乘以判断旋转方向。根据右手定则,我们一般以前一帧的位置为起点,藉此确定绝对方向。

叉乘的其他作用还包括判断4个点是否在一个平面内、计算立方体的体积、获取平面的法向量等。内积也是计算两条线之间位置关系的利器。通过这两个工具我们很容易把点之间的关系转换为线和面的关系,而不需要真的构建一条线或一个面。

最后,由于装甲板定位的噪声(四个点会飘动)以及中心确定的误差,角速度难免有波动,常常可以达到0.5~0.7 rad/s级,因此还需要对采用数据进行低通滤波以去除高频分量。5.3.1里介绍的方法都是可以用的,笔者推荐使用基于变化量权重的指数加权平均,即计算前帧数据和当前帧数据的差值并由此得到当前帧在更新中占的权重(显然变化量越大权重越小)。

6.2.4.2. 拟合旋转函数
  • 离散傅里叶变化DFT

只要你学习过信号与系统,第一时间想到的应该就是这个方法。持续通过帧差采集角速度信息,当采样总时间至少达到一个周期时进行离散傅里叶变换获取信号的频率谱,找出最大的响应峰即能量机关的旋转频率。由于其旋转频率随机,因此至少保证采样时间为可能周期的最大值() 。OpenCV中恰好提供了cv::DFT()的接口,只需要构建一个一维的Mat(或者直接使用OpenCV对vector封装得到的Vec)并将采样点保存在其中,传入函数后就可以得到返回的结果。遍历结果中的频率谱,找到最大值对应的下标即要找的频率。然后在对应频率上查看相位谱,即可得到采样开始时刻的初项。

很经典的图片,如果不断增加项数,上面的级数最终会逼近一个方波

使用DFT有一个好处,就是不用对原始数据进行滤波。只要计算得到的角速度是无偏的,方差不是那么的离谱,那么得到的频谱中的最大分量始终会是正确的结果。另外,需要注意频率和角频率的转换,两者相差。偏置b在频率0处获得,由于可能对低频噪声敏感,可以在频率0处使用高斯窗进行加权以获得最终的偏置。频率峰处的幅值的倍,就是时域的幅值。

计算得到的角速度无偏时,虽然有数据波动,但是时域的总能量应当保持不变(角速度的均值和真值相同)。那么根据傅里叶变换的能量谱定理,我们要找的分量的能量始终保持定值。因此,即使加入了低通滤波器,也只是将高频能量滤除,并不影响低频处的响应。对于频率峰幅值的两倍才是时域幅值的解释:一般进行变换的时候只取半边谱,而的傅里叶变换为。

从信号处理的角度出发,还可以使用相关滤波的方法解决拟合问题。互相关运算的结果代表两个信号的相似程度。我们首先确定,在范围内任意选取幅值和偏置,得到一个模板函数。将规则手册内的范围离散成10个(当然可以更多,精度会提高)值,对于每一个值都通过给模板函数设定一系列的相位,并用其与采样得到的数据进行互相关运算。得到的结果最大者就是我们要找的频率和相位。接下来通过零均值化确定偏置b,最后对幅值A进行拟合。

理解这种方法,需要对线性空间和傅里叶变换有较为深入地学习。傅里叶变换将时域上的函数转换到以频率基表示的频域,因此在用不同频率的模板函数与采样数据进行相关运算的时候,实际上就是计算采样数据和模板函数的内积。只有两者包含的分量方向一致的时候,相关运算才会得到最大值。

由于以上两种方法都需要较为复杂的pipeline,代码编写起来也比较麻烦。最关键的问题是,拟合速度较慢,因为根据离散傅里叶变换的性质,我们至少要采集一个周期的数据才能开始拟合。对于2021年的能量机关来说,这两者是比较适合的方法,因为那时候的所有参数都是固定的,不需要进行大量的计算。像后者,只需要用固定的频率不同的相位和采样数据进行相关运算即可。由于偏置和幅值已经给定,也不需要对这两个参数进行后续处理。深圳大学RobotPilots战队在2021青年工程师大会上分享了相关运算的方法,戳这里。

虽然这里的方法基本已经弃用,但是了解它也大有脾益,能够更好地帮助你构建旧方法和新方法之间的过渡。学习一些其他知识也是不错的。

  • 非线性最小二乘法

看名字就知道和最小二乘法很相似,只不过我们要解决的是非线性的拟合问题。这个问题是非线性优化中的一种,而解决非线性优化问题有一个非常出名的C++库——Ceres Solver。这是Google开源的非线性优化库,这个名字是为了纪念高斯在1801年利用最小二乘法处理观测数据,计算出谷神星(即Ceres,一个希腊女神的名字)的轨道并完美地预测了其在太阳另一侧出现的时间。相比SLAM中常用的优化库g2o,Ceres的文档更丰富,上手难度也要低一些,因此推荐使用。

Ceres的官方文档里详细介绍了非线性优化问题和其工作原理,并编写了相应的Tutorial以供入门使用,这里不再赘述。神经网络其实也是一个优化问题,确定损失函数,网络中待学习的参数就和本问题中的四个待求解参数一样,最后确定优化方式(梯度下降),开始求解。Ceres的求解方式也是一样的,确定优化问题,编写损失函数,然后选择优化方式就能开始求解了。

需要注意的问题是,Ceres在求解的时候(尤其是样本点数多,模型又很复杂时),对于CPU的占用往往很高,其内部对多线程调用进行了大量的优化,因此可能会导致其他线程的CPU时下降,使得我们的识别线程、采样线程等性能下降。因此要合理选择待优化参数,并分阶段解决问题以降低此情况的影响。

最简单的优化方法是牛顿法,还有高斯牛顿以及一系列基于GN法的改进,它们统称拟牛顿法(利用低阶微分近似高阶微分)。还有一些基于信赖域的方法,能够分析对高阶导数拟合的精确程度从而选择迭代方向。随机梯度下降是最经典的方法,也是深度学习中使用最多的方法。

  • PROSAC

在5.5.5.4中我们介绍过PROSAC,对于任意用样本点拟合数学模型的方法,采样一致性都能够帮的上忙。为了保证最终迭代结果的准确性,至少确保采样点数在100以上。同时,随着时间的推移,可以移动这个滑窗不断重新开启迭代,并对每一次的迭代结果进行加权。此法对点的离散程度要去较高,如果没有什么点落在ground truth附近,获得的结果往往不尽如人意。

  • 梯度下降(自适应滤波)

学习了这么多神经网络的内容,对梯度下降还会陌生吗?这里,我们的目标函数(损失函数)就是拟合得到的曲线和真值之间的误差,L1 loss足堪大用,想要尝试其他损失函数自己修改即可。以笔者的经验,学习率的设置大概是10e-3~10e-2的数量级,mini-batch的大小为100(大小为100的滑窗,还不够100的时候就一直往里面加)。对于每个mini-batch,在两帧数据之间迭代尽可能多的次数(这里不存在过拟合,就是在线学习)。我们拟合的参数只有四个,这种方法和5.3.1.6介绍的自适应滤波是一致的,只不过这里解决的是非线性问题。同样,你可以把他当作一个单层感知机。四个参数可以随机初始化,也可以用规则所限定区间的一半(都放在正中间,直观上看平均的收敛时间应该会比较小)对参数进行初始化。此法的效果也非常不错,在2s内基本可以收敛,用C++手撸一个即可,而且优化空间非常大,比如自行加入momentum和RMprop,甚至直接上Adam。学习率和mini-batch都可以细调。因为对于每个mini-batch实际上可以迭代任意的轮数,如何优化以最大程度提高迭代速度、收敛速度是最关键的问题。有兴趣的同学可以尝试对其进行多线程的实现。个人猜测最快收敛速度大概能达到1s级,相比Ceres非常有竞争力。

既然有梯度下降,5.2.4.2中的更高阶优化方法自然是可以尝试的!由于不单单使用了Jacobi,还引入了Hessian等函数的高阶变化趋势信息,拟牛顿法等应该能以更快的速度和精度找到损失函数的最底点。


到这里,就只差最后一步了。利用拟合出的旋转函数,我们将所有延迟和滞后时间加在一起放入前者,计算出这段时间过后目标所处的位置(实际上得到的是目标装甲板的位置,因为我们需要4个点进行解算)。和装甲板识别一样,我们需要把待打击的位置通过位姿解算,转换成电机的控制信息发送给电控,请看 6.3 。

insight:实际上我们在拟合一个模型已知参数位置的公式(你也可以直接称其为模型),对于这个模型,只要能合理地建立出代价函数(也可以叫目标函数/损失函数),那么就可以通过一般的优化方法求解。如果这个优化问题有一些特性如其为凸函数、可以转化为凸问题,以及SDP问题等,那么我们就能找到更高效或精度更高的方法进行求解。一般来说,如果优化问题有解析解,那么我们一般先求取解析解作为初值(解析解一般没有考虑带噪数据),然后在初值上再应用一步或多步的非线性优化以获得精确解。



6.3.测距与深度估计

在前面的解算中我们已经知道了装甲板、能量机关等在图像坐标系中的位置,那我们要如何通过相机成像模型中的投影关系把它们还原回世界坐标系中的三维点,得到装甲板相对于相机中心的距离和偏角呢?接下来就会讲解距离测量算法。由于比赛中涉及到的测量和3维重建算法不是很多,并且和相机成像模型有一定的重合,故删去了第五部分的相关内容,统一在第六部分进行介绍。

6.3.1. 传统的定点解算方法

指的是使用单目相机通过标定得到的内参以及真实物体的尺寸对距离进行估计的方法。

6.3.1.1. 比例法

最简单的方法,即根据小孔成像原理,利用像和物体的相似关系,并对物体的位置给予近似或约束,从而求解。我们已经在5.6.4.2中大致介绍了原理。现在针对不同的问题尝试利用此法计算距离。

  • 装甲板

显然应该利用灯条作为比较目标。由于装甲板的安装固定,而机器人底盘始终和地面水平,因此发生移动的时候只会在地面水平的方向转动,而不会发生俯仰方向的翻转。这就保证从相机的视角观测灯条时,不论其怎么运动长度都不会发生大的变化(正视和侧视时相差很小,装甲和底盘的倾角大小为75度,参考设计规范手册)。在获取装甲板灯条在像平面上的长度后,就可以根据和真实长度的比例估算距离。当然,此法强烈依赖灯条定位的准确性,对于选点策略的要求比较高,否则装甲板轻微移动,角点位置却偏移很多(传统算法形态学操作、轮廓的确定以及椭圆/旋转矩形拟合是三个影响最大的环节,神经网络则依赖于数据集和检测头的设计)。要提高精度,则可以根据灯条倾斜的角度改变比例系数,灯条接近垂直的时候则认为装甲板正对相机,因此应乘上一个修正系数();灯条倾斜说明装甲板没有正对相机,因此修正系数应该在0.96~1之间,根据角度变化。不过这种方法显然也只是权宜之计,因为你无法保证对方的机器人是否在水平地面上移动(你的机器人也可以在斜坡上,不过自己的倾斜可以通过IMU修正)。

  • 能量机关

能量机关的选点策略比较丰富,因为转角多直线也多,可以多选择几段直线进行估计取平均值。而且,能量机关激活点几乎是正对能量机关,使得我们可以假定获取的图像是正视图。另外,可以通过6.2.3中介绍的投影变换方法将视图“摆正”,从而使得此法也能获取相当的精度。其他的缺点和可能出现的问题同装甲板。

  • 其他定位特征

如场地中的定位标签等,对精度要求不高的场合可以使用比例发确定距离。倘若能对物体的姿态加以较严格的约束,这个方法也可以获得较高的精度,而且使用起来非常方便,计算开销几乎为零。

6.3.1.2. PnP

同样在第五部分第六小节进行了介绍。我们一般直接使用OpenCV提供了solvePnP()接口。这里分析一下几种不同的求解方法(通过传入枚举参数选择)。

官方文档在此:Perspective-n-Point (PnP) pose computation

  • SOLVEPNP_ITERATIVE :利用非线性方法优化,通过迭代找到重投影误差最小的一组位姿变换;同一个平面需要四个点,不同平面则需要至少六个点。增加点数将提高精度,但增加开销。精度较高。同面四点的典型运行时间为1ms(i7-8625U)。
  • SOLVEPNP_P3P :速度最快,只需要求解三个二元二次方程组,典型开销为40-60μs,精度稍差,但稳定。需要3个以上的点。
  • SOLVEPNP_AP3P :P3P的改进,将平移矩阵和旋转矩阵分开求解,并用迭代方法解二次方程。典型开销和P3P相同,稳定性更强,精度差不多。需要且只能用四个点。
  • SOLVEPNP_EPNP :对ITERATIVE方法进行了改进,利用高斯-牛顿法迭代。速度相比前者小1-2个数量级,精度更高一些,但不太稳定,正对容易出现奇异情况。要求选点准确。需要3个以上的点。
  • SOLVEPNP_IPPE :新奇的方法,也是基于迭代。至少需要四个同平面的点(新增的点也必须在同一平面),速度超快,稳定性尚可,精度一般。求解策略为先求解平移向量t再求解旋转向量r(opencv返回的旋转描述是angle axis形式)
  • SOLVEPNP_IPPE_SQUARE :前一种方法的进一步改进,要求已知点必须是正方形的四个角点,增加这个限制后速度更加炸裂,其他同上。

  • SOLVEPNP_SQPNP :很新的工作,对噪声的容忍程度特别高,稳定,但是速度很慢,典型运行时间达到2~2.7ms。

对于装甲板,显然就选灯条的四个角点,可以根据尺度的大小选择方法:对于视野中较小的、倾斜程度较高装甲板,选择稳定性强的解算方法;反之,选择高速的方法。但不论如何,对于我们的应用场景来说,提高角点的定位精度才是最重要的,不能让两帧之间的角点定位出现很大的偏差,否则得到的结果会有较大的抖动(实测100-300mm都很常见,远距离时若不进行特殊处理,常达到500mm,这也和相机的分辨率有关,分辨率越高定位肯定越准,否则远处物体隔一个像素结果都相差甚远)。

能量机关由于距离基本固定,拍摄效果又好,可以选择速度最快的方法。而且选点策略多、数量广,根据一片扇叶的位姿便能够推断其他扇叶,因此可以做结果的融合。

定位标签、矿石、障碍块等的定位基本非常准确,拐角多特征多,用PnP无压力。

PnP返回的结果是一个目标到相机的旋转矩阵和平移矩阵,对平移矩阵取欧几里得范数就可以获得目标到相机的距离,我们还将通过平移矩阵把物体的装甲板等的变换到云台系,稍后再根据IMU的数据变换到绝对系以供运动预测使用。目标的旋转矩阵对反陀螺有很大的用处。

另外,虽然精度较高的PnP典型时间开销都在ms级,但是相比神经网络动辄十几二十ms的势头,可谓九牛一毛。再对比电机的动作、子弹的飞行,更是几乎可以忽略不记。而且巧用多线程能较好地掩盖开销,保持相同的吞吐率(注意latency仍然增加,因为无论如何每一张图片都是要经过所有流程的,但是throughput可以保持不变)。还有一个需要注意的点是,调用solvePnP()的时候会传入相机畸变矩阵的参数,因此在解算之前不需要自行再矫正!

值得一提的是,今年的CVPR(2022) best paper颁发给了做二维图像姿态估计的论文,这也是在神经网络连续几年霸榜后难得的一篇扎实的应用理论工作。这也一定程度上说明了大会的风向标。说不定相关的顶会顶刊在之后会回归初心,出现更多的相关工作。


6.3.2. 双目和深度相机

  • 双目相机

笔者对双目不太熟悉,可能等回学校之后会完成对双目双光相机的测试后再行补充。若你对这方面有自己的理解,欢迎补充!

4.5已经大致介绍了工作原理。通过两个相机拍摄图片的视差,可以构建深度图像。为了提高深度,没有必要对所有像素都计算深度,在确定ROI之后再计算内部点的深度可以极大地减小开销。为了保证输出的数据稳定,可以将ROI内所有点的深度都计算出来,取平均作为估计值。在RM 教程 6-特征点,双目相机与光流这篇教程中,作者非常详细地介绍了将双目相机应用到深度估计中的所有知识,包括双目位姿估计(相当于两个单目,根据图像估计相机在拍摄两幅图像时的位姿变换,是单目SLAM的基础)、特征点匹配、对极约束、双目标定和双目测距。我们在前面的教程中几乎已经覆盖了学习双目所需的所有知识,相信理解起来不会有太大的难度。

若使用双目,则需要找到两个相机同一时刻拍摄图片的世界系下相同点的匹配,进而计算时差。这在相机曝光较低时很难实现。一种可行的方法是同时在两张图片上运行装甲板检测并先用pnp计算出粗略的位置,并在世界坐标系下匹配这些装甲板,得到的同一块装甲板,再对灯条和数字进行两种图片上的匹配,计算精确的视差。

  • 深度相机

没难度,直接从D通道提取对应位置的深度,为了准确可以对ROI取平均(一般是对装甲板区域,注意能量机关扇叶是有空洞的),距离远了之后不好使。

  • 补充:单点测距

思路来源于大连理工大学凌bug战队,相关设计的青工会分享在这里:大连理工大学-青工会分享;结构、代码开源:RM2021-大连理工大学-凌BUG-指哪打哪激光测距云台开源

这是一个非常有意思的思路,对于打击静态目标非常有效,尤其是英雄击打固定目标很适合这种策略。以英雄为例,在发射机构下方安装一个单点激光测距模块,两者通过一个舵机连接,方便云台pitch变化时后者仍能保持原来的角度。这个激光测距模块实际上就是一个点型的激光雷达,很多线阵激光雷达就是将一个单点测距模块固定在电机上,在电机旋转同时于每个角度都进行测量,加上电机编码器的数据即可获知某一帧数据是在哪个角度获取的,这样便可以建立2维的深度地图(扫地机基本使用这种设计)。

瞄准时,首先转动云台使得激光器发出的光点对准目标,测得测距模块到目标的距离;随后,根据距离、弹速和加入空气阻力的弹道模型,计算出枪口需要抬高的补偿角度(参见6.6弹丸重力补偿),从而实现“指哪打哪”。此法为半自动瞄准,因为需要操作手先手动将激光打到目标上,才能进行后续的解算。

有同学可能回想,为什么我不能先在图像上检测出打击目标然后再用激光测距呢?这是因为检测得到目标后,你只能获取物体相对相机的偏角,由于没有深度信息,物体可能存在射线的任意位置上,因此激光测距器不知道应该对准哪个位置,无法完成测量:

激光传感器不知道应该转到哪个位置,因为总能产生交点

但是,如果使用双目相机,组合之后就会产生两条射线,并且两个相机之间的变换矩阵我们是知道的。而激光测距传感器显然和相机不处于一个平面上,因此射出的激光和前两条光心与物体的连线恰好是线性无关的!这样,通过三个向量的组合就可以获知空间中任意位置,从而完成对准。因此使用双目相机的话可以通过此法完成更精确的定位。


6.3.3. 激光雷达与相机联标

这里不介绍如何标定两者的位姿变换矩阵,假设它们的关系已知,我们看看如何将点云投影到2维图像上。此法一般用于面阵激光雷达。当然不排除将激光雷达装载在其他机器人上的情况。

点云可视化(上)和投影到二维图像的效果(下),不同颜色表示深度不同;图源KITTI DataSet

我们可以得知点云中每个点到激光雷达的偏移量,而激光雷达和相机之间的位姿转换矩阵可以通过标定获取。因此,将所有点都施以变换,就能得到这些点在相机坐标系下的位置。还记得我们在5.6.4.1介绍的重投影吗?这时候,我们只需要用相机的内参(已经通过相机标定获取)构建相机成像模型(即前面章节中的相机系的像平面的变换),利用该模型把所有点都投影到像平面上,就能得到上图所示的效果了!这时候,我们还可以将图片像素点的RGB信息反向投影回点云,如上图(上)所示,每个点都增加了对应位置的颜色,使得点云包含的信息更加丰富。两者在融合后可以实现优势互补:二维可见光图像更加稠密,并且拥有颜色信息;三维点云稀疏,但有深度信息。

对于我们的应用来说,首先通过相机拍摄图片,并运行2D目标检测;得到检测输出的ROI后,把相同时间采集到的点云数据投影到图像上,寻找那些落在ROI内的点,从而利用其深度信息。为了减小误差,常取深度的平均值。特别注意,相机和激光雷达肯定不是重叠排布的,因此两者必然会产生视差,对于一些物体的边缘,激光雷达可能会出现采集到背景或出现遮挡而出现空洞的情况(从相机角度理解也相同,因为两个传感器的位置不同;类比人眼,极端情况下遮住左眼,右眼还看得到,如果你的左眼是LiDAr右眼是相机,遮住右眼左眼仍然可以采集点云,然而右眼实际上啥也没看见,点云和二维图像就会出现错配)。

面阵雷达一次可以采集很多的点,因此能够通过“广撒网”的方式覆盖相机图像的每一个角落。在 6.2 中介绍的单点激光测距实际上也是一个激光雷达,因此同样可以将其采集到的数据投影到图像上,只不过只有一个点。我们使用深度信息主要是为了给pitch轴设定重力补偿值,若一定要使用单点激光测距传感器进行测距,可以使其在pitch方向上转动(转动的范围需要覆盖图像的纵向),采集一系列的点并投影到图像上。显然这种方法过于缓慢。

可我们刚刚不是说点式激光测距传感器不能确定物体到底在哪个位置吗?这是因为我们增加了只采样一次的前提。若能多次采样,实际上可以实现和面阵激光雷达相同的效果。还是使用6.3.2的那张动图,通过逆向思维,一种直观的解释如下:对于测距传感器在每一个角度测量到的深度值,我们都可以计算出该点是否在光心和物体成像点(像平面上的点)形成的射线上,因此只需要连续改变测距传感器的角度,不断采集数据,直到计算出的点恰好位于射线上,这个位置就一定是物体所在的真实位置。

真的要用也不是不行,也应该采样离散抽样的方式,不过速度还是有点慢,只适合静态目标


6.3.4. 深度学习方法

本小节是实验性的方法,有兴趣可以自行尝试。

一张depth map很容易让人联想到卷积神经网络生成的feature map,和语义分割任务。而基于卷积神经网络的深度估计方法也已经兴起许久。语义分割是逐像素的分类任务,而深度估计则可以看作逐像素的回归任务,要得到图像上每一个点距离相机光心的距离(再次感慨神经网络真的是万能的)。不过,单目深度估计对于标准的CNN设计来说可能有些困难,如果仅仅只是模仿全卷积网络构建一个深度估计网络,可能并不会达到太好的效果。单目相机本就缺乏深度信息,并且前述网络设计中并没有添加任何关于物体、背景的额外先验知识,这使得深度估计变得更为困难。从前面介绍的PnP方法便可以得知,我们至少需要获取同平面的3个定点对才能进行解算,而用神经网络去拟合P3P的解算过程,精度显然是不如解析计算高的,甚至完全不会收敛。

实际上单目深度估计已经成为一个较大的研究方向,有兴趣的同学可以参考:单目深度估计方法综述。即使是现在,也尚未开发出一种精度达到我们应用场景需求的估计方法,由于物理限制,单目估计的深度误差总是较大的。

双目更适合深度估计,因为两幅图像中包含了视差和遮挡相关的信息。一种可能的架构是5.4.2中介绍的FlowNet的架构。当然,为了添加先验,肯定要在不同的阶段通过精心设计的模块将相机成像模型、两个相机的位姿变换等信息引入网络。

另一个较大的问题是,深度估计的训练数据集较难获取,更换相机、更换镜头甚至改变像距都会影响GT的标注,本质上还是要归结为位姿解算需要大量的先验信息:相机内外参;而这些先验又较难在网络中引入,并且训练标注的代价过于昂贵,内外参发生轻微变化都可能导致无法收敛。

最后是速度的限制,深度估计需要一个单独的网络来解算(是否能共享前面的feature map?不过融合到目标检测网络中于检测头部并行,有可能在一定程度上可以利用分类特征)。


得到和目标的距离之后,就可以利用它计算弹道模型了!这时候直接通过 6.6 介绍的方法获得pitch需要补偿的角度,便可以把最终的云台目标角度发送给电控,让电机转到对应位置实现自动瞄准。但是不要忘记能量机关和装甲板都是会移动的,能量机关的运动尚且可以预知,而装甲板的移动却是对方机器人的操作手决定的,因此我们还要根据目标的运动模型,估计出一个合理的超前量,让电机瞄准预测位置而不是当前帧所处的位置。



6.4.轨迹预测与跟踪

我们采集到的数据只可能是离散的,并且对数据的处理需要时间,而最终利用计算结果对机器人施加控制的时候,也仍然需要响应时间,弹丸在空中飞行更是占了大头。而在数据采集到执行机构完成动作的这段时间里,有可能控制目标已经发生了新的变化,不再位于原来的位置。因此我们需要建模目标的运动状态,预测其在一定时间后的位置,从而实现超前打击。6.4.1 和 6.4.2 针对的是装甲板识别,因为能量机关的位置固定且到此时已经获取旋转函数。

请注意,本节的所有模型都是建立在我方机器人在绝对系下静止的基础之上 的。这时认为底盘(yaw转轴)就是绝对系的原点(底盘仍然可以转动,小陀螺也没问题)。

6.4.1. 运动学建模

5.3.2.2中已经介绍了运动学建模的基本方法。这里我们以匀速模型(CV)为例,将加速度视为零均值的高斯噪声,利用卡尔曼滤波器融合观测值和预测值,实现运动学建模。这里假定我们已经得到目标在绝对系下的坐标(已经完成深度估计,通过相机到世界系的变换矩阵将目标点转换到绝对参考系下)。

匀速模型的状态空间建模(一阶微分方程组)

其中$\mathbf{x}=\begin{bmatrix} x\ y\ z \end{bmatrix} \mathbf{\dot x}=\begin{bmatrix} \dot x\ \dot y\ \dot z \end{bmatrix}\omega(t) \sim N\left(0, \sigma^{2}\right)$,是服从正态分布的加速度过程噪声。因为我们处理的是离散数据,差分获取的离散形式如下:

为采样间隔时间,它是可变的,取决于模型两次更新的间隔。对于我们的场景来说,就是两帧数据的更新时间/相机的采样间隔时间。将上式展开,就得到了我们的过程模型,即状态根据数学模型的推演在下一时刻的变化:
为状态转移矩阵,为过程噪声作用矩阵,它们展开之后如下:

其中$\boldsymbol{x_k}=\begin{bmatrix} x_k\ y_k\ z_k\ \dot x_k\ \dot y_k\ \dot z_k\end{bmatrix}$,均为绝对系下的数值。

而量测模型的获得取决于深度信息的获取,若使用单目PnP获取(即深度),观测方差取决于PnP的解算方法和选点策略,此过程的非线性程度很大甚至难以建模;使用双目相机则因为三角测量的模型比较清晰,容易对噪声进行建模;使用激光雷达也是如此,因为激光雷达的观测模型也容易获取。在这里为了方便大家理解,暂时假定相机系下获得的目标的三个坐标位置相互独立(当然这肯定会降低模型的准确性,而且无法得到无偏的最优估计)。

令目标在绝对参考系下的坐标为量测向量以提供观测信息,则观测模型为:

是绝对系下目标位置的观测方差。由刚刚假设的相机系观测方差独立,即:

通过时刻相机系和绝对参考系的旋转矩阵(这个变换由IMU的数据获知)将相机系的观测方差其变换到绝对系下:

有同学可能会问,坐标变换不是还有一个平移变换吗?为什么不用考虑进来?这是因为平移变换不会改变协方差阵。回归到方差的定义,方差描述的是数据点之间的离散程度,而施以平移变换则将把所有点都在空间中移动一个距离,然而离散程度不会发生任何变化。因为旋转需要用新的坐标表示方差,所以要把原来的协方差投影到新的坐标轴下,因此要考虑旋转矩阵。

这样就拥有了量测更新的全部信息,下面看看整个滤波器更新的流程:

上述即Kalman Filter的“黄金五式”,第四步还要用上卡尔曼增益的计算:,一轮更新就结束了。你只需要把这些矩阵运算用代码实现,就可以获得自己的运动模型。OpenCV在video/tracking模块下提供了卡尔曼滤波器的实现,详情见cv::KalmanFilter Class Reference;它还提供了包括粒子滤波、增强/无迹卡尔曼滤波等的实现,有需求的同学自行查阅文档。

关于运动学建模,还有一些细节需要注意。IMU的采样频率一般为1kHz或500Hz(以C型开发板板载的BMI088为例),而装甲板识别和解算的帧率显然远低于前者(100~200Hz)。因此,过程模型的状态更新和量测更新需要异步进行,保证在没有量测的时候,控制量目标仍然可以根据过程模型的更新实时变化,以达到插帧平滑的效果。具体实现为当有新的量测数据进入时进行状态转移和量测更新,没有量测数据时仅进行状态更新。后者更新状态的时候,只执行式(1)和(2),不断迭代,一般要加入的限幅防止状态发散。若没有采用相机+IMU的硬触发,还需要对齐两者采集数据的时间戳保证相机系到绝对系的位姿变换正确。

至于采用其他的运动学模型,对于滤波器的设计来说,只是过程模型有所不同。为了得到更精确的量测更新,应该根据你使用的解算方法和传感器,对相机成像模型、测距的观测噪声作进一步的分析。


6.4.2. 估计超前量

前一小节我们对目标的运动进行了建模,根据得到的运动学模型,我们假定现在已经能够获取目标运动的较为可靠的信息。首先我们需要对所有可能产生滞后和延时的环节进行统计。列表如下:

延迟环节 延迟时间 重要性 备注
相机曝光 由设置的参数决定,典型时间为 3~8h 忽略
和下位机通信 和线程设计有关,一般有波动,范围在3~15 ms 不重要 接收IMU数据
装甲板识别 传统算法约1~3 ms,神经网络约5~20 ms 不重要
距离解算 PnP典型开销为1 ms 忽略
和下位机通信 和线程设计有关,一般有波动,范围在3~15 ms 不重要 发送控制数据
CAN通信(电机) ~2 ms 忽略
电机动作 和电机控制算法的性能有关,~100 ms 较重要
弹链空程 ~100 ms,取决于弹速 较重要
子弹飞行时间 取决于弹速和距离,300~700 ms 极其重要

子弹飞行时间最少将占据2/3以上的时间,因此距离的解算和弹道模型是否准确是超前量估计最重要的问题。可以选择利用下位机的定时器、C++中的chrono库对软件部分的时间开销进行定量测算,然后实时更新需要超前的时间。也可以对“忽略”和“不重要”的项目总耗时进行拟合,而其他的部分还是通过计算得到。最暴力的方法是只考虑子弹飞行时间,将剩下所有开销视作黑箱,通过实弹测试获取数据,把超前量当作距离目标的函数进行拟合。

获取超前量后,根据前面得到的速度,用超前时间乘以速度获取这段时间将发生的位移,当前时刻的模型输出中加上得到最终打击位置在绝对系下的坐标。此信息就是我们需要的控制信息,将其转化为云台的pitch和yaw,发送给电控。

如果追求击杀效率,则通信、电机动作、弹链空程的延时都不需要考虑。这时我们假设发射机构和目标之间的弹丸速度极快(想象成水枪!),那么只需要我们的枪管一直指向准确的预计位置,就可以保证“水”始终冲刷目标装甲板。若你对命中率的要求较高,则仍需要考虑上面的因素。

实际上,超前时间的计算依赖与云台和目标的距离(由距离计算弹丸飞行时间),然而为了得到距离(超前时间*目标速度),我们又需要知道弹丸的总飞行时间,似乎陷入了死循环。上面介绍的方法只是一种近似,不过好在目标的运动速度不是特别快,误差还算可以接收。关于子弹飞行时间的计算,我们将在6.5介绍。


6.4.3. 跟踪器

在一帧图像中往往会识别到多个目标,我们需要确认这些装甲板在不同帧之间的继承关系以建立运动模型;对于能量机关的圆拟合、旋转函数拟合我们也想选取多片扇叶计算取平均获得更精确的结果,同样需要追踪每片扇叶的状态。因此,对每个目标进行跟踪是非常有必要的。不过,由于在我们的应用场景下,目标数量较少,且识别算法帧率较高,帧间物体在图像上的位移并不会很大,使用一些简单的跟踪策略即可。并且因为我们在每帧都会进行检测,实际上采用的是track by detection的方法。至于5.4中介绍的目标跟踪算法,我们有别的用途。本小节主要介绍跟踪器的构造和目标切换时的策略。

  • 跟踪器的构造

用vector保存目标对象,要保存的信息包括上一/几帧中目标的大小/位置/速度等状态,以及最后一次更新的时间戳等。如对装甲板的跟踪,首先要保证两个装甲板的贴纸相同以确保其来自同一个机器人。为了提高复用性和类的内聚程度,最好包含该目标用于后续解算的所有信息:四个角点的位置、速度和中心的位置。对每一个装甲板,我们都要维护一个不断更新的运动模型。跟踪的策略也非常简单,使用该目标的运动模型先进行状态转移,计算出运动模型的预测值,然后从检测出的所有装甲板中选择与该值最接近的一个,最后使用选中装甲板对该跟踪器维护的运动模型进行量测更新。

对于能量机关而言,图像中的内容更少更干净,一般只需要保存每一个扇叶在上一帧的位置,然后在当前帧中选择距离最小的扇叶更新跟踪器即可。

  • 目标切换策略

如对方机器人开启小陀螺时,一块装甲板消失的时候另一块装甲板会恰好出现;装甲板被击打后会发生闪烁,于 0.1 s 后亮起;又或是目标装甲板被物体短暂遮挡后再次出现。因此跟踪器的更新应该要考虑目标丢失和目标短暂丢失的情况。一般设置一个阈值,当连续数帧都没有出现的时候删除该跟踪器。显然即使没有出现目标我们也应该进行状态转移,注意和以提高控制为目的补帧的异步更新进行区分,前者是因为目标丢失了数帧,后者则是因为控制频率高于采样频率。所以,当目标短暂丢失的时候,也可以当作是识别帧率突然下降,在中间插入了额外的只进行状态转移的模型更新。

对于能量机关,若目标被击中则它将在短暂的延迟后变成激活的扇叶。为了提高模型的收敛速度避免重新初始化跟踪器,可以让该扇叶继承之前的信息。

  • 一些补充

跟踪器还可以和检测ROI结合起来使用。对于哨兵这种运动状态单一的目标,一般只会有一个追踪器(你不可能看到两面的装甲板吧),因此,我们可以加入检测ROI,下一帧图像只在ROI上运行检测算法。传统算法尤为适合这种策略,因为其时间开销是和图像大小正相关的。神经网络其实也可以使用此法,为ROI也训练一个轻量化的网络,以加速识别。实际上其他机器人的检测同样能利用ROI:对于所有拥有相同号码的装甲板,取它们角点在四个方向的最大值作为边界,获取该边界bbox的几何中心并以其为参考点将bbox扩大1.5~2倍,生成最终的ROI。之所以要扩大ROI是为了确保该机器人开启小陀螺后,所有装甲板都能被检测到。对同一个机器人的多个装甲板分别维护跟踪器最主要的目的就是稍后要介绍的反陀螺。



6.5.弹道模型

组委会在ICRA开发平台RoboRTS上提供了一种对空气阻力分析的重力补偿思路:projectile model;务必看完以后再继续阅读本节。

加入空气阻力以后原本的斜抛运动方程转化为微分方程,直接求解微分方程显然比较困难,我们常用迭代法逼近其解。官方的projectile model迭代方案也可以看作是利用离散的比例控制器对竖直方向的弹丸落点进行控制。不过它并没有考虑加入预测后的情形,这里我们主要分析使用预测后应该如何计算pitch补偿值和超前时间T,解决6.4.2提出的问题。

暂时先不管重力补偿,得到目标水平距离和时间的关系(需要确定水平方向初始速度):

令超前总时间为,目标当前和云台的水平距离为,目标水平速度为(注意两者都是矢量,即在两个方向上;轴不需要考虑),则弹丸飞行时间满足下式:
因为除了弹丸飞行时间外的延迟时间可以当作一个常量,因此真正计算的时候式左边只要加上即可,不影响我们的讨论,为方便起见式左边暂时只考虑飞行时间。上式的形式就是一条直线(左)和一个对数函数(右),相信大家高中都做了不少这样的题目,在这里使用迭代法求解,定义目标函数为:
导数为:
选取一个初值,利用梯度下降法迭代至损失小于设定阈值或达到轮数便得到结果。这时候就可以确定超前量,以及总的水平距离了。

现在,根据官方文档建立的模型,我们再考虑重力补偿。由于假定竖直方向不受空气阻力,因此竖直方向的运动方程为:

为击中目标时竖直高度,为当前云台的pitch角,即上一步计算得到的弹丸飞行时间(这时候不需要考虑其他延迟,因为总延时已经确定,我们是在其基础上考虑子弹的飞行)。而目标的实际高度则为,因此误差为:
用文档中的模型迭代上式使误差小于阈值,记录此时的pitch角度;实际上修正的角度一般很小,为1~4°,若你认为已经满足了需求或在实战中表现良好,则可以就此打住。然而实际上显然和初始角度不同,因此水平方向的初速度将发生改变,要想提高精度,应该对第一步和第二步进行反复迭代:我们将新的初速度带入第一个步骤,重新计算,并用新的更新,如此往复直到两个值都趋于稳定收敛,则认为两者都逼近真实值。

实际上,我们利用水平方向的弹道运动方程估计,用竖直方向的落点估计,这是参数相互依赖的两个模型,耦合程度非常高。我们采用的轮流更新参数的方法,可以看作是机器学习中非常经典的期望最大化(EM,expectation maximum)算法,将时间或仰角当作隐变量,首先根据己经给出的观测数据(角度初始化得到),估计出模型参数的值(时间);然后再依据上一步估计出的参数值估计缺失数据的值(在第二步利用计算新的),再根据估计出的缺失数据加上之前己经观测到的数据重新再对参数值进行估计,然后反复迭代,直至最后收敛,迭代结束。如果你了解这个算法,应该能更好地理解我们用于迭代的这两个过程。 注意,外圈有一个大的“迭代“,而对于我们的方法,每一小步也通过迭代法求解微分方程或进行反馈求解。

这里还考虑了竖直方向的运动,一般只要在上坡时可能发生,故在上面的算法中忽略了竖直方向的运动

图源哈工程创梦之翼战队自瞄系统设计

在所有迭代结束后,最后别忘了水平方向上也要进行预测,在秒后云台的目标航向角(Yaw)应该为:

不需要在每轮迭代都更新计算,虽然它的改变会导致云台和目标的水平距离发生变化,但前面在计算的时候已经考虑到水平距离是二维平面上的矢量了。所以只要在迭代结束后用最后一次的进行求取即可。

至此,对于静止目标、哨兵、能量机关,你已经拥有了全部的知识,可以尝试实现完整的程序了。在 6.6 讲完反陀螺之后,我们会再梳理一遍自瞄的框架。

whenever where there is a constrain, here comes optimization. 既然我们可以得到预计的弹丸轨迹,理论上就能计算真实轨迹和预计弹道的差异,从而执行误差的反向传播以优化指定的参数如空气阻力系数等。上海交通大学在2023的青年工程师大会中分享了这个方法。

6.6.反陀螺

前面对目标的建模都是针对平动的,虽然在很短的时间内装甲板绕机器人自身的旋转也可以等效为平动,但是对于舵轮、全向轮步兵等转速高达数r/s的目标,没有考虑旋转的预测模型就无能为力了,预测值常常会飘到机器人外。

机器人发生旋转的时候,由于装甲板转到另一侧导致预测失效超出车体;图源华广机器人RMUA技术报告

6.6.1. 状态识别

假设我方云台静止,对方机器人底盘开启小陀螺处于自旋状态,小陀螺最显著的特征就是连续有装甲板从一侧出现并在另一侧消失,而且每一块装甲板相对相机的位姿都以侧向倾斜-正对-另一侧倾斜的规律变化。因为视野中最多同时有两块相同ID的装甲板出现,跟踪器的数量应该会呈现...-2-1-2-1-...的规律变化,并且变化的频率保持固定。用计数器统计上述两种情况的出现次数,如果达到阈值则认为目标处于陀螺状态。另外,根据机器人制作手册的限定,两两相对的装甲板必须安装在对称位置,因此只需要识别到任意两块相邻的装甲板,实际上就可以估计出剩下两个装甲板的位置。并且此模型一旦建立起来,我们将能够用跟踪器对所有装甲板保持追踪。

一般建立一个专门的模块用于检测陀螺状态,该模块存储相同ID的装甲板信息和跟踪器的数量,以及其数量变化的周期或时间戳信息,还有一个统计陀螺状态的计数器。

6.6.2. 运动学建模

我们在识别和目标的运动建模阶段可以为每一个装甲板都维护一个跟踪器,因此各自的模型更新可以利用6.4.1的方法独立完成。这里所说的建模是指对小陀螺、四块装甲板或整个机器人底盘的建模。当识别到底盘处于小陀螺状态,根据同一个机器人上两块装甲板的速度方向和大小,可以计算旋转的中心的旋转角速度:旋转中心为垂直于两块装甲板的直线的交点;角速度为其速度除以旋转半径。

每个装甲板的位姿都是可以通过PnP计算出来的,获取上面的数据也不是什么困难的事情,只是一些简单的位姿变换。若认为用平移近似转动的误差太大,则考虑扩大量测范围,将新装甲板出现的那一帧和装甲板消失前的最后一帧中的装甲板法向向量记录下来,由两个向量的叉乘计算转过的角度,除以时间便可以算出角速度。

如果已经确认进入反陀螺状态,在装甲板转到背面后,我们可以选择不删除它对应的跟踪器,而是采取新的策略,利用它对面/相邻的装甲板的状态信息继续对其进行更新,这样当它从另一侧再次进入视野时仍然可以快速获取它的状态而不用重新初始化。

利用重投影,还可以将转到背面的装甲板的角点投影到像平面上进行可视化,以评估位姿估计的准确性:

四块装甲板识别及底盘建模的可视化,蓝色为被遮挡的,绿色为可检测到的

6.6.3. 击打策略

如果如上图所示建立了底盘的运动模型,则可以则任意一块装甲板在何时会出现都可以通过模型获知,可以考虑将枪口对准对方机器人的中心,当任意一块装甲板那的预测极大位置包含中心时开火;或者跟随预测一起运动,保证预测量不超出机身范围,一旦超出就切换瞄准下一块装甲板。对于第二种方法也未必需要计算底盘旋转的角速度,直接用经验法/数据拟合法建立装甲板速度和出现在正面的时间的相关关系也是一种不错的方法。

若没有建立底盘模型,只是为每个能看到的装甲板维护跟踪器,可以保存最后一帧装甲板消失的位置和新装甲板出现的位置,取两者的中点作为打击点,采用和前者相同的策略(也可以不自动开火,就对准中央让操作手决定开火时机,减少云台晃动以提升操作体验);或始终瞄准在旋转方向上落后的(也就是第二块)装甲板,因为它的“存活时间”比较长,预测的有效时间相应的会更长。一旦新的装甲板出现,立刻转向瞄准它,此时还可以将老装甲板的信息继承给新出现的装甲板以加速收敛,不过此法只对静止的陀螺有较好的效果,否则因为速度耦合,击打运动陀螺的继承关系很难确定,且有速度误差。

6.6.4. 观测器示例

这里以rm-vision中提出的匀速转动+平动的运动学模型作为示例,分析机器人运动状态观测器的构建。

P.S. 这里介绍的方法和rm-vision原始版本不一致。rm-vision的建模只考虑持续跟踪当前最新的装甲板,而没有考虑两块装甲板共现的情况。通过对系统能观性的分析,可以得出这样的观测方程会导致底盘半径不可观。(通过matlab的obsv()可以很方便地计算能观性,然后通过obsvf()进行能观性分解以得到不能观的部分)

但是如果你采用守株待兔的策略,预测每一块装甲板转到我方机器人和对方底盘中心连线位置时被击中,那么仍然是一种可行的策略。实际上即使使用追踪击打的方法大部分时候也能奏效,因为半径不收敛到准确值带来的影响较小(只在水平方向上相差较小的距离,基本在一块装甲板之内)。

首先,规则手册规定机器人的装甲板安装必须成轴对称设计且对侧安装的装甲板两两之间的连线相互垂直。那么,机器人底盘的运动可以分解为绕中心的转动和底盘刚体的平动(这里暂时不考虑平衡步兵机器人、偏心小陀螺和机器人上坡[ 此处特指倾斜 ]的情况)。因此,为了获取机器人的运动,我们需要获得几个信息:底盘中心的位置和速度,以及底盘自旋的速度;还有底盘中心到两组装甲板中的各一块的距离(上图中的相同颜色的装甲板到中心的两个距离);由于机器人两组相对安装的装甲板高度很可能不同(设计者很可能希望通过高低装甲板的错落来降低那些“泼水”方式射击自己的机器人的命中率),我们还需要估计较高的一组装甲板距离较低一组装甲板的距离(在实践中也可以直接估计),注意此处将底盘的中心高度和较低的一组装甲板高度视作相同。

统计一下,这里一共有9个状态量,他们都是无法直接得到的,在状态空间建模中可以观测得到的数据为(实际上这里的并不是线性的)。在识别到装甲板之后,看看我们能得到什么有用的观测,从而得到和之间的关系以计算量测方程的雅可比矩阵。当视野中出现一辆车的单块装甲板时,我们可以获取这块装甲板在绝对系(即转换到车体系)下的位置和姿态,一共六个值。当出现同一辆车上的两块装甲板时,则可以获得两块装甲板的位姿,同时,根据规则的限定,两对相对安装的装甲板各自的连线相对另一对一定是垂直的且呈轴对称(可以视为“先验”),这时便可以根据简单的几何关系推算底盘中心的位置以及。

知道一块装甲板的水平朝向(yaw,下面用 替代)之后,剩下四块装甲板的方向就可以被计算出来(相邻的两块只能计算朝向不能计算位置)。不过,如果观测独立,观测方程的数量肯定是多多益善——在机器人底盘转动的过程中至多有2块装甲板同时面向相机,而这两块装甲板对应的半径又恰好不同,如果将两对装甲板中各取出一个,将它们的朝向yaw也加入状态估计量中,它们对侧装甲板的位姿便能够推出,这样就可以利用匀速模型预测任意一块装甲板在一段时间之后的位姿了。

当然,为了确定装甲板属于哪一对、选用正确的量测方程,我们需要维护跟踪器以处理装甲板的消失和出现。

对机器人的建模,注意两对装甲板的高度一般不同

现在,状态转移方程已经确定(匀速模型),我们只需要根据单块装甲板、两块装甲板出现在视野内的这两种情况,分别列写它们的量测方程即可。

  • 状态转移

最终确定的状态向量为。状态转移方程为:

空白处皆为0。

  • 量测方程(单块装甲)

对于单块装甲板的观测,不失一般性这里假设观测到了第一块装甲板:

底盘中心可以根当前观测到的装甲板位置和朝向计算出来:
上述存在三角函数的非线性环节,因此需要求量测方程的雅可比,将上面的方程写成以状态为变量观测为结果的函数,然后求导即可。

我们记第一块装甲板的中心高度为底盘中心高度为,另一对装甲板的中心高度为

写成矩阵形式:

  • 量测方程(两块装甲)

直接上图:

这个瞬间,armor1在armor2左侧;有时候位置相反,这需要你在程序中自己处理

即通过两块装甲板和各自的半径都可以计算得到中心点的位置。对于两对装甲板的错落,直接通过计算(符号取决将哪一块装甲板视为在下方,这里我们假设r1这一对装甲板更低。若实际更高,则最后计算得到的结果为负数,但不影响)。

这时候的观测是8维的。然而,在将观测写作观测方程的时候却会出现问题:当编写和对应的观测方程时,我们发现上图中的两式都可以推出结果——这实际上对应着两个观测数据来源。当一个状态存在多个观测方程时应该如何进行后验融合?

一种方法是进行两次量测更新,由于高斯分布的特性,使用最大后验的方法融合先后2个高斯分布,其结果仍然是高斯分布,且和融合的顺序没有关系(你可以自己验证一下)。

另一种方法则是先对不同的量测方程进行融合,因为我们假定量测值服从高斯分布,自然而然两个高斯分布可以融合得到一个新的高斯分布,再把这个“新的”量测加入与状态先验的融合中。

若从概率视角入手,这里求解的是,假设测量z1、z2以及状态先验相互独立,那么就有,前两个就是观测方程,第三项是状态方程 。

这里为了方便起见,以上的两种方法都不会采用(你可以自己试着推导,实际上并不困难),而是直接计算,将它们作为直接能够获得的观测量。我使用matlab符号系统化简得到r1和r2的表达式:

像 6.4.1 介绍的单块装甲板匀速直线运动的模型一样,实际上通过单目PnP深度估计得到的平移向量之间也并非独立解耦的,而是存在较为复杂的(依赖于PnP的约束)关系。在这里,半径和装甲板的位置和朝向也存在着复杂的变换(从而噪声分布也复杂),但我们简化模型,将半径视为服从高斯分布。

最后,将观测扩充到10维:,两块装甲板和底盘中心的关系与单块装甲板观测的形式相同,只是新增了半径的观测。最后列写矩阵的形式:

为了保证横向宽度不超出范围,c=cos,s=sin

  • 跟踪器维护与装甲板切换

得到的状态方程和两种不同情况的观测方程,我们只需要在每一帧解算完成后利用卡尔曼滤波的“黄金五式”进行融合即可。现在只差最后一个问题没有解决,如何判断当前使用哪一个量测方程?视野内的装甲板是armor1还是armor2?

很直觉的想法是,在追踪器尚未初始化时,将出现在视野内的第一个装甲板视为armor1。若初始化时视野内有两块装甲板,则可以人为约定在左侧的为armor1。跟踪则使用经典的tracking by detection的方法,将上一帧位置和这一阵出现位置最近的装甲板视为相同的目标。当装甲板离开视野,记录标号(1or2)并删除跟踪器,同时赋予下一个出现的装甲板的标号为之前记录的那个。

当然,你还需要解决一些工程上的细节,以处理短暂遮挡、装甲板受击闪烁以及观测值的异常检测等情况。



6.7.自瞄算法框架

现在我们可以把 6.1~6.6 学习的所有知识连接在一起,创建自瞄算法的框架了。这里我们对前述每个小节的功能都进行封装和抽象,不再看其中的小步骤,而是以功能模块的形式进行连接。我们先只考虑算法处理顺序。

  • 装甲板识别流程

deprecated,哨兵已经不再被固定在轨道上运行。

  • 能量机关识别流程
  • 线程设计、互斥锁和下位机通信

如果算法的总耗时小于相机的采集时间间隔,那么只需要为线程之间的通信设计一个容量为1的共享数据队列。由于IMU和下位机发送的频率远高于相机采集和算法处理,因此一般为它们预留大小至少为相机图像buffer 5~10倍的数据队列,以方便进行时间戳对齐。如果算法时间开销大于相机采集速度,应该增大共享数据队列的容量保证其能够进行同时读写,用流水线处理掩盖算法多出来的latency。也可以考虑使用线程池,一旦有新的任务进入(即新图像到来),立刻取出一个线程发射新任务。

这里选择使用三个buffer,实际上如果算法处理速度较慢,可以在算法的内部子环节中插入额外的FIFO buffer以掩盖延迟的时间开销。从下位机处的数据接收可以采用轮询或者中断的方式,根据需要自行选择。

对于线程的数据交互,FIFO需要一个标志位确定内部数据的时间顺序,消费者线程始终读取FIFO的头部,生产者线程不断向尾部填充,buffer size大于1的时候不会出现读写冲突。buffer size等于1的时候需要为数据添加互斥锁,为了提高效率防止阻塞,应该在消费者拿走数据后立刻解锁,让生产者能够将数据放入buffer。

图像处理的线程可以根据电控发来数据的工作模式确定当前的任务是激活能量机关还是装甲板识别。相机采集、串口线程为了保证稳定工作,可以添加守护线程以防止连接线松动、工作异常、串口号变动等问题,及时重启线程。主函数在启动其他线程后进入等待即可,也可以把四个线程中的一个放到主线程中,减小线程的创建数量。

别忘了给你程序加上一个自启动脚本。当然,你需要在代码中添加重启串口、相机掉线重连的支持。要使用自启动,在bios里开启power strategy after s5设为s0(即上电开机),下面是一份包含了守护线程的脚本:

#!/bin/bash 
# execute using bash
# 特别注意特别注意特别注意特别注意特别注意特别注意特别注意特别注意特别注意特别注意特别注意特别注意:请将/bin/sh链接到/bin/bash,ubuntu18.04后的默认shell为dash,dash无法source动态加载环境变量!

# 检查间隔时间,每隔sec秒查看一次当前线程是否正常运行
sec=1
# 异常统计数,超过5次重新make,超过20次reboot
remakecnt=0
rebootcnt=0
#进程名字,取决于CMake的项目名称,即可执行文件的名称,也可以在启动程序后在终端输入top/ps来查找
PROC_NAME=HNU

# init openvino environment, you can add your own env variable here
source /opt/intel/openvino_2021/bin/setupvars.sh
# remember to add these if your programme depends on other dependencies
source /etc/profile
source ~/.bashrc
# be sure that vars in profile will be loaded
# source /etc/environment
# source ...

# 进入编译文件里面
cd /home/rm/hnuyue-lu-rm2022-vision/build # 路径改成你自己的项目build文件夹
# 清除并重新make一下,防止文件损坏(-j提高效率,根据核心数选择,一般2倍) 如果有必要你甚至可以重新生成makefile,cmake一下
make clean && make -j12 
# 赋予可执行文件读写权限,通过管道将密码传递给sudo -S(-S参数接收密码),HNU是我们生成的可执行文件名
echo " " | sudo -S chmod 777 ./HNU
# 赋予串口权限,若有多个一次性赋予(*),建议直接将当前用户加入串口所有者的用户组中,就不需要这一步了
echo " " | sudo -S chmod 777 /dev/ttyUSB* 
# 新建一个terminal在里面运行程序,方便监视程序的情况,若要ssh链接则直接运行,替换为./YueLuRMVision2022
gnome-terminal -- ./HNU
echo "[daemon]: ${PROC_NAME} has started!"

while [ 1 ] # 中括号的判断条件两边要空格
do
# 判定进程运行否,是则继续,否则重新启动
# ps列出进程,-ef给出进程的详细信息,grep捕获我们的进程,
# grep -v把grep进程去掉,否则grep也会把自己统计进来,wc(word count) -l(line)统计进程数
# 当程序正常运行的时候,tcount>=1(多线程)
tcount=`ps -ef | grep ${PROC_NAME} | grep -v "grep" | wc -l` 
echo "[daemon]:Thread count: ${tcount}"
if [ ${tcount} -ge 1 ];then  # -ge greater equal,大于等于1情况下 进程没被杀害
	echo "[daemon]: ${PROC_NAME} is running..."
	sleep ${sec} 
else  # 出现异常,进程未运行
	# 异常统计数累加
	rebootcnt=`expr ${rebootcnt} + 1`
	remakecnt=`expr ${remakecnt} + 1`
	if [ ${remakecnt} -gt 4 ];then # greater,5 times failed: try to remake the project
		# echo -e "\033[31m你要输入的字符串\033[0m",可以换颜色,xxm是颜色代码,还可以换背景
		echo -e "\033[31m[daemon]:trying to remake project...\033[0m"
		cd /home/rm/hnuyue-lu-rm2022-vision/build  #移动到你要编译运行的文件夹里面
		make clean && make -j12  #清除并重新make一下
		echo -e "\033[31m[daemon]:Remake successfully! \033[0m"
		remakecnt=$[ 0 ]
	fi
	
	if [ ${rebootcnt} -gt 20 ];then # 大于20,直接重开!
		echo " " | sudo -S reboot
	fi
	
	# 小于五次,尝试重新运行程序
	echo -e "\033[31m[daemon]:${PROC_NAME} error! \033[0m" 
	echo -e "\033[31m[daemon]:Trying to restart ${PROC_NAME}...\033[0m" 
	echo -e "\033[32m[daemon]:retry counts: ${rebootcnt} \033[0m"
	echo " " | sudo -S chmod 777 /home/rm/hnuyue-lu-rm2022-vision/build/HNU
	cd /home/rm/hnuyue-lu-rm2022-vision/build
	gnome-terminal -- ./HNU
	echo -e "\033[34m [daemon]:${PROC_NAME} restart successfully! \033[0m"
	sleep ${sec}
fi  
done


7.视觉组成员需要掌握的知识

在前文当中零散地介绍了一些需要学习的知识。在这个部分,会系统性的列出一张视觉组知识框架供大家参考。当然这只是笔者个人目前接触到的东西,限于个人水平必然有很多欠缺,望读者提供补充建议。

下图是一个典型的完整自瞄算法的流程,下列相关的知识都可以从中找到应用场景。当你遇到一个新的工程问题,也可以尝试将技术分类列点,形成一张这样的流程框架图:

建议打开大图查看,Image_base文件夹下有矢量格式图片,可以更清晰的看到内部的小图像

如果你对于这张图中的细节都没有疑虑,能够联系各个部分对应的算法和实际应用过程,说明你已经大致掌握了作为视觉组成员所需的知识体系了。

7.1.硬件相关

工欲善其事必先利其器,了解底层原理和封装的细节有助于软件算法的设计与优化。地基不牢就妄想万丈高楼拔地而起是万万不可的。

7.1.1. 计算机组成原理

  • 硬件组成

了解现代计算机的基本架构:冯诺依曼和哈佛架构。熟悉CPU的构成,包括ALU(理解常见的运算器如加法器乘法器除法器)、高速缓存、通用寄存器组和指令控制器,指令流水线等,知道CPU如何调度其他的硬件还有硬件中断的实现。了解内存的工作原理、寻址的概念以及计算存储的层次结构;内存管理以及cache-内存映射、内存分页等也可以作了解。掌握总线结构的原理和构成,学习DMA、PCI、USB等计算机内部常用的连接总线。了解GPU和CPU的不同,以及GPU是如何在CPU的调度下进行图形运算和渲染的。熟悉常用的I/O设备以及计算机和I/O设备的交互方式和驱动的概念。

  • 软件组成

了解指令集和机器语言-汇编语言-编译型语言的关联,以及解释型语言和编译型语言的异同。明白操作系统和实时系统的异同,知道内核的概念和内核调用,以及如何实现任务并发、硬件资源共享、硬件资源抽象和虚拟化和异步执行这四个操作系统的基本功能。操作系统还要提供对处理器的调度和管理,包括:进程控制,进程同步,进程通信,作业进程调度;存储器的管理:内存分配,内存保护,地址映射,内存扩充;外设管理:缓冲管理,设备分配,设备处理;文件管理:存储管理,目录管理,读写管理和保护;以及最后的接口设计:系统调用、C/C++ Runtime,命令接口和GUI。


7.1.2. 微处理器结构

这部分主要是关于片上系统,也就是我们所谓的单片机,即电控使用的控制核心MCU。除了一般不搭载操作系统,单片机几乎和现代计算机一模一样。以比赛中常用的STM32F4系列为例,重点了解各种外设的寄存器配置,学习标准库、HAL库对底层资源的抽象实现,时钟树和时钟管理,还有包括GPIO、SPI、CAN、USART、I2C、TIM、ADC、DMA、EXTI、WDG在内的常见外设。了解软件中断和实时系统的原理,尤其是FreeRTOS。

视觉必须的硬件,包括相机等已经在第四部分进行了详细地介绍。



7.2.软件相关

  • Linux系统

可以了解一下UNIX、LINUX、Windows三大家系统的渊源。一定要掌握Linux终端的基本使用和简单的shell编程。理解环境变量的意义和各种配置文件的作用。知道Linux的文件系统构成和修改读写控制权限的方法。会使用Linux下常见的小工具如ping、ifconifg、wget、tar、ssh、ftp、fd、htop等。学习Linux下软件包的管理。学会自己将驱动写入内核。

  • GNU工具链

了解编译型语言从源码到执行的过程。会使用G++编译带有少许动态链接库的程序。能在无GUI的界面下使用GDB进行简单的断点插入、查看变量等操作。

  • 网络和通信

了解五层网络协议栈以及DNS、DHCP、URL等的基本概念。线程通信和进程通信的常用手段也需要掌握。学会自己定制定长、边长的frame-based和stream-based协议。

关于低速外设的通信已经在硬件部分介绍。



7.3.语言相关

  • C/C++

至少掌握C语言,并且能够将C++用作C with class的程度。知道C++的内存模型(最好还能学习一下MMU,虚拟内存和地址映射的概念),了解指针和引用的区别。学习一下C++的各种编程范式,包括泛型、面向对象、面向过程和函数式编程。对标准库(STL)的构成有基本的了解,能够熟练运用各种容器和算法。在合适的地方使用模板和继承,对类方法和成员进行合理的封装。

  • Python

了解脚本语言和编译型语言的区别,大致明白动态类型的实现原理,知道Python对内存的管理方法。熟练使用切片、推导式和各种数据结构,以及迭代器/生成器的使用。特别注意python的I/O和文件操作,因为我们在大部分时候使用python编写解析文本的脚本,如xml、txt、json等。对于python的位置参数、关键字参数和命名关键字参数也要有所了解。如果有兴趣,还可以学习闭包和闭包、map/reduce/zip/filter等高级函数,以及装饰器。

  • 正则表达式

编写python/shell脚本,索引文件、匹配模式的时候用。掌握基本操作即可,不要弄太复杂的。



7.4.计算机视觉相关

第五部分都是介绍相关算法,这里不再赘述,只是做一个小小的总结。

图像处理不管在什么时候都一定是计算机视觉的基础。而2D目标检测则是踏入深度学习时代的第一步,和比赛相关的还包括关键点检测。3D视觉我们暂时只需要学习坐标变换、位姿解算和简单的点云处理,若要研发哨兵机器人,SLAM相关的状态估计内容也是必须的,因此还要学会3D的位姿描述以及李群上的优化工具的使用,比如g2o和gtsam。本教程并不包含任何必须的数学知识和机器学习相关的内容(虽然已经涉及了很多ML的概念),这部分是需要读者自行补充的。OpenCV是计算机视觉的大宝库,也是经历了时间检验的值得使用值得学习的开源库,一定要好好利用。在当下这个优化的时代,Eigen、Ceres也是很好的帮手。注意,学完本教程你只能算是刚刚入门。

更多的内容都属于探索和深造方向的,包括分割、3D重建(爆火的NeRF)、看图说话、动作检测、姿态检测、视频理解、多模态、给句作画、对抗学习、超分辨率、图像降噪、深度估计、知识整理、神经网络架构搜索、大规模预训练(包括自监督和对比学习)、LLM、图像增强、风格迁移、图像检索......



7.5.机械、硬件和电控

为了和其他技术组的同学更好地协作,有必要了解相关的基础知识。

7.5.1. 伺服驱动单元

主要是比赛常用的电机。

  1. DJI出品的各种智能电机,它们都是以定子直径-定子厚度命名的,方便安装。如3508代表35mm的定子直径和8mm的定子厚度。它们都可以通过CAN/PWM进行控制。 - 3508电机:拥有1:19的齿轮减速箱,持续输出扭矩2.8N·m,适合作为轮组电机或其他速度伺服/开关类型的机构。配合C620 FOC电调使用。自带电流传感器和霍尔磁编码器,对磁编码器积分后可以实现三闭环控制。拆卸减速箱后还可以作为摩擦轮电机。 - 6020电机:大扭矩直驱电机,内置FOC电调,持续输出扭矩1.2N·m,并且带有一个直径18mm的通孔方便安装电滑环/弹链。由于转速低,不带有电流传感器,有霍尔磁编码器(可以检测速度和位置)。由于能够进行位置伺服,一般用作云台电机。 - 2006电机:加入1:36齿轮减速箱,体积较3508小很多,持续峰值扭矩1N·m,配合C610电调使用。一般作为拨盘电机,或者其他速度伺服/开关类型的机构。同样有电流传感器和霍尔磁编码器。 - snail电机:超高转速的航模直流无刷电机,需配合C615电调使用。由于不自带速度编码器,需要自行改装以实现速度伺服。一般作为摩擦轮电机使用,现在大批量被3508替代。很可能被淘汰。 - 退役电机:包括6623、3510、RM35、6025、EC60等,由于性能已经过时且体积重量不合适,已经基本消失在赛场上。

FOC的介绍戳这里,高中物理知识就能看懂:【自制FOC驱动器】深入浅出讲解FOC算法与SVPWM技术 - 知乎 (zhihu.com)

  1. 包括海泰、Ktech、宇树、脉塔、高擎、达妙等出品的机器人、云台电机。包括大减速比的机器人关节电机,和一些中低速的多绕组大直径直驱电机。这些电机可以作为云台电机以提供更强劲的性能,或是作为平衡步兵的驱动电机/轮腿关节电机,比如瓴控9025,海泰04等;

一些队伍还会自制电机减速器(减速箱),通过匹配合适的减速比以获取极致的性能,或改进减速器结构以提高可靠性。

  1. 电推杆、步进电机和舵机,一般用在对响应速度要求不高的位置伺服场景,如机器人的抬升机构、夹爪、弹舱、障碍块夹爪等。可以提供较大的力矩/扭矩。

  2. 气动执行单元,包括气缸、气动转台、气阀门、真空泵。一般用在开/关类型的机构上,可以实现非常快速的响应,常见于抬升、夹取机构等,也有用于英雄机器人的发射。


7.5.2. 电力电子硬件模块

主要是硬件组制作的开发板、用于功率控制、升降压以及调试使用的模块。

  1. 超级电容:由于规则对机器人的底盘功率有限制,我们可以制作超级电容将功率暂时转移,在低功率移动的时候可以把额外的能量存入超级电容,以便在高速运动时使用。一般根据原理分为Buck、Buck-Boost和数字电源超级电容。
  2. 功率测量模块:用于限制机器人的底盘功率防止超限。一般配合功率估计算法使用,将电机的功率模型作为量测来源之一,使用功率测量模块和裁判系统的功率测量修正功率模型的误差(这可以看作一种observer,试试kalman filter!)
  3. 主控单元:就是单片机,有些队伍为了方便接线/实现定制功能会自己制作PCB并焊接。所谓的single-chipped microcontroller和一般的电脑区别在于,单片机的cpu、ram、rom和其他总线控制器以及基本外设都集成在单块晶片上。apple silicon打造的M系列芯片已经在一定程度上向单片机靠近,可以将单片机看作是把acpi,pcie、SATA等总线以及其上连接的芯片和设备都缩小并集成在芯片内部的电脑。
  4. 继电器:电子开关,一般通过mos管等电力电子器件实现,开关信号的通断会决定其之路的通断与否,用电替代了机械开关,方便控制。英雄机器人由于规则限制必须使用继电器。一般我们也会在电源上增加一个可以远程控制的继电器,方便在机器人出现问题、疯转等的时候能够立刻停止。也有使用机械式的电控继电器(磁簧管等)
  5. 荧光充能模块:安装在小弹丸供弹链路上的紫外线发射模块,让小弹丸内的荧光材料受紫外激发跃迁以发出可见光。
  6. USB转TTL(串口)/CAN:在第四部分已经介绍过,用于和下位机通信或直接控制电机。
  7. 调试器(下载器):用于连接单片机和电脑,提供异构平台的调试支持。包括stlink、jlink、daplink、cmsis-dap、ulink等。

7.5.3. 控制理论和姿态估计

对于7.5.3 和 7.5.4,如果你是机械自动化/控制或其他EE专业出身,这些介绍似乎没有必要;如果你没有相关背景,笔者写了也是白写。所以我建议你看一看,图一乐。

这里的控制主要是伺服电机的控制,其他开/关类型的执行单元只有两个挡位(bang-bang)。最常用的算法是PID,即比例-积分-微分算法,这也是在目前的工业控制中用的最多的方法(约70~80%),显然这是最简单的控制方法。有时候,还会加入前馈控制(forward),和PID构成PIDF。这里有一个对PID算法的直观解释:PID控制算法原理(实际上并不本质)。PID算法还有一些唾手可得的改进,参见这个系列:PID库与PID基本优化。当你可以对系统进行较为准确的建模时,可以使用model-based PID,直接配置模型的极点使其以接近临界阻尼的方式收敛到期望输入位置,此时的PID则用作反馈回路,以补偿建立的模型与实际情况的差异。实际上,状态空间建模中使用线性反馈控制器,就相当于为每个状态建立了一个model-based PD controller,若引入状态误差积分项,则一般被称为内膜控制。

基于状态空间法的现代控制理论,也可以利用一些线性方法进行控制。LQR(linear quadratic regulator)是使用的最广泛的线性调节器(调节器的目标和控制器不同——让所有的状态变为0),通过线性状态反馈将控制问题转化为二次型的优化问题进行求解,它可以解决一大类的控制问题(包括非线性问题,在稳定点附近线性化即可)。MPC(model predicate control)是一种动态LQR,它通过一个滚动窗口实时更新并预测控制对象当前和未来一小段时间的行为,并将其转化为优化问题求解。包括非线性自适应、和滑膜控制/高增益控制等鲁棒方法对部分未知的模型也有很好的效果。如果你是EE相关的背景,可以跟着这两个系列一起学习:现代控制理论、非线性控制。如果你来自其他major,今后不打算从事相关领域工作也不想学习,那么刚好上面这两段简短的介绍就只能让你了解一些概念,以便在电控/机械的同学进行讨论时也可以插一嘴。当然,你最好掌握一些云台伺服控制的知识,否则在电控甩锅给你的时候你只能默默接锅。

IMU也是状态估计有关的学科或应用的核心之一,比如INS,SLAM和SFM。我们已经在4.3大致讲解了MEMS IMU的原理和一些数据处理方法,这里提一下具体的使用。在大部分时候我们会选择直接利用C型开发板上面的那颗BMI088,将C板固定在云台上和相机一同旋转。有时候为了更好地进行手眼标定(即IMU和相机的联合标定)以及两者的硬件触发采集,也会选择自制一个IMU将其与相机安装在一起,此IMU通过串口或CAN和运算平台进行通信。

学习控制理论,可以关注笔者的专栏:希尔伯特空间视角下的LTI - 知乎 (zhihu.com)


7.5.4. 悬挂系统

如果你了解非独立悬挂,可以对本节进行补充。

这是笔者认为最核心的部分之一。当前RM有两种流行的悬挂,一种是每个轮组单独配备一套悬挂,称为独立悬挂;另一种是不同轮组间的悬挂会发生相互关系以约束自由度,被称为自适应悬挂(笔者也有点没搞懂为什么叫自适应,可能是上坡的时候通过性能更好)。悬挂系统一般由阻尼器和弹性储能装置组成,是一个典型的二阶系统。此系统的输入是悬挂形变量,目标可分为跟随形变(即恢复到原来的状态)或保持高度不变。该系统有几个参数特别重要,罗列如下:

  • 截止频率:允许截止频率以下的震动通过,滤除高频振动。此值和稳定时间以及共振频率有很大关系,要在高低之间取得权衡。
  • 稳定时间:受到冲击输入/阶跃输入后,系统进入稳态所需要的时间。一般来说稳定时间应该在长和短之间取得权衡。
  • 共振频率:输入频率和该频率相同的时候,系统会引发幅度最大的震荡。最好让共振频率高于截至频率,避免产生大幅共振。
  • 最大振幅:对于某个频率的输入,引起的最大超调/最大响应值。此值越小说明系统对外界输入的衰减越强,一般阻尼大的系统衰减速度快,此值越小越好。

利用二阶系统的方法对悬挂进行分析,根据需求可以设计合适的阻尼系数和弹簧的劲度系数,以确保其性能满足需求(对独立悬挂而言)。由于二阶系统需要在截止频率和响应时间上取得权衡,不能做到美美与共,更先进的设计方法是采用分级悬挂,在不同的行程上采用不同的参数使得悬挂在所有位置都能取得令人满意的性能。最高境界则是主动悬挂,轮腿式的设计(尤其是平衡步兵)自然地将机械悬挂替代,当然也可以选择保留机械悬挂以降低电机的负荷。使用主动悬挂则可以对任意频率、任意幅值的输入进行针对性设计并实时更新参数,合理的算法设计甚至能够使其达到如履平地的效果。

将单个轮组建模,得到的二阶系统,M为近似为四分之一车身重量


7.6 知识体系和学习路线

笔者整理了一张计算机视觉学习的思维导图,和一张成为一名合格的初级机器人工程师需要了解的知识的思维导图,链接戳这里:

第七章的开头也给出了一张关于自瞄实现的pipeline,以它为脉络学习各个部分的知识也是可以的。

7.6.0. 遇到问题怎么办

学习一定要循序渐进,如果某些知识要求一些前置知识,请务必把基础夯实再开始学习前者,否则很容易不知其所云。另外,一定要成体系地进行学习,尽量一次性建立该学科或该领域的主干,不要总是四处寻找碎片化的知识。当你遇到一个难以解决的bug或者异常,请先问问自己,我是否有相关领域的基础知识,如果有了再尝试查找资料解决;如果没有,请立刻着手学习,否则你可能根本无法解决这个问题,或是即使查到了也不理解它到底是如何帮助你修复这个bug的,只是按部就班地根据其解决步骤机械地复制罢了。对于一些一直无法理解的公式推导或者知识,同理。

另外,在上述遇到问题的情况下,你自己可能无法解决(但是也请仔细看一看报错信息,这并不困难,而且很多时候其实很快就能定位问题所在,不要看到一片红色的error就不知所措),这时候就要利用搜索引擎或者向它人求助。一般来说,技术类的问题,Google可以得到比较好的结果,bing次之,最好不要不用baidu。搜索的时候,请使用陈述句而不要加入语气词/问号/疑问词等。如“C++ vector的初始化”就比“怎么样初始化一个vector”好得多。善用搜索引擎的规则搜索,大部分搜索引擎都支持类似正则表达式的功能,可以限定搜索内容,请参考:搜索引擎实用语法。请有意识的在相关的平台寻找对应的知识,在该软件平台的官方网站/说明文档中寻找解答是一个很好的方法。对于编程问题,StackOverflow是比较合适的选择,系统出现的问题如Ubuntu就去askubuntu。谨慎使用CSDN上的解决方法,很多完全不奏效甚至会造成意想不到的后果,并且质量良莠不齐,存在大规模的复制黏贴。How-To-Ask-Questions-The-Smart-Way详细地介绍了提问的艺术,良好的问题描述有助于它人了解你的情况。

广为流传的“提问收费表”,表明了问题描述的重要性;这也是视觉交流群“救命啊”的梗来源

下面笔者将会分门别类,大致按照学习的时间顺序和知识的依赖关系,把所有精心挑选的资源和书籍一一罗列。已经在正文部分出现的相关算法学习教程和书籍资料等均已在出处给出链接,这里不再重复罗列。

如果你有相关方面的课程和教材推荐,欢迎补充。若认为还需要增加方向,也欢迎提出建议。

7.6.1. 数学

本科期间学习的微积分、线性代数和概率论,以及离散数学(有些专业/学校会开设)是必须的。数学是一切一切的基础。在课内学习之外,应该自行探索,着重理解关键的概念。下面的介绍大概按照学习的顺序排列。

  • 微积分

分为微积分基础和多元微积分,一般在大一上和大一下分别开设。这是对工科生来说高等数学中最核心的部分。

  • 教材推荐:《普林斯顿微积分读本》,相对国内的定义式教材有着无与伦比的优势,直观性会强很多,笔者认为这点非常重要,要先有直观的认识然后再加入严谨的定义。
  • 课程推荐:麻省理工-微积分重点,涵盖了微积分中最重要的概念,可以用作预习/复习。高等数学-国防科技大学由朱建民教授讲授,个人认为是国内上的最好的慕课,不仅贴合应试内容还注重实用,可以保证你在考试也拿高分,一共有五个系列,分别对应高数上下册。它也贴心的提供了先修课微积分CAP。
  • 重要概念:映射、极限、导数和微分、泰勒公式、定积分和不定积分、向量值函数和多元函数、偏导数和全微分、条件极值(拉格朗日和KKT非常重要之后会遇到n多次)、多重积分、坐标变换(结合线性代数,jacobian)、梯度/散度/旋度、格林公式/高斯公式/斯托克斯公式、傅里叶级数。

  • 线性代数

是机器学习、矩阵分析、凸优化和积分变换等后续课程的基础。个人认为是计算机视觉最核心的数理基础课。

有人说,程序员一生要学五遍线性代数。

  • 教材推荐:《linear algebra done right》,抛开行列式,直接从最本质的角度分析线性变换,由线性映射的概念导出矩阵、特征向量、相似等的概念,同时涉及了一些简单的泛函分析。《Introduction to Linear Algebra》,结合了大量工科应用实例,并且深入分析了线性空间(丢掉你的同济绿皮吧!)。
  • 课程推荐:MIT-18.06——永远的神!配合《Introduction to Linear Algebra》食用。还有广受欢迎的3blue1brown的线性代数的本质,这个系列课程通过大量直观有趣的可视化,给你提供直觉的理解。这里还有一个精心制作的可视化线性代数课程网站,包含了大量可以自行操作和交互的图像和模型,戳这里:Immersive linear algebra。18.06还有后续课程:线性代数在数据分析、信号处理、机器学习中的应用。
  • 重要概念:线性空间、列空间/零空间/行空间/左零空间、线性映射和线性变换、投影和正交矩阵、特征值/特征向量和对角化、马尔可夫矩阵和矩阵幂、线性微分方程组的求解、相似矩阵和二次型、奇异值分解。不仅要学习概念会做题,还要有抽象和直观的理解!
  • 3blue1brown关于线性代数的视频可以给你一些直观的理解。知乎上也有很多关于线性代数究竟在学什么、所谓“线性”和“代数”究竟是什么,以及线性代数的“本质”相关的优质回答。

  • 概率论

机器学习和深度学习的根基,进阶课程包括随机过程、数理统计等。课内知识大概已经足够。

  • 教材推荐:《概率导论》、《普林斯顿概率论读本》,优点和前述的教材一样。
  • 课程推荐:MIT 概率论
  • 重要概念:全概率公式、贝叶斯公式、极大似然估计、最大后验估计、正态分布、中心极限定理和大数定律、马尔科夫链。
  • 机器学习极度依赖概率论和统计,尤其是贝叶斯优化、极大似然估计,正态分布更是在各个学科和应用中反复出现。这里也推荐3blue1brown关于概率的视频,尤其是中心极限定理和正态分布相关的视频。

  • 进阶(新人跳过)

待补充。

其实很大一部分都是同学们的专业课或专选课,但是课内学的与达到掌握的程度并将其应用在实际问题的分析中之间有很大的鸿沟。

集大成者,MIT 18.086 Mathematical Methods for Engineers II,很适合工科生看,相当于把下面常用的一些数学方法糅合在一起。

  • 凸优化,非常重要
  • 复变函数(结合线性代数、信号与系统和自动控制原理)
  • 信号与系统,推荐奥本海姆绿皮书以及他的配套视频课程,在网易云课堂可以找到
  • 自动控制原理(time+freq domain),推荐笔者的专栏。
  • 李群优化,主要当作工具,用于姿态估计和机械臂控制、轨迹规划等。
  • 非线性优化(以梯度下降为基础展开的一系列改进),通用性非常强。
  • 偏微分方程,多元微积分其实已经学了一些。

7.6.2. 计算机科学基础

我们按照计算机从门电路到软件的顺序,层层抽象。

  • 计算机速成课

crash course for computer science,百万播放量的超人气课程,深入浅出短小精悍入门必看。crash course的其他课程也制作的非常精良且成体系,是作为闲暇时间扩展的不二之选。该系列从最底层的逻辑门逐层向上抽象,让你理解每一个抽象层次,而且包含了丰富的计算机科学史,从ENIAC讲到当下的技术前沿。看完了它你就对计算机有了基本的了解。

  • 模拟电路和数字电路速成

编程前你最好了解的基本硬件和计算机基础知识-模拟电路篇和编程前你最好了解的基本硬件和计算机基础知识-数字电路则是着重关注晶体管与逻辑门、数字逻辑电路层面的知识,对于进一步学习很有帮助。

  • 从逻辑门开始打造一台计算机

从0到1设计一台计算机是上面两个课程的进阶,学完这个,足以让你造出一台8位计算机!内含Logisim仿真实验,有兴趣的同学可以动手做一做。NandGame - Build a computer from scratch也是一个有助于硬件学习的网站,配合视频使用。

  • 学院派的计算机科学导论

CS50-Harvard University也是非常广为流传的课程。其课程体系来自学校的真实授课,课时数充足而且很详尽,老师讲得生动且深入浅出,也有很多梗。看完这个,对于计算机科学的基本理解就大差不差了,接下来在你感兴趣的领域进行深入即可。

  • (新人跳过)若有兴趣,可以继续学习CSAPP《以程序员的视角看待的计算机系统》、计算机组成原理、操作系统等CS专业的课程。

  • missing course in CS

一门非常流行的对CS传统教育的补充课程。


7.6.3. 编程语言和软件框架

  • C/C++

推荐浙江大学翁恺教授的C语言程序设计和C++面向对象程序设计,非常经典,不需要任何教材,跟着敲代码就行。推荐新手使用DevC++进行学习。

若你要阅读教材,《C++ primer plus》可以当作工具书进行查阅,《C++新经典》也是不错的选择。

  • CMake

Linux下管理和构建C++项目的必需品。CMake实践这个教程基本涵盖了所有常用指令。你只需要学习如何包含库头文件、链接动态库、将源文件加入编译这些基本操作即可。

  • C++多线程编程

C++11终于支持了平台无关的多线程库,基本的方法看这个就够了:c++11并发与多线程视频课程,课是好课但是老师废话有点多,建议4倍速观看。

  • Boost(新人跳过)

对标准库的扩展,有兴趣的自行了解,也不需要专门学。

  • Eigen & Ceres(新人跳过)

一个是矩阵运算库,一个是非线性优化库,跟着官方的tutorial学就可以了。

  • Python

推荐Python教程 - 廖雪峰和Python 3 教程-runoob,打开编辑器跟着敲即可。

  • Numpy

主要是数据类型和基本函数接口的使用,不需要跟着课程学,看官方文档即可。RM的主要场景是数据集处理和简单的图像处理。

  • Matplotlib

熟悉一些基本的画图、坐标系、换颜色等api就行。训练模型可能用得到。

  • Matlab(新人跳过)

也是脚本语言,写起来很方便关键是工具箱贼多,开发demo非常好用。跟随官方tutorial即可。simulink和simscape配合工具箱是强大的仿真工具。

  • Shell

掌握Shell编程,一篇就够了;Shell 编程快速入门;更多详细编程指引看Shell 教程


7.6.4. 数据结构和算法

(新人跳过)

个人认为没有必要专门买算法书来看(CS专业学生除外),在实践中遇到相应的问题,查找资料即可。

推荐一个OI网站:算法基础简介 - OI Wiki。可以当工具书查,过于复杂和艰深的算法和数据结构没必要学习。

Data Structure Visualization 包含了很多算法和数据结构执行过程的可视化,可以帮助理解。


7.6.5. OpenCV

开源计算机视觉的集大成者,重要性不言而喻。可以选择和教材一起学,如《学习OpenCV3》,《OpenCV3编程入门》等。也可以直接跟着官方的文档学,OpenCV提供了大量的example和tutorials。最好是学习某个算法后,去OpenCV寻找相应的接口和实现,然后自己跑一跑。本教程仓库下的opencv_test文件夹中提供了传统视觉算法中装甲板识别会使用的大部分api介绍,同时配有视频教程,戳这里:提取码-6666。

可能会用到的关键API和数据类型都已经在5.1介绍。OpenCV提供了CMake example,因此即使你还不会编写cmake,也可以直接将exmaple_cmake拿来照抄。

若你要在多视图几何以及3D视觉上开发深度学习的应用,可以使用Kornia,这是一个基于python/pytorch打造的兼容pytorch的库,其中使用到的数据使用pytorch基本的数据结构tensor。这意味着算法中的所有算子都是可微分的。Nerf、深度位姿估计等基于深度学习的3d视觉算法chagn'cha


7.6.6. PyTorch

PyTorch基本上是现在深度学习开发的标配了,虽然有很多其他的构建框架如Tensorflow、华为的MindSpore、百度的paddlepaddle,但是MetaAI(原facebook)还是使用起来最方便文档最详细最流行的深度学习构建框架。

不需要购买教材或者看什么课程。官方有一个系列7个视频,分别介绍了基本数据类型、和Numpy的数据转换、数据集处理、搭建模型、训练、评估和推理。看完之后就对整个框架有大致的理解了。这时候,自己尝试搭建一些简单的模型比如logistic regression、多层感知机,复现一下LeNet-5和YOLO,基本上算是上手成功了。李沐的动手学深度学习也是一个很好的范例,提供了Jupyter脚本供学习。

在大部分时候,编写数据集处理脚本,构建DataSet/DataLoader,确定预处理pipeline是最重要的,其他部分有大量开源可用。


7.6.7. 有用的网站和资源

善用软件和程序的官方文档,能在上面找到的就不要去第三方网站查找。大规模的开源程序一般也有自己的社区,如askubuntu、opencv-soc等。github上的issues可能也包括了你会遇到的问题。欢迎补充。




8.心得体会和给新人的学习建议

终于来到本文的最后一个部分了,这也算是笔者个人向的一些思考和观点。求同存异哦~

8.1. 我的RM历程

先简单说说我自己的RM经历,再总结一下感想。主要就是讲讲故事,嫌太长可以跳过不看。

笔者在2020年11月加入湖南大学跃鹿战队。因为我读的是自动化专业,最开始的时候为了“专业对口”本来想去电控组,后来视觉组的老学长把视觉吹的天花乱坠的,我直接就被带进了坑里。那时候视觉组只剩下两个人,一个组长,一个视觉项管。我们那一批进去的新人一共9个,都是经过了“非常严厉”的考核。记得考试试卷总共300分,我只拿了98分。当时组长让我们在30天内一天学一个topic,C++/python/3d视觉/强化学习/目标检测/多线程/双目相机/IMU啥都有,反正我基本没看懂什么,就是凭借印象在答题。之前只学过c,在招新之前的一个月自己速成了C++,还觉得自己挺牛逼的,身边的人都不会编程。以前还玩过arduino,就感觉自己快上天了。

进队之后我和一个研究生学长被分配做和电控对接的工作,编写串口通信的模块。那个研究生本科是搞电赛的,和我说这个非常简单,于是让我去写串口驱动。我就跑去搜索linux下的串口驱动,源码打开看根本看不懂。过了一个星期组长问我做的怎么样了我说没什么进展,他让我把这几天做的东西告诉他我就如实说了。没过多久研究生学长就跑了,剩下我一个人看着serial port programming under Linux这本书,慢慢啃。后来把东南大学2018年开源的代码拿来复刻,结果linux下qt不会写qmake和cmake,一直运行不起来,报错。那时候根本不知道makefile和动态库是啥玩意。

组长却对我们有很高的期望,认为我们这个赛季就可以做出自动步兵(2020赛季规则第一次出现),给我们每个人都分配了任务,我负责用卡尔曼滤波进行状态估计,然而我连KF和状态估计是什么都不知道。组长还要求我们每天刷3道洛谷的题并截图在群里发出来。一个信科院的同学受不了这个压力没过多久就溜了。组长和我们说,我希望我们组在一年后,至少有3个人考CCF高于400分,剩下的同学也是至少350分的水平。现在让我去考,估计也就是两百出头。因为啥也不会,传统的装甲板识别实际上也没写明白,都不知道调曝光,用开源代码跑出来的效果也不好,于是就天天摸鱼,摆烂,想着抱其他队员的大腿混一个参赛名额。

组长也发现这样下去不行,于是就和我们说你们先慢慢发育,有什么事我扛着,我会把大体的框架搭好,以后你们只需要改进算法填进去,框架不用动,可维护性很强。我听着一愣一愣的的,继续看我的串口编程手册。寒假回家了,组长让我们学ROS,可我连linux下的opencv都没装好,更不知道ROS为何物。组长问我串口什么时候能写完,如果ddl前还写不完,我就不用继续干了,他来写,我就只能当梯队队员。我赶紧临阵磨枪,找了好几个开源代码的串口通信来看,勉强缝合了一个丢包率为5%,每秒只有100帧的通信模块。

开学以后队长找到我,问我视觉的进度怎么样了,他觉得组长不靠谱。几个电控的学长也说,你们新人要自己成长,组长弄了那么久也没搞定,再这样下去进度会爆炸。可我连装甲板识别都还不会,这是其他同学负责的。二月中旬,原本专门负责做自动步兵跑路了,他去负责我们学校赛车队的无人车,于是人又少了一个。组长三月在群里问有谁愿意接任,他要去创业了,没空管队里的事。群里没人回应。然后他挨个问。我不知道哪里来的勇气,和他说如果其他人都不来那就我来吧。于是我就成了视觉组组长,最年轻的,啥也不会的组长。

4月份马上就要高校联盟了,做装甲板识别的同学还没搞定,因为我们这时候还不会调曝光。我开始天天往实验室跑,各种找开源。终于在CSDN上看到哈工程的一个老学长发的教程,这才明白问题出在相机的曝光上。这时候我们终于安装了qv2l4,开始调节相机的参数。传统装甲板识别算法最后以每秒50帧的速度跑起来了,程序里有各种冗余的操作,for(for(for)))循环,以及内存泄漏。电控有一个超强的老学长威哥,也就是vSeasky(可能很多圈内的人都认识他),他看不下去了开始监督我搞串口,我终于在一周之后看懂了前组长和威哥一起定制的协议,成功和电控进行了第一次双向通信。我那天开心了好久,觉得自己做的事总算有回报了。晚上还特意学习了cp210x和ch340的区别,顺便看了看Linux的文件系统。在这之前,我都不会用终端也不懂shell,觉得用linux就是为了装逼。

高校联盟赛来临前的两个星期,我终于在自己的电脑上用cmake成功编译OpenCV,虽然是跟着教程来的,只是按部就班的操作,祈祷他不要出现任何问题。传统装甲板识别很容易发生误识别,因为我们没有数字识别的步骤,我看不懂SVM。模板匹配的速度实在太慢,我还不知道要降低ROI的分辨率,加上匹配之后速度直接掉到1秒10帧,我们用的还是i7-8625U的minipc。信科院的一个同学果然还是有底子,他用ROCO官方数据集,跟着nanodet的readme把网络成功train了起来。对比我那个垃圾的传统算法,神经网络实在太惊艳了,我们全实验室都在高呼牛逼,那个晚上大家都觉得,高校联盟赛我们会崛起。

但是因为机械出车不久,电控还在调车,而且出现各种问题。那个时候我还不清楚什么叫DBUS,超级电容是怎么做的。我只是听它们说威哥做的硬件很牛逼,好像机械拉了电控后腿,因为我们的步兵车有22kg重。到了武汉工程大学的比赛现场,本来打算调一下英雄的视觉。那是我第一次看到云台跟随装甲板移动。神经网络的鲁棒性很好,不论亮度如何,我们的bbox都准确地框住了装甲板。彼时还没有很多队伍使用神经网络,好多学校都围上来看,说你们的识别好牛逼,结果这时候,英雄的云台疯转把hdmi线扯断了,小屏幕也摔了。我和另一个视觉的同学就在武汉度过了无所事事了3天。

结果显然是很惨的,小组都没有出线,和别的队伍菜鸡互啄。我看着那些实现了基本功能的队伍,感慨做这些东西有这么难吗我也要好好做,甚至只要把最基本的运动和击打实现了,就能取得不错的成绩啊。这时候我就开始恶补传统算法,找开源来看。那时候的开源代码,我是直接在搜索引擎里面搜“RoboMaster装甲板识别算法“找到的,我不知道RM论坛,也还不会用github。后来我和何老板发现了吉林大学的开源,里面用了多分类的支持向量机。下载下来跑,发现效果非常好。何老板说干脆就用它们的算了,都不知道自己写要写到什么时候,我说不行,我一定要自己弄。我每天就呆在实验室里,尝试各种腐蚀和膨胀的操作,不知所以然地拖动相机参数的拖条,一个个试。不知道把参数写到文件里去读取,每次改参数都要重新编译,而且还是单线程编译,每次至少15秒。

到了五月,我彻底放弃了传统算法,何老板说干脆就用神经网络吧,不要用pnp算距离了,直接用bbox的面积和补偿角度拟合出一个函数关系。于是我们就去实验室楼下拿车测射表,拟合两者的关系。这时候,组里实际上只剩下我,何老板和黄老板了。我们都开始尝试卡尔曼滤波器做预测,但直到后来我才发现做了也是徒劳,因为我直接在图像坐标系上预测装甲板的运动,我还不知道相机要和IMU进行联合标定。玩不明白Mahony解算,Yaw轴每秒漂移一度。好歹这时候可以打静止目标了,动起来就没办法,因为没有预测总是落后一截,也没有反陀螺,全靠蒙。但我和何老板还是戏称我们的算法是超级自瞄,为此骄傲了好久。我们还决定,认真地给代码加上注释,提高可读性,我们要冲击开源特等奖,有10w奖金。他甚至已经想好怎么花了。现在回头看看之前写得代码,好垃圾啊。

因为南部赛区和总决赛紧凑在一起,我们恰好有机会前往春茧体育馆。分区赛的结果也是显而易见的。一个月造好的工程机器人、像坦克一样的步兵机器人,没有加上功率限制,弹道也是一塌糊涂。英雄在生死战还把电机烧坏了,最终没有上场。但是在深圳的旅途还是很愉快的,我亲眼目睹了哈工大、深大和上交的机器人。当时和机械的同学感叹,它们的机器人好整洁,走线好漂亮。在离开的前一个晚上,我们聚在陈老师的房间里,反思这个赛季的不足,畅想着未来。大家都说明年会留下来继续做,每个同学都像要做出让自己满意的机器人。我们似乎都踌躇满志,认为来年一定是跃鹿的复兴之年,因为之前的学长学姐都做的不行,所以不是我们的锅,2022赛季一定会充满希望。

在家里,因为总决赛变成了技术交流,我只能用腾讯会议看现场同学的直播。那个晚上,我愿称之为RM的春晚。川大如精准的飞镖一度让人认为拥有它能够制导;华南虎和哈工大的舵轮步兵实在太强了,东大的机器人还是那么稳定,深大展现出的能力也让人慨叹进步速度之快......在总决赛的总决赛上,哈工大和上交强强对话,整个圈子的RMer都在关注这场巅峰之战。哈工大的哨兵反击如洒水,舵轮步兵1秒能跑六米;虽然斯基的工程也很强,但是上交的工程机器人更强。两边队伍激活能量机关的能力旗鼓相当。这是无可争议的决战。最终,由于工程机器人发生损坏,哈工大败给了上交,但它仍是我心中的无冕之王。上交用绝对的实力告诉大家,这是机器人该有的样子。

暑假的时候我开始搞深度学习,终于弄明白了之前为什么会把紫色装甲板认成红色装甲板。上交的四点模型让我夜不能寐。我每天的生活就是,刷教程,听课。到现在来看这应该是我成长最快的时间段,而且在这时候,我知道了RM视觉交流群的存在,我才发现,我们真的好拉。何老板距离保研差一名只能考研,黄老板要保研,其他组员实际上只是挂名。历史再次重演,我成为了光杆司令。这时候,我还在b站发现了dr_can,我开始学习控制理论,对于机器人整体也有了新的理解。那段时间沉浸在知识海洋里的感受非常令人愉悦,每天都有新的收获,每天都真切地感觉到自己在变得更“厉害”。

在9月开学的时候,我们要招新了。我们收了10个大二的小朋友,和四个大一的小小朋友。当时的我也觉得,这是质量最高的一届视觉,我们要起飞了。不过我终于和前组长感同身受,一些在我看来想当然的东西,新队员都要学习很久。而且我常常对他们的效率和能力有过高的估计,从来没有想过当时的我也是经常摸鱼,需要大量的休息时间的。不过不管怎么说,他们当时的水平显然远超一年前的我。我给他们写了非常多的培训资料,在这个过程中也给自己的知识体系查缺补漏。十月份的我终于算是拥有了一个合格的视觉组成员所有的知识。那是2021年10月6号,我开始编写这篇教程,我把教程目录放在了csdn上,并给自己立了一个flag,两个月内编写完全部内容。

我很快确定了程序框架,只待新队员把里面的算法填充完毕。但是队伍的进度总是不尽如人意,机械迟迟没有出车,巨蜥一个人画了3个机器人;电控除了队长之外,竟然没有一个人能够写整车的代码。还好,因为我们比赛参加的多,踩了狗屎运变成了甲级战队,不需要提交中期考核。在寒假的时候我就进入了半摆烂的状态,自认为自动控制原理和信号与系统学得很好,但是却考出了79和74分高分,彻底击碎了我的保研梦。

2022年伊始,大家都在空间和朋友圈里展望新的一年,我却开始为队伍可能的分崩离析而担心。那些曾经说要圆自己机器人梦的同学,一个个都离开了战队。机械组组长变成了大二的龙爷,他取代我成为最年轻的技术组组长。我是唯一留在队里的大三成员,江哥是最后一个18级学长。我继续编写这份教程,它的篇幅也随着知识的增多发生了超级加倍。原本想写六万字,变成了12万字,写着写着发现详略不得当,字数又开始疯狂地往上蹭。

寒假回来还是没有可以调的车,我认为我已经教给新队员所有必须的知识,可他们的进度也还是一筹莫展。我们又招了两个新同学,然后有4位新同学退出。队里的矛盾似乎越来越多,我也曾经尝试写下大段的鸡汤和自己所谓的肺腑之言,现在看来好像也没有任何用处。不知道从什么时候开始,我有了新的称呼:曾老师。这个名字也让我有些许的得意,听起来我在机器人方面已经颇有造诣。甚至有一次,实验室楼道里的两个大一同学真的以为我是老师,拿着他们的TI寻迹小车过来问问题。

从21年开始,我就再也不去教室听课,之前好歹还会意思一下给老师点面子。每天就在宿舍和实验室往返,两点一线。但是在实验室里我也没有做什么和比赛直接相关的事情。我对自己说,我可能已经预料到这个赛季的结果了,于是我开始提前展望未来。这时候我也开始憧憬学术道路,每天都花费数个小时的时间阅读目标检测相关的论文,直到现在才发现,这里已经没什么可做的东西了,3d重建、多模态和超大规模的预训练模型才是深度学习的未来。我的目标又开始动摇了,我慌慌张张地买来几本控制理论相关的经典作品,对着上千页纯英文的书就是一顿啃。好歹还算是有些收获,多少对经典控制理论和现代控制理论有了新的理解。但是当我发现非线性控制和鲁棒控制中艰深的数学时,我又开始怯步了。

到了四五月份,对于比赛,打心底我就属于彻底摆烂的一类。因为疫情高校联盟被推迟到八月或九月。完整形态考核的时候恰逢期中考,江哥一个人调完了所有的车,秦姐也回来帮忙,抢在ddl前提交了考核视频和材料。赛务和我们说,我们的完成度很危险。最后,我们还是被扣了25块钱。这段时间,我和龙爷偶尔会在晚上抽空去后湖滑板,滑完之后就在路边思考人生。

疫情在这个时候开始再次反复,在一定程度上它也拯救了我们的进度。期末考后又是一个混乱的星期,我一个星期速成的电机学期末考了77分,我一直担心会挂。带着搭载着屎山码的还没装好的机器人,我们坐着动车前往常州。由于行程安排不当,我们在武汉转车时错过了第二趟车,全体成员只得改签——这也是我们这个混乱赛季的缩影。

结果是不出意料,再次无法出线,拿着5比1的战绩耻辱下播。赢的一小局,还是因为对方两个机器人突发状况发生损坏。我望着赛场和颁奖台,心底尽是无奈和自嘲,也许我不摆烂,做点什么就会让结果不一样呢?新队员们亲临现场,看到强队之间的对决,说不定也有了不一样的感受。离开的前一天晚上,陈老师再次让我们聚在一起,进行本赛季的最后一次会议。老队员和新队员都聊了很多,大家都有不同的想法,实际上也好像没有畅所欲言。新队员们高低充满了斗志,老队员们把遗憾藏入心底,也许要再战斗一个赛季,或是带着别样的情绪离开。总之,大家都会有光明的未来。

回到长沙后,留校的同学开始准备在九月份可能到来的高校联盟赛。龙爷和楠神大改实验室布局。还有很多问题没有解决,也许在某个时间某个角落就埋下了不定时的炸弹。好在至少表面上看起来,我们重新走入了正轨。2022,才是我们队伍出发的元纪年。

于是现在,我在福建的一个小山城里,坐在电脑前,对这份尚有许多缺陷的教程进行收尾工作,再把它献给屏幕前的你。

从一开始的半桶水都没有还喜欢装逼,逐渐在做机器人的过程中跨越了愚昧的山顶,到最后终于明白自己有太多东西需要学习。在RoboMaster里,认识了非常多志同道合的朋友,也真正让我感受到什么叫人外有人。当我在为自己懂得编程而沾沾自喜的时候殊不知别人在高中甚至初中就已经于OI里大杀四方,当我觉得自己会用单片机高人一等的时候,早已经有人在它上面跑起了MPC和FOC;我也曾认为自己在目标检测上颇有理解,大神们却默默地把自己的算法迭代了好几次。

经典老图永不过时

不能单单考虑理想状况,在诸多内外部因素的限制下,RoboMaster对于很多同学来说是一个美丽而容易破碎的梦。我不敢说我是因为参加比赛而错失保研的机会,但我不后悔参加RM。在大学的这三年,每个学年的学费是6500CNY,虽然我没有去上课,但这实际上像是我加入机甲大师的入场费。我在实验室里学到了别人四年学不来的东西,交到了别人四年接触不到的朋友。至少,这对我来说是一个局部最优解。

人大概都是喜欢安逸的,我也是如此。社会学研究发现,当人处在一个集体当中时,会降低自己承担责任的意愿。一开始的时候,我进队就只想抱抱别人的大腿,只要别人不催,我的工作就原地踏步。现在想来,我的学弟学妹们可能处在相似的境遇吧。这在某种程度上,也是一种畏难情绪,好像你不去做它,它就难不倒你似的。有很多队员都会把参加机甲大师看做是在学校的实验室献爱心,用爱发电。但是谁又需要你的这点施舍呢?你所做的一切都是为了你自己,为了你心中的那个梦想。

我的梦想是什么?小时候的梦想是成为一名伟大的科学家,要留名青史。但越是长大,就越感觉到有一些对我很重要的东西被剥夺。直到参加RoboMaster后的某个时刻,我突然明白我能在这里找到心中失落的遗迹。我不想讨论一些关于目的、意义和存在的宏大话题,也可能要暂时抛开对人生价值与真理的追求。因为我的故事还没有结束,还需要收起我的“感慨万千”继续前进。更多的精彩,更多的风景,也许是更多的遗憾。我会把它们留给回忆,待到画下句点再向你娓娓道来。

下个赛季,我会继续战斗,不摆烂的那种。

至于到底有什么感想,百闻不如一见。

2023/06/01增补:最后一个赛季以遗憾收尾,小组赛与另外两支队伍积分相同,小分落败。

愿你拥有一个不留遗憾的青春,加油。


8.2. 常见Q&A

欢迎补充。如果你有问题,也可以提出。

  • 感觉视觉要学的东西好多好难,怕自己坚持不下来,配个环境都快劝退了

算法组的成长路线确实比较长,要点的技能较多,不过只要按照顺序,把基础知识学好,往后的学习就不会有太大困难。如果你一上来就想要装Linux系统想要安装OpenCV,然而连计算机是怎么工作的都不清楚,那必然是难过登天的。当你觉得某些事物无法理解的时候,基本上不是所谓智商或者能力的问题,而是缺少前置知识。所以一定要登高自卑,从基础开始学习,万丈高楼平地起。

  • 自学的效率太差,而且学完很快就忘了

如果你希望有老师讲,跟着网上的视频教程一起学就完事了。自学最重要的其实是没人给你布置作业,一定要在学完以后立刻实践,自己把代码敲出来。如果不写代码,看完就觉得自己会了显然是不现实的,参考艾宾浩斯遗忘曲线。用得越多记得越牢固。至于效率可以找一些相关方面的方法论看一看,摸索出适合自己的学习方法。可以尝试记笔记、写文档等,也可以在学完一个知识点后尝试向别人讲解、写博客,教学相长也。

  • 参加RM要花的时间多吗,会不会影响课内成绩,还能参加其他项目吗

我们不谈RM在成绩之外可能带来的好处。

显然参加比赛你要学习的知识在课内很可能不会包含,因此必然要在空余时间学习计算机视觉和机器人相关的知识。至于是否影响课内成绩,笔者认为这因人而异。首先,学习比赛相关的知识在很大程度上会帮助你理解课内知识并将它们应用到实践中去,你在学习扩展知识的过程中,也可能对基本的概念有了新的理解。同时,课外知识的学习在相当程度上依赖于基础课,包括高数、物理等,它们会对这些知识的掌握提出更高的要求,也有利于你课内知识的学习。其次,比赛中很多学到的知识在课内是用不上的,这点毋庸置疑,所以这些知识的学习对于你的成绩没有任何帮助。当然你可以倾向于在选修课环节选择那些和计算机视觉、机器人、机器学习等有关的课程,这有可能可以帮你刷出高分。

至于学习比赛相关的内容是否会挤占课内知识的学习,这点也是必然的。如果你认为成绩更重要,GPA对于保研、留学申请确实必不可少,那你应该减少在RM上的投入,专心学习课内知识,多刷题。在RM上付出的时间增加了,就意味着你课后只有更少的时间用于预习/复习和反刍知识。你应该权衡利弊,考虑自己安排时间的能力和学习的效率。

在RM圈子里,不乏一些成绩竞赛科研多点开花的大佬;也有一些因为全心全意投入比赛完全抛弃课内学习的例子(笔者不建议这样)。你应该想清楚,你要追求的事情是什么,再决定在不同方向上的时间分配。

  • 自己不是CS、EE相关背景的,做这个会不会不合适

笔者个人认为不会。21世纪是信息科学和大数据的时代,学习计算机相关的知识是必要必须的。第四次工业革命已经席卷了人类生产生活的各个角落,不会使用计算机简直是天方夜谭。如果你已经打算转专业、跨考/跨保,那么这就更不成问题了。另外,当今也是学科高度交叉的时代,学习计算机视觉和机器人说不定可以在你的major发挥别样的作用,你也会比别人多一项有用的技能。据笔者所知,视觉算法组不乏来自物理学科、机械学科、生物医学工程和数学专业的同学。

此外,考虑当今(注意是当今)的就业情况,计算机/电子/自动化/机器人行业的薪酬普遍较高(虽然996,但是其他厂也996,吧?),根据供需关系你也知道,学视觉和算法不亏。

  • RM做的东西以后用得上吗?是不是为了比赛而比赛?

学校上的课以后可能用不到,但是RM做的东西,在之后的科研或工作中一定用得上,只要你投身于相关行业。我们不谈所谓思维和方法论的突破,就看在这里学习的知识是否能用上:人脸识别已经深入千家万户,自动驾驶更是百家争鸣称为一片红海;你的手机相册能够帮你把照片分门别类放好,打开相机屏幕上自动出现的取景框内的物体位置和分类;搜索引擎可以用图片进行索引,AI自动生成视频字幕,甚至能告诉你视频里发生了什么;计算机作的画媲美人类画家,甚至能写小说谈时政;P图如此方便,美颜一键年轻20岁;庞大的生产线上只有机器运作的白噪声,百平面的车间空无一人······

笔者这里说的直白一些,很多同学都知道互联网+、挑战杯、创新创业大赛等。把RM中打造的机器人改头换面,开开脑洞实现新功能,找一个导师带你参赛,你也可以在它们那里大杀四方。至于各种横向小项目,拥有了这些知识基本上手到擒来。


8.3. 我对计算机视觉的一些理解

鄙见,管中窥豹,欢迎交流。

参考文献

所有的图片已在对应注释中添加引用来源,此处不再重复引用。介绍到的相关论文、大部分网络资源,也已超链接的形式放在文中相应位置。这里只添加一些无法在文中确定详细引用位置的内容。

[1] Lowe, D.G. Distinctive Image Features from Scale-Invariant Keypoints. International Journal of Computer Vision 60, 91–110 (2004). https://doi.org/10.1023/B:VISI.0000029664.99615.94

[2] Local Feature Descriptors: Harris and Hessian Corner Detector.[CV] 10. Local Feature Descriptors: Harris and Hessian Corner Detector | by jun94 | jun-devpBlog | Medium

[3] Signals & Systems,Alan V. Oppenheim, Alan S. Willsky, Syed Hamid Nawab,Pearson Educación, 1997.

[4] 《学习OpenCV3》, Adrian Kaehler ,Gary Bradski,清华大学出版社, 2018-7.

[5] 《Robomaster——关于视觉组,你想要了解的都在这里》,Coo㏒ ∮的CSDN博客,https://blog.csdn.net/weixin_42754478/article/details/108159529

[6] RoboMaster_Raring_Ringtail的博客-CSDN博客,https://blog.csdn.net/u010750137/category_8994384.html

[7] Steve Brunton - YouTube,https://www.youtube.com/c/Eigensteve

[8] Lecture 05 - Scale-invariant Feature Transform (SIFT) - YouTube,https://www.youtube.com/watch?v=NPcMS49V5hg&t=1128s

[9] SEU-SuperNova-CVRA/Robomaster2018-SEU-OpenSource,https://github.com/SEU-SuperNova-CVRA/Robomaster2018-SEU-OpenSource

[10] Harry's Blog,https://harry-hhj.github.io/

[11] 《机器人视觉测量与控制》,徐德,谭民,李原,第3版,国防工业出版社,2016. 1

[12] FMCW激光雷达科普(上):基本概念、技术路线、优势及争议和误解,https://zhuanlan.zhihu.com/p/424953394

[13] RoboMaster创梦之翼战队自瞄系统设计,王洪玺,哈尔滨工程大学

[14] 《计算机视觉中的多视图几何》,理查德哈特雷

[15] 《数字图像处理》,冈萨雷斯

[16] 《SLAM视觉14讲》,高翔

[17] 《机器人学导论》,约翰克雷格

[18] 《机器学习》,周志华