mirror of https://github.com/opencv/opencv.git
Merge pull request #14792 from dkurt:dnn_ie_min_version_r5
* Remove Inference Engine 2018R3 and 2018R4 * Fix 2018R5pull/14747/head
Dmitry Kurtaev
6 years ago
committed by
Alexander Alekhin
parent
54a14c877c
commit
eba696a41e
35 changed files with 216 additions and 1608 deletions
@ -1,205 +0,0 @@ |
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2013, OpenCV Foundation, all rights reserved.
|
||||
// Copyright (C) 2017, Intel Corporation, all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#include "../precomp.hpp" |
||||
#include "../op_inf_engine.hpp" |
||||
#include "layers_common.hpp" |
||||
|
||||
namespace cv |
||||
{ |
||||
namespace dnn |
||||
{ |
||||
|
||||
class CropLayerImpl CV_FINAL : public CropLayer |
||||
{ |
||||
public: |
||||
CropLayerImpl(const LayerParams& params) |
||||
{ |
||||
setParamsFrom(params); |
||||
startAxis = params.get<int>("axis", 2); |
||||
const DictValue *paramOffset = params.ptr("offset"); |
||||
|
||||
if (paramOffset) |
||||
{ |
||||
for (int i = 0; i < paramOffset->size(); i++) |
||||
offset.push_back(paramOffset->get<int>(i)); |
||||
} |
||||
} |
||||
|
||||
virtual bool supportBackend(int backendId) CV_OVERRIDE |
||||
{ |
||||
#ifdef HAVE_INF_ENGINE |
||||
if (backendId == DNN_BACKEND_INFERENCE_ENGINE) |
||||
return INF_ENGINE_VER_MAJOR_LT(INF_ENGINE_RELEASE_2018R5) && crop_ranges.size() == 4; |
||||
else |
||||
#endif |
||||
return backendId == DNN_BACKEND_OPENCV; |
||||
} |
||||
|
||||
bool getMemoryShapes(const std::vector<MatShape> &inputs, |
||||
const int requiredOutputs, |
||||
std::vector<MatShape> &outputs, |
||||
std::vector<MatShape> &internals) const CV_OVERRIDE |
||||
{ |
||||
CV_Assert(inputs.size() == 2); |
||||
|
||||
MatShape dstShape = inputs[0]; |
||||
int start = clamp(startAxis, dstShape); |
||||
for (int i = start; i < dstShape.size(); i++) |
||||
{ |
||||
dstShape[i] = inputs[1][i]; |
||||
} |
||||
|
||||
outputs.resize(1, dstShape); |
||||
|
||||
return false; |
||||
} |
||||
|
||||
void finalize(InputArrayOfArrays inputs_arr, OutputArrayOfArrays) CV_OVERRIDE |
||||
{ |
||||
std::vector<Mat> inputs; |
||||
inputs_arr.getMatVector(inputs); |
||||
CV_Assert(2 == inputs.size()); |
||||
|
||||
const Mat &inpBlob = inputs[0]; |
||||
const Mat &inpSzBlob = inputs[1]; |
||||
|
||||
int dims = inpBlob.dims; |
||||
int start_axis = clamp(startAxis, dims); |
||||
|
||||
std::vector<int> offset_final(dims, 0); |
||||
if (offset.size() == 1) |
||||
{ |
||||
for (int i = start_axis; i < dims; i++) |
||||
offset_final[i] = offset[0]; |
||||
} |
||||
else if (offset.size() > 1) |
||||
{ |
||||
if ((int)offset.size() != dims - start_axis) |
||||
CV_Error(Error::StsBadArg, "number of offset values specified must be " |
||||
"equal to the number of dimensions following axis."); |
||||
|
||||
for (int i = start_axis; i < dims; i++) |
||||
offset_final[i] = offset[i - start_axis]; |
||||
} |
||||
|
||||
crop_ranges.resize(dims); |
||||
for (int i = 0; i < start_axis; i++) |
||||
{ |
||||
crop_ranges[i] = Range(0, inpBlob.size[i]); |
||||
} |
||||
for (int i = start_axis; i < dims; i++) |
||||
{ |
||||
if (offset_final[i] < 0 || offset_final[i] + inpSzBlob.size[i] > inpBlob.size[i]) |
||||
CV_Error(Error::StsBadArg, "invalid crop parameters or blob sizes"); |
||||
|
||||
crop_ranges[i] = Range(offset_final[i], offset_final[i] + inpSzBlob.size[i]); |
||||
} |
||||
} |
||||
|
||||
void forward(InputArrayOfArrays inputs_arr, OutputArrayOfArrays outputs_arr, OutputArrayOfArrays internals_arr) CV_OVERRIDE |
||||
{ |
||||
CV_TRACE_FUNCTION(); |
||||
CV_TRACE_ARG_VALUE(name, "name", name.c_str()); |
||||
|
||||
std::vector<Mat> inputs, outputs; |
||||
inputs_arr.getMatVector(inputs); |
||||
outputs_arr.getMatVector(outputs); |
||||
|
||||
Mat &input = inputs[0]; |
||||
input(&crop_ranges[0]).copyTo(outputs[0]); |
||||
} |
||||
|
||||
#ifdef HAVE_INF_ENGINE |
||||
virtual Ptr<BackendNode> initInfEngine(const std::vector<Ptr<BackendWrapper> >&) CV_OVERRIDE |
||||
{ |
||||
#if INF_ENGINE_VER_MAJOR_LT(INF_ENGINE_RELEASE_2018R5) |
||||
InferenceEngine::LayerParams lp; |
||||
lp.name = name; |
||||
lp.type = "Crop"; |
||||
lp.precision = InferenceEngine::Precision::FP32; |
||||
std::shared_ptr<InferenceEngine::CropLayer> ieLayer(new InferenceEngine::CropLayer(lp)); |
||||
|
||||
CV_Assert(crop_ranges.size() == 4); |
||||
|
||||
#if INF_ENGINE_VER_MAJOR_GT(INF_ENGINE_RELEASE_2018R3) |
||||
for (int i = 0; i < 4; ++i) |
||||
{ |
||||
ieLayer->axis.push_back(i); |
||||
ieLayer->offset.push_back(crop_ranges[i].start); |
||||
ieLayer->dim.push_back(crop_ranges[i].end - crop_ranges[i].start); |
||||
} |
||||
#else |
||||
ieLayer->axis.push_back(0); // batch
|
||||
ieLayer->offset.push_back(crop_ranges[0].start); |
||||
ieLayer->dim.push_back(crop_ranges[0].end - crop_ranges[0].start); |
||||
|
||||
ieLayer->axis.push_back(1); // channels
|
||||
ieLayer->offset.push_back(crop_ranges[1].start); |
||||
ieLayer->dim.push_back(crop_ranges[1].end - crop_ranges[1].start); |
||||
|
||||
ieLayer->axis.push_back(3); // height
|
||||
ieLayer->offset.push_back(crop_ranges[2].start); |
||||
ieLayer->dim.push_back(crop_ranges[2].end - crop_ranges[2].start); |
||||
|
||||
ieLayer->axis.push_back(2); // width
|
||||
ieLayer->offset.push_back(crop_ranges[3].start); |
||||
ieLayer->dim.push_back(crop_ranges[3].end - crop_ranges[3].start); |
||||
#endif |
||||
return Ptr<BackendNode>(new InfEngineBackendNode(ieLayer)); |
||||
#else |
||||
return Ptr<BackendNode>(); |
||||
#endif // IE < R5
|
||||
} |
||||
#endif |
||||
|
||||
std::vector<Range> crop_ranges; |
||||
}; |
||||
|
||||
|
||||
Ptr<CropLayer> CropLayer::create(const LayerParams& params) |
||||
{ |
||||
return Ptr<CropLayer>(new CropLayerImpl(params)); |
||||
} |
||||
|
||||
} |
||||
} |
||||
Loading…
Reference in new issue