1
0
mirror of https://github.com/opencv/opencv.git synced 2026-07-31 00:03:03 +04:00

LSTM from ONNX works

This commit is contained in:
Dmitry Kurtaev
2020-03-15 23:21:58 +03:00
parent 14da5ec311
commit 8d69dbdf49
5 changed files with 66 additions and 168 deletions
-6
View File
@@ -79,12 +79,6 @@ public:
netSoftmax.setInput(ref);
ref = netSoftmax.forward();
}
std::cout << "ref: " << ref.size << '\n';
std::cout << "out: " << out.size << '\n';
std::cout << ref.reshape(1, 1) << '\n';
std::cout << '\n';
std::cout << out.reshape(1, 1) << '\n';
normAssert(ref, out, "", l1 ? l1 : default_l1, lInf ? lInf : default_lInf);
if (checkNoFallbacks)
expectNoFallbacksFromIE(net);