Enable SSD models from TensorFlow with OpenCL plugin of Intel's Inference Engine

This commit is contained in:
Dmitry Kurtaev
2018-06-08 16:55:21 +03:00
parent 1187a7fa34
commit 40765c5f8d
4 changed files with 75 additions and 44 deletions
+1
View File
@@ -877,6 +877,7 @@ TEST_P(Layer_Test_DWconv_Prelu, Accuracy)
int shape[] = {1, num_input, 16, 16};
Mat in_blob(4, &shape[0], CV_32FC1, Scalar(1));
net.setPreferableBackend(DNN_BACKEND_OPENCV);
net.setInput(in_blob);
Mat out = net.forward();