mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-29 13:47:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #22653 from WanliZhong:issue22597

Browse Source 
DNN-TF: let StridedSlice layer support const input
This commit is contained in:
Alexander Smorkalov 2022-10-21 17:51:00 +03:00 committed by GitHub
commit e4cd430710
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
modules/dnn/src/tensorflow/tf_importer.cpp
View File

@ -1706,6 +1706,19 @@ void TFImporter::parseStridedSlice(tensorflow::GraphDef& net, const tensorflow::
layerParams.set("begin", DictValue::arrayInt((int*)begins.data, begins.total()));
layerParams.set("end", DictValue::arrayInt((int*)ends.data, ends.total()));
Pin inp = parsePin(layer.input(0));
if (value_id.find(inp.name) != value_id.end())
{
// The input is constant.
LayerParams lp;
lp.name = inp.name;
lp.type = "Const";
lp.blobs.push_back(getTensorContent(getConstBlob(layer, value_id, 0)));
int constInpId = dstNet.addLayer(lp.name, lp.type, lp);
layer_id[lp.name] = constInpId;
}
int id = dstNet.addLayer(name, "Slice", layerParams);
layer_id[name] = id;