potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 15:23:05 +04:00
Code Packages Releases Wiki Activity

Merge pull request #28018 from satyam102006:fix/dnn-torchimporter-pad-allocation

Browse Source 
Update torch_importer.cpp
This commit is contained in:
Alexander Smorkalov
2025-11-16 18:47:33 +03:00
committed by GitHub
parent 723bcb9a7b 27d106d574
commit 42cf71285e
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/torch/torch_importer.cpp
+12 -6
View File
@@ -782,12 +782,16 @@ struct TorchImporter
int dim = scalarParams.get<int>("dim") - 1; // In Lua we start from 1.
int pad = scalarParams.get<int>("pad");
std::vector<int> paddings((dim + 1) * 2, 0);
AutoBuffer<int> paddingsBuf((dim + 1) * 2);
int *paddings = paddingsBuf.data();
int paddingsSize = (dim + 1) * 2;
for (int i = 0; i < paddingsSize; ++i) paddings[i] = 0;
if (pad > 0)
paddings[dim * 2 + 1] = pad; // Pad after (right).
paddings[dim * 2 + 1] = pad;
else
paddings[dim * 2] = -pad; // Pad before (left).
layerParams.set("paddings", DictValue::arrayInt<int*>(&paddings[0], paddings.size()));
paddings[dim * 2] = -pad;
layerParams.set("paddings", DictValue::arrayInt<int*>(paddings, paddingsSize));
curModule->modules.push_back(newModule);
}
@@ -935,12 +939,14 @@ struct TorchImporter
// Torch's SpatialZeroPadding works with 3- or 4-dimensional input.
// So we add parameter input_dims=3 to ignore batch dimension if it will be.
std::vector<int> paddings(6, 0); // CHW
int paddings[6] = {0, 0, 0, 0, 0, 0};
paddings[2] = padTop;
paddings[3] = padBottom;
paddings[4] = padLeft;
paddings[5] = padRight;
layerParams.set("paddings", DictValue::arrayInt<int*>(&paddings[0], paddings.size()));
layerParams.set("paddings", DictValue::arrayInt(paddings, 6));
layerParams.set("input_dims", 3);
if (nnName == "SpatialReflectionPadding")