创建自己的重参数class

PaddlePaddle / PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.

Apache License 2.0

12.38k stars 2.84k forks source link

创建自己的重参数class #6004

Closed sdreamforchen closed 2 years ago

sdreamforchen commented 2 years ago

问题确认 Search before asking

[X] 我已经搜索过问题，但是没有找到解答。I have searched the question and found no related answer.

请提出你的问题 Please ask your question

ppyoloe的convert_to_deploy这个函数是在什么条件下调用呢？我如果新写一个重参数的class。需要怎么修改才能得到正确的结果呢

yghstill commented 2 years ago

@sdreamforchen 在模型导出时调用，https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.4/ppdet/engine/trainer.py#L708 你可以在模型组网中新增相同命名的convert_to_deploy函数即可。

sdreamforchen commented 2 years ago

@sdreamforchen 在模型导出时调用，https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.4/ppdet/engine/trainer.py#L708 你可以在模型组网中新增相同命名的convert_to_deploy函数即可。

感谢