paddledection中ppyoloe+导出模型时怎么把NCHW格式转化为NHWC

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

a andeluleidisi1 发布于2023-02

我在paddledection中训练了ppyoloe+

在ppyoloe+导出模型时怎么把NCHW格式转化为NHWC

全部评论(6)

宇宙物语

#2 回复于2023-02

用hook把所有NCHW的结构调换

andeluleidisi1

#3 回复于2023-02

宇宙物语 #2

用hook把所有NCHW的结构调换

那个hook是个什么东西？

宇宙物语

#4 回复于2023-02

register_forward_post_hook()。添加预处理

andeluleidisi1

#5 回复于2023-02

宇宙物语 #4

register_forward_post_hook()。添加预处理

我是在paddledection中训练的，我看不明白官方代码，所以才不知道加在哪里，你这个应该也是往源码里添加吧，所以比较难搞，你知不知道paddledection中导出的py文件的那个参数是管这个的吗

宇宙物语

#6 回复于2023-02

。。那我也不太清楚了。py文件应该是模型结构。

李长安

#7 回复于2023-02

大佬厉害呀