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

Fix conv/deconv/fc layers FLOPS computation

This commit is contained in:
Dmitry Kurtaev
2017-12-06 21:58:36 +03:00
parent ef04ca9e0f
commit ef0650179b
3 changed files with 4 additions and 3 deletions
+1
View File
@@ -87,6 +87,7 @@ public:
size_t weightsMemory = 0, blobsMemory = 0;
net.getMemoryConsumption(netInputShape, weightsMemory, blobsMemory);
int64 flops = net.getFLOPS(netInputShape);
CV_Assert(flops > 0);
net.forward(outputLayer); // warmup