1
0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 23:33:05 +04:00

Set output layers names and types for models in DLDT's intermediate representation

This commit is contained in:
Dmitry Kurtaev
2018-06-28 09:09:11 +03:00
parent e4b51fa8ad
commit 346871e27f
4 changed files with 81 additions and 49 deletions
+4
View File
@@ -925,6 +925,10 @@ TEST(Layer_Test_Convolution_DLDT, Accuracy)
Mat out = net.forward();
normAssert(outDefault, out);
std::vector<int> outLayers = net.getUnconnectedOutLayers();
ASSERT_EQ(net.getLayer(outLayers[0])->name, "output_merge");
ASSERT_EQ(net.getLayer(outLayers[0])->type, "Concat");
}
// 1. Create a .prototxt file with the following network: