diff options
| author | Dmitry Kurtaev <dmitry.kurtaev+github@gmail.com> | 2018-04-20 16:44:05 +0300 |
|---|---|---|
| committer | Dmitry Kurtaev <dmitry.kurtaev+github@gmail.com> | 2018-04-20 16:51:38 +0300 |
| commit | d959d7b9f05e99a6d4c3deec485a59120eac5a35 (patch) | |
| tree | 88b5ad3a1c6b167e0481a9498d0c0cb46f5a4a3b /modules/dnn/src/tensorflow/tf_importer.cpp | |
| parent | 7ea5029ae506291ce384d6f429536fa2fb354689 (diff) | |
| download | opencv-d959d7b9f05e99a6d4c3deec485a59120eac5a35.tar.gz opencv-d959d7b9f05e99a6d4c3deec485a59120eac5a35.tar.bz2 opencv-d959d7b9f05e99a6d4c3deec485a59120eac5a35.zip | |
Fuse deconvolution layer subgraphs from Keras
Diffstat (limited to 'modules/dnn/src/tensorflow/tf_importer.cpp')
| -rw-r--r-- | modules/dnn/src/tensorflow/tf_importer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/dnn/src/tensorflow/tf_importer.cpp b/modules/dnn/src/tensorflow/tf_importer.cpp index f5809163bd..a401f715be 100644 --- a/modules/dnn/src/tensorflow/tf_importer.cpp +++ b/modules/dnn/src/tensorflow/tf_importer.cpp @@ -1303,8 +1303,8 @@ void TFImporter::populateNet(Net dstNet) const int strideY = layerParams.get<int>("stride_h"); const int strideX = layerParams.get<int>("stride_w"); Mat outShape = getTensorContent(getConstBlob(layer, value_id, 0)); - const int outH = outShape.at<int>(2); - const int outW = outShape.at<int>(1); + const int outH = outShape.at<int>(1); + const int outW = outShape.at<int>(2); if (layerParams.get<String>("pad_mode") == "SAME") { layerParams.set("adj_w", (outW - 1) % strideX); |