dnn: preserve name, type strings for ShuffleLayer

2024-11-25 03:30:34 +08:00 · 2018-07-11 08:19:23 +02:00 · 2018-07-11 08:19:23 +02:00 · a7b502f04a
commit a7b502f04a
parent 0bb2c115aa
1 changed files with 1 additions and 0 deletions
--- a/modules/dnn/src/layers/shuffle_channel_layer.cpp
+++ b/modules/dnn/src/layers/shuffle_channel_layer.cpp
@ -14,6 +14,7 @@ public:
    ShuffleChannelLayerImpl(const LayerParams& params)
    {
        group = params.get<int>("group", 1);
+        setParamsFrom(params);
    }

    bool getMemoryShapes(const std::vector<MatShape> &inputs,