mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-18 16:11:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #12025 from Triplesalt:tfimport-relu

Browse Source
This commit is contained in:
Alexander Alekhin 2018-07-26 15:08:05 +00:00
commit c37d1a53b5
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/dnn/src/tensorflow/tf_importer.cpp
View File

@ -1293,7 +1293,13 @@ void TFImporter::populateNet(Net dstNet)
if (!next_layers.empty()) if (!next_layers.empty())
{ {
int maximumLayerIdx = next_layers[0].second; int maximumLayerIdx = next_layers[0].second;
ExcludeLayer(net, maximumLayerIdx, 0, false);
CV_Assert(net.node(maximumLayerIdx).input_size() == 2);
// The input from the Mul layer can also be at index 1.
int mulInputIdx = (net.node(maximumLayerIdx).input(0) == name) ? 0 : 1;
ExcludeLayer(net, maximumLayerIdx, mulInputIdx, false);
layers_to_ignore.insert(next_layers[0].first); layers_to_ignore.insert(next_layers[0].first);
layerParams.set("negative_slope", scaleMat.at<float>(0)); layerParams.set("negative_slope", scaleMat.at<float>(0));

7
modules/dnn/test/test_tf_importer.cpp
View File

@ -230,6 +230,13 @@ TEST_P(Test_TensorFlow_layers, flatten)
runTensorFlowNet("unfused_flatten_unknown_batch"); runTensorFlowNet("unfused_flatten_unknown_batch");
} }
TEST_P(Test_TensorFlow_layers, leaky_relu)
{
runTensorFlowNet("leaky_relu_order1");
runTensorFlowNet("leaky_relu_order2");
runTensorFlowNet("leaky_relu_order3");
}
TEST_P(Test_TensorFlow_layers, l2_normalize) TEST_P(Test_TensorFlow_layers, l2_normalize)
{ {
runTensorFlowNet("l2_normalize"); runTensorFlowNet("l2_normalize");