mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-07-24 14:06:27 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

support broadcast on axis > 1 for Expand

Browse Source
This commit is contained in:
fengyuentau 2023-04-14 15:52:27 +08:00
parent f5a92cb43f
commit 88cacd35c5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/dnn/src/onnx/onnx_importer.cpp
View File

@ -2529,8 +2529,11 @@ void ONNXImporter::parseExpand(LayerParams& layerParams, const opencv_onnx::Node
node_proto.set_input(0, constParams.name);
node_proto.set_input(1, srcName);
}
else if (broadcast_axes.size() == 1 && broadcast_axes[0] <= 1)
else if (broadcast_axes.size() == 1)
{
// FIXME: this will end up creating massive amount of Identity nodes for broadcasting,
// for example, broadcast 1 to 256 needs 256 Identity nodes and 1 Concat node.
// Possible improvement is to use "Scale".
expandMid(layerParams.name, node_proto, srcName, targetShape[broadcast_axes[0]]);
layerParams.set("axis", broadcast_axes[0]);