mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-12 20:42:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #20674 from rogday:prelu_slope

Browse Source 
Fix PReLU negative slope access pattern

* fix prelu negative slope access pattern

* change begin() to ptr()
This commit is contained in:
rogday 2021-09-10 14:07:16 +03:00 committed by GitHub
parent ac0fd6aa9a
commit d31b93b513
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dnn/src/layers/elementwise_layers.cpp
View File

@ -1542,7 +1542,7 @@ Ptr<Layer> ChannelsPReLULayer::create(const LayerParams& params)
if (params.blobs[0].total() == 1) if (params.blobs[0].total() == 1)
{ {
LayerParams reluParams = params; LayerParams reluParams = params;
reluParams.set("negative_slope", params.blobs[0].at<float>(0)); reluParams.set("negative_slope", *params.blobs[0].ptr<float>());
return ReLULayer::create(reluParams); return ReLULayer::create(reluParams);
} }
Ptr<ChannelsPReLULayer> l(new ElementWiseLayer<ChannelsPReLUFunctor>(ChannelsPReLUFunctor(params.blobs[0]))); Ptr<ChannelsPReLULayer> l(new ElementWiseLayer<ChannelsPReLUFunctor>(ChannelsPReLUFunctor(params.blobs[0])));

5
modules/dnn/test/test_onnx_importer.cpp
View File

@ -250,6 +250,11 @@ TEST_P(Test_ONNX_layers, ReLU)
testONNXModels("ReLU"); testONNXModels("ReLU");
} }
TEST_P(Test_ONNX_layers, PReLU)
{
testONNXModels("PReLU_slope");
}
TEST_P(Test_ONNX_layers, Clip) TEST_P(Test_ONNX_layers, Clip)
{ {
testONNXModels("clip", npy); testONNXModels("clip", npy);