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

dnn(tflite): fix wrong axis normalization

Browse Source
This commit is contained in:
Alexander Alekhin 2023-02-21 21:20:37 +00:00
parent 5a227352b4
commit 9eb5e39ff3
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/dnn/src/tflite/tflite_importer.cpp
View File

@ -497,9 +497,8 @@ void TFLiteImporter::parseConcat(const Operator& op, const std::string& opcode,
DataLayout inpLayout = layouts[op.inputs()->Get(0)]; DataLayout inpLayout = layouts[op.inputs()->Get(0)];
if (inpLayout == DATA_LAYOUT_NHWC) { if (inpLayout == DATA_LAYOUT_NHWC) {
// OpenCV works in NCHW data layout. So change the axis correspondingly. // OpenCV works in NCHW data layout. So change the axis correspondingly.
CV_Check(axis, -4 < axis && axis < 4, ""); axis = normalize_axis(axis, 4);
int remap[] = {0, 2, 3, 1}; static const int remap[] = {0, 2, 3, 1};
axis = axis > 0 ? axis : 4 + axis;
axis = remap[axis]; axis = remap[axis];
} }
layerParams.set("axis", axis); layerParams.set("axis", axis);