3.6_训练与测试

文档摘要

3.6、训练与测试 3.6.1 模型训练前面的章节，我们已经对目标检测训练的各个重要的知识点进行了讲解，下面我们需要将整个流程串起来，对模型进行训练。目标检测网络的训练大致是如下的流程：设置各种超参数定义数据加载模块 dataloader 定义网络 model 定义损失函数 loss 定义优化器 optimizer 遍历训练数据，预测-计算loss-反向传播首先，我们导入必要的库，然后设定各种超参数按照上面梳理的流程，编写训练代码如下：其中，我们对单个epoch的训练逻辑进行了封装，其具体实现如下：完成了代码的编写后，我们就可以开始训练模型了，训练过程类似下图所示：剩下的就是等待了～ 3.6.2 后处理 3.6.2.