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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user