Using IE backend for normalize layer tests

pull/13774/head
Liubov Batanina 6 years ago
parent f67b197d49
commit b068d26fad
  1. 12
      modules/dnn/src/layers/normalize_bbox_layer.cpp

@ -65,16 +65,12 @@ public:
{ {
if (backendId == DNN_BACKEND_INFERENCE_ENGINE) if (backendId == DNN_BACKEND_INFERENCE_ENGINE)
{ {
if (pnorm != 2) if (!(pnorm == 2 || pnorm == 1)) {
return false; return false;
if (!blobs.empty()) }
return true; return preferableTarget == DNN_TARGET_MYRIAD ? !acrossSpatial : startAxis == 1;
if (preferableTarget == DNN_TARGET_MYRIAD)
return !acrossSpatial;
return startAxis == 1 && (!acrossSpatial || endAxis > 1);
} }
else return backendId == DNN_BACKEND_OPENCV;
return backendId == DNN_BACKEND_OPENCV;
} }
bool getMemoryShapes(const std::vector<MatShape> &inputs, bool getMemoryShapes(const std::vector<MatShape> &inputs,

Loading…
Cancel
Save