Skip to content

Commit 09d2ca1

Browse files
Merge pull request opencv#19449 from SamFC10:tf-leakyrelu
* add LeakyRelu to tf-importer * fix whitespace error * added test for leakyrelu
1 parent 863ecde commit 09d2ca1

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

modules/dnn/src/tensorflow/tf_importer.cpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2414,6 +2414,16 @@ void TFImporter::parseNode(const tensorflow::NodeDef& layer_)
24142414

24152415
connect(layer_id, dstNet, parsePin(layer.input(0)), id, 0);
24162416
}
2417+
else if (type == "LeakyRelu")
2418+
{
2419+
CV_CheckGT(num_inputs, 0, "");
2420+
CV_Assert(hasLayerAttr(layer, "alpha"));
2421+
layerParams.set("negative_slope", getLayerAttr(layer, "alpha").f());
2422+
2423+
int id = dstNet.addLayer(name, "ReLU", layerParams);
2424+
layer_id[name] = id;
2425+
connectToAllBlobs(layer_id, dstNet, parsePin(layer.input(0)), id, num_inputs);
2426+
}
24172427
else if (type == "Abs" || type == "Tanh" || type == "Sigmoid" ||
24182428
type == "Relu" || type == "Elu" ||
24192429
type == "Identity" || type == "Relu6")

modules/dnn/test/test_tf_importer.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -463,6 +463,7 @@ TEST_P(Test_TensorFlow_layers, leaky_relu)
463463
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && target == DNN_TARGET_OPENCL)
464464
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_OPENCL, CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_VERSION);
465465
#endif
466+
runTensorFlowNet("leaky_relu");
466467
runTensorFlowNet("leaky_relu_order1");
467468
runTensorFlowNet("leaky_relu_order2");
468469
runTensorFlowNet("leaky_relu_order3");

0 commit comments

Comments
 (0)