Fix bug that input width and height are reversed

JDAI-CV · Jun 1, 2019 · 4d918b9 · 4d918b9
1 parent 276f2dd
commit 4d918b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dnnlibrary/src/ModelBuilder.cpp b/dnnlibrary/src/ModelBuilder.cpp
@@ -40,7 +40,7 @@ ModelBuilder::Index ModelBuilder::AddInput(string name, const uint32_t batch,
                                            const uint32_t height,
                                            const uint32_t width,
                                            const uint32_t depth) {
-    const vector<uint32_t> dimen{batch, width, height, depth};
+    const vector<uint32_t> dimen{batch, height, width, depth};
     return AddInput(name, {Type::TENSOR_FLOAT32, dimen});
 }