让我们一起来看看我们第一位投稿大佬在 hook 机制方面的分享吧！

欢迎大家在留言区积极参与讨论哦！

本文由 Yimian Dai 投稿，OpenMMLab 社区专属ID：429212，目前正在攻读博士学位，研究领域涵盖图像处理、计算机视觉及遥感。

本文内容

为什么要设计 Hook 机制？

Hook 机制的运作流程

Hook 机制的底层实现

案例：mmseg 中的 Hooks

1. 为什么要设计 Hook 机制？

OpenMMLab 系列的一个显著特点就是其采用的 Hook 机制。与易于理解且对新手友好的 Gluon-CV 相比，Hook 机制无疑增加了初学者使用 OpenMMLab 系列工具箱的难度。那么，为什么要引入 Hook 机制呢？

实际上，OpenMMLab 中的 Hook 机制是面向切面编程（Aspect Oriented Program, AOP）思想的一种体现。从抽象的角度来看，问为什么要引入 Hook 机制，实际上是在问在软件开发中，为什么要采用这种面向切面编程的设计模式。

我个人浅显的理解，提出面向切面编程是为了解决面向对象编程（Object Oriented Programming，OOP）中的代码重复性问题。面向对象编程的思想是职责分配，将功能分散到不同的对象类中，在不同类中设计不同的方法。如果两个类 A、B 都需要使用同一种方法，可以将该方法写在一个独立的类 C 中，然后让这两个类各自继承类 C。然而，这么做存在两个问题。

第一个问题是，对于不支持多重继承的语言，如 Java，继承类 C 后就不能再继承其他类了。如果类 C 中的功能并不是类 A、B 的主要功能，通过继承类 C 来获取其方法就不合适了。简单的解决方案是分别在类 A、B 中实现类 C 中的那个子函数，这样一来，相同的代码就会在两处存在，代码的重复性大大增加。如果你要修改该方法，必须在两处都进行修改。两处还可以处理，如果这种情况有 m 个，每个重复 n 处呢？

第二个问题是，即使能够继承，类 A、B 与类 C 就耦合在一起了。如果存在一个与类 C 具有相似但不同子函数的类 D，我希望能够让类 A、B 通过用户配置选项动态地选择是调用类 C 还是类 D 中的子函数，那么这种直接继承的方案也没法提供这种动态选择的灵活性。

本质上，除了继承之外，面向对象编程所追求的封装特性切断了类与类之间的联系和共享。然而，为了降低代码的重复性、提升软件的模块化水平，需要将分散在各个类内的重复代码统一起来，两者之间就存在了矛盾。

这种在程序运行时，动态地将所需代码插入到类的指定方法、指定位置上的编程思想就是面向切面编程。其中，几个类共同需要调用的、为此被抽取出来的代码片段称为切面，其会在程序运行时被插入到指定类的指定方法中，而被插入的那些类、那些方法称为切入点。面向切面编程，使得我们可以把和当前的业务逻辑无关的部分抽到单独的一层中去，实现无侵入式的功能扩展。

正是通过 Hook 机制，OpenMMLab 系列能够对网络实现以及算法训练、测试流程进行抽象和解耦，从而达到了相当高的模块化水平，即重复代码大大减少。

2. Hook 机制的运作流程

Hook 机制并不是 OpenMMLab 独有的，只是由于我代码经验不足，第一次见到而已。钩子编程（hooking）是计算机程序设计中的术语，指通过拦截软件模块间的函数调用、消息传递、事件传递来修改或扩展操作系统、应用程序或其他软件组件的程序执行流程。其中，处理被拦截的函数调用、事件、消息的代码被称为钩子（hook），也就是前文 AOP 编程中的切面。

在 OpenMMLab 中，Hook 机制由 Runner 类（如 IterBasedRunner, EpochBasedRunner）和 HOOK 类（如 EvalHook）配合完成，共同构成了训练框架的架构规范。

首先，在 OpenMMLab 中，负责网络训练测试全流程的 Runner 类在训练测试周期中定义了一系列触发器，如下所示：

# 省略 ...self.call_hook('before_train_epoch')for i, data_batch in enumerate(self.data_loader):    # 省略 ...    self.call_hook('before_train_iter')    # 省略 ...    self.call_hook('after_train_iter')    # 省略 ...self.call_hook('after_train_epoch')

其次，在与 Runner 类配合的 Hook 类及其子类中，也定义了一堆与上面 Runner 类的触发器中 before_run, before_epoch, before_train_iter, after_train_iter, after_epoch, after_run 等步骤/时刻/节点同名的函数，被称为钩子函数，如下所示：

class Hook:    def before_run(self, runner):        pass    def after_run(self, runner):        pass    def before_epoch(self, runner):        pass    def after_epoch(self, runner):        pass    def before_iter(self, runner):        pass    def after_iter(self, runner):        pass    # ... 省略

当然，上面这个 Hook 类是最原始的实现，也就是基本什么功能都没有实现。如果想定义一些操作，实现一些功能，可以继承这个类并定制我们需要的功能，比如 mmcv.runner.hooks.evaluation 模块中的 EvalHook 类继承了最原始的 Hook 类，将里面的子函数基本都具体实现了；而 mmseg.core.evaluation 模块中的 EvalHook 类则进一步继承了前一个 EvalHook 类，重写了 after_train_iter 和 after_train_epoch 两个子函数。

有了相互配合的 Runner 类和 Hook 类之后，Runner 类实例运行到特定时刻，就会通过触发器函数调用各个 Hook 类中的钩子函数，从而完成特定的功能。例如，每个或者隔几个 after_epoch 或者 after_train_iter 触发器时刻，可以通过 EvalHook 的 after_train_iter 函数调用 _do_evaluate 函数完成对 validation set 的 evaluation。

个人感觉，这套 Hook 机制很像通信系统中的轮流询问机制，是一套在算法生命周期中规定好了种种操作的训练框架规范。其之所以起作用，是因为在 Runner 类的被调用方法中，每一个节点都规定了 call 相应 hook 函数的操作。Runner 类在训练过程中会依次轮流询问端口，也就是依次 call 下每个节点的 hook 函数，如果对应钩子函数有被专门定制过，那就执行该功能。如果没有，那就是个空函数，直接 pass 了，继续执行下一步，从而实现了拦截模块间的函数调用、消息传递、事件传递，从而修改或扩展组件的行为。

3. Hook 机制的底层实现

在清楚了 Runner 类与 Hook 类配合实现 Hook 机制的运作流程后，还剩下两个问题。第一个问题是，怎么让 Runner 类实例知道去调用某个具体的 Hook 类实例的子函数，也就是怎么将 Runner 类实例和 Hook 类实例关联起来？第二个问题是，Runner 类实例可能会调用多个 Hook 对象，每个 Hook 对象都会有各自同名的子函数，比如 after_train_iter，这种情况是如何处理的？

对于第一个问题，是通过 Runner 类的 register_hook 函数将 HOOK 类实例注册进 Runner 类实例的。我们以 MMSegmentation 为例，在训练模型的时候，会调用 mmseg.apis 模块的 train_segmentor 函数。其中有两步是给 IterBasedRunner 类实例 runner 注册 training hooks 和 validation hooks：

runner.register_training_hooks(cfg)runner.register_hook(eval_hook(val_dataloader, eval_cfg))

Runner 类提供了两种注册 hook 的方法：

1. register_hook 方法是直接传入一个实例化的 HOOK 对象，并将它插入到 Runner 类实例的 self._hooks 列表中；

2. register_hook_from_cfg 方法是传入一个配置项 cfg，根据配置项来实例化 HOOK 对象，然后再将其插入到 self._hooks 列表中。

实际上，第二种方法就是先调用 mmcv.build_from_cfg 方法生成一个实例化的 HOOK 对象，然后再调用第一种 register_hook 方法将实例化后的 HOOK 对象插入到 self._hooks 列表中。

有了存有注册了的 Hook 类实例的 self._hooks 列表，Runner 类在运行中调用注册了的 Hook 类实例的子函数也就顺理成章了。看一下 BaseRunner 类中 call_hook 函数的定义，其中 fn_name 就是 self.call_hook('after_train_iter') 传入的 after_train_iter。getattr(hook, fn_name)(self) 其实就是在调用 self._hooks 列表中的 hook 对象的名为 fn_name 的函数，比如 EvalHook 类实例的 after_train_iter 方法。至此，第一个问题，如何动态地将想要的 Hook 类实例的某个方法插入到 Runner 类实例的运行过程中已经实现了。

def call_hook(self, fn_name):    """Call all hooks.    Args:        fn_name (str): The function name in each hook to be called, such as            "before_train_epoch".    """    for hook in self._hooks:        getattr(hook, fn_name)(self)

对于第二个问题，从上面 call_hook 函数的定义也可以看出，在 Runner 实例的 run 函数运行过程中，在每一个设置 call_hook 函数的节点，都会就轮流执行一遍 self._hooks 列表中所有 hook 实例中该时刻对应的方法。比如，对于 after_train_iter 这个时刻，就是遍历一遍所有 hook 实例的 after_train_iter 方法。如果只有一个 Hook 实例重写了该方法，而其他实例的该方法都是 pass，那也无所谓。但如果有两个及以上实例的该方法实现不是 pass，那这就涉及到一个哪个实例的方法该先被调用的问题，具体到程序中，则是每个 Hook 了实例被插入到 self._hooks 列表的位置的前后，因为 call_hook 函数是依次调用的。

优先级这点，在注册 hook 的时候就已经实现了，priority 是默认变量。从下面 register_hook 函数的定义就可以看出，对于新注册的一个 Hook 实例，按照其指定的优先级，没有指定就默认 'NORMAL' 优先级，插入到 self._hooks 中，优先级越高的，越靠前。如果新注册的 Hook 实例与就有的 Hook 实例优先级相同，那就按照先来后到，先来的排在更前面。至此，第二个问题也解决了。

def register_hook(self, hook, priority='NORMAL'):    """Register a hook into the hook list.    The hook will be inserted into a priority queue, with the specified    priority (See :class:`Priority` for details of priorities).    For hooks with the same priority, they will be triggered in the same    order as they are registered.    Args:        hook (:obj:`Hook`): The hook to be registered.        priority (int or str or :obj:`Priority`): Hook priority.            Lower value means higher priority.    """    assert isinstance(hook, Hook)    if hasattr(hook, 'priority'):        raise ValueError('"priority" is a reserved attribute for hooks')    priority = get_priority(priority)    hook.priority = priority    # insert the hook to a sorted list    inserted = False    for i in range(len(self._hooks) - 1, -1, -1):        if priority >= self._hooks[i].priority:            self._hooks.insert(i + 1, hook)            inserted = True            break    if not inserted:        self._hooks.insert(0, hook)

4. 案例：mmseg 中的 Hooks

在下图中，我整理了 mmseg 的 tools/train.py 整个运行周期中会用到的所有 hooks 对应的具体的 Hook 类以及相应被调用的时刻。

（点击查看大图）

另外，以 IterBasedRunner 为例，整理了这些 Hooks 被调用的时刻以及相应的优先级（先后顺序）。

（点击查看大图）

大家是否在 Yimian Dai 同学的启发下有所收获呢？

欢迎各位大佬多多投递想法，也欢迎大家一同交流

# 面向对象编程  # 操作系统  # 工具  # ai  # 为什么  # red  # Java  # 架构  # Object  # 面向对象  # 封装  # 子类  # 继承  # 堆  # 多重继承  # 对象  # 事件  # 算法  # 低代码  # 第一个  # 子函数  # 第二个  # 问题是  # 类中  # 那就  # 大佬  # 列表中  # 实现了  # 几个 

相关栏目： 【 网站优化151355 】 【 网络推广146373 】 【 网络技术251813 】 【 AI营销90571 】

相关推荐： Laravel怎么使用Blade模板引擎_Laravel模板继承与Component组件复用【手册】  网站制作报价单模板图片,小松挖机官方网站报价？  Laravel如何使用withoutEvents方法临时禁用模型事件  如何在万网利用已有域名快速建站？  详解Android——蓝牙技术 带你实现终端间数据传输  如何实现javascript表单验证_正则表达式有哪些实用技巧  HTML5空格在Angular项目里怎么处理_Angular中空格的渲染问题【详解】  大连网站制作费用,大连新青年网站，五年四班里的视频怎样下载啊？  如何正确下载安装西数主机建站助手？  Laravel怎么集成Vue.js_Laravel Mix配置Vue开发环境  网站制作企业,网站的banner和导航栏是指什么？  如何在 Telegram Web View（iOS）中防止键盘遮挡底部输入框  Angular 表单中正确绑定输入值以确保提交与验证正常工作  如何获取上海专业网站定制建站电话？  魔毅自助建站系统：模板定制与SEO优化一键生成指南  Laravel模型事件有哪些_Laravel Model Event生命周期详解  ai格式如何转html_将AI设计稿转换为HTML页面流程【页面】  香港服务器选型指南：免备案配置与高效建站方案解析  国美网站制作流程,国美电器蒸汽鍋怎么用官方网站？  Laravel如何发送邮件和通知_Laravel邮件与通知系统发送步骤  Laravel如何使用Scope本地作用域_Laravel模型常用查询逻辑封装技巧【手册】  javascript事件捕获机制【深入分析IE和DOM中的事件模型】  如何安全更换建站之星模板并保留数据？  Laravel如何实现数据导出到CSV文件_Laravel原生流式输出大数据量CSV【方案】  如何在阿里云ECS服务器部署织梦CMS网站？  Android okhttputils现在进度显示实例代码  详解jQuery中基本的动画方法  Bootstrap整体框架之JavaScript插件架构  怎么用AI帮你设计一套个性化的手机App图标？  如何在景安云服务器上绑定域名并配置虚拟主机？  利用vue写todolist单页应用  Laravel如何处理JSON字段的查询和更新_Laravel JSON列操作与查询技巧  如何为不同团队 ID 动态生成多个独立按钮  如何在 Pandas 中基于一列条件计算另一列的分组均值  高防服务器：AI智能防御DDoS攻击与数据安全保障  使用PHP下载CSS文件中的所有图片【几行代码即可实现】  高配服务器限时抢购：企业级配置与回收服务一站式优惠方案  Laravel怎么使用artisan命令缓存配置和视图  如何在阿里云虚拟机上搭建网站？步骤解析与避坑指南  EditPlus中的正则表达式 实战(2)  Win11怎么关闭专注助手 Win11关闭免打扰模式设置【操作】  智能起名网站制作软件有哪些,制作logo的软件？  悟空浏览器如何设置小说背景色_悟空浏览器背景色设置【方法】  如何在IIS中新建站点并配置端口与物理路径？  极客网站有哪些,DoNews、36氪、爱范儿、虎嗅、雷锋网、极客公园这些互联网媒体网站有什么差异？  在centOS 7安装mysql 5.7的详细教程  潮流网站制作头像软件下载,适合母子的网名有哪些？  Laravel如何实现事件和监听器？（Event & Listener实战）  浅谈redis在项目中的应用  Laravel任务队列怎么用_Laravel Queues异步处理任务提升应用性能