核心思想就是通过Yaml文件主体模块和可拔插的模块组成一个完整pipline.

train.py流程解析

初始化训练参数

        1 parser=ArgsParser() #读取命令行传递参数加载yaml文件参数

        2 整合参数检查参数配置是否正确

        3 检查是否使用GPU加速

        4 检查paddledet版本是否正确

        5 进入run()函数

                配置阶段

                        系统变量配置、初始化、得到GPU数量

                        创建数据读取

                        创建网络结构

                        创建学习率类

                        创建优化器类

                        初始化模型权重加载训练模型模型优化整合

                        是否多卡并行训练

                开始训练

                        遍历数据,开始循环训练,根据时间计算时间

                        模型前向推理反向传播

                        每一个iter结束输出日志

                        打印log

eval.py流程解析

初始化训练参数:

        parser=ArgsParser() #读取命令行传递参数,加载yaml文件参数

        参数整合,检查参数配置是否正确

        使用GPU加速

        查看paddledet版本是否正确

        进入run()函数

                配置阶段

                        创建网络结构

                        初始化模型权重加载训练模型

                        创建数据读取

                开始评估

                        遍历数据,开始前向推理,收集结

                        选择Metric评估标准

                        输出日志

infer.py流程解析

初始化训练参数:

        parser=ArgParser() #读取命令行传递参数,加载yaml文件参数

        参数整合,检查参数配置是否正确

        GPU使用是否正确

        查看paddledet版本是否正确

        进入run()函数

                配置阶段:

                        创建网络结构

                        初始化模型权重,加载预训练模型

                        初始化模型权重,加载预训练模型

                        创建数据读取类

                        选择推理标准

                开启推理
                        遍历数据,开始前向推理,收集结

                        保存推理结果

                        使用visualdl保存图片结果

export_model.py流程解析

        初始化训练参数

                设置cpu环境执行parser=ArgsParser() #读取命令行传递参数,加载yaml文件参数

                BN参数转换推理参数,将参数整合,检查参数配置

                GPU加速

                查看paddle版本

                进入run()函数

                        配置阶段

                                创建网络结构

                                 初始化模型权重

                                转为静态图,保存模型

                        

参考PaddleDetection整体结构概述_无情铁铲的博客-CSDN博客

原文地址:https://blog.csdn.net/qq_40107571/article/details/134646463

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任

如若转载,请注明出处:http://www.7code.cn/show_36220.html

如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱suwngjj01@126.com进行投诉反馈,一经查实,立即删除

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注