首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
pool2d专题
【pytorch】——ONNX export failed: Couldn‘t export operator aten::adaptive_avg_pool2d
问题: 语义分割网络pspnet导出成onnx的时候报错 分析: 主要是ppm中使用了AdaptiveAvgPool2d 解决方案: onnx中最新的版本好像也是没有AdaptiveAvgPool2d的因为pytorch是动态尺寸的,所以有AdaptiveAvgPool2d,首先要固定尺寸进行推理将AdaptiveAvgPool2d替换成AvgPool2d 参考自:https://www.c
阅读更多...
PyTorch 的 Pooling 和 UnPooling函数中的 indices 参数:nn.MaxPool2d/nn.MaxUnpool2d、F.max_pool2d/F.max_unpool2d
这篇博文主要介绍 PyTorch 的 MaxPooling 和 MAxUnPooling 函数中涉及到的 indices 参数。 indices 是“索引”的意思,对于一些结构对称的网络模型,上采样和下采样的结构往往是对称的,我们可以在下采样做 MaxPooling 的时候记录下来最大值所在的位置,当做上采样的时候把最大值还原到其对应的位置,然后其余的位置补 0 。 indices 参数的作
阅读更多...