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:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user