Commit 0be1bf55 authored by Alexander Alekhin's avatar Alexander Alekhin

Merge pull request #10244 from dkurt:issue_10239

parents ef04ca9e ef065017
......@@ -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
......
......@@ -1022,7 +1022,7 @@ public:
int64 flops = 0;
for (int i = 0; i < inputs.size(); i++)
{
flops += total(outputs[i])*(2*kernel.area()*inputs[i][1] + 1);
flops += total(outputs[i])*(CV_BIG_INT(2)*kernel.area()*inputs[i][1] + 1);
}
return flops;
......@@ -1440,7 +1440,7 @@ public:
for (int i = 0; i < inputs.size(); i++)
{
flops += 2*outChannels*kernel.area()*total(inputs[i]);
flops += CV_BIG_INT(2)*outChannels*kernel.area()*total(inputs[i]);
}
return flops;
......
......@@ -397,7 +397,7 @@ public:
int innerSize = blobs[0].size[1];
for(int i = 0; i < outputs.size(); i++)
{
flops += 3*innerSize*total(outputs[i]);
flops += CV_BIG_INT(3)*innerSize*total(outputs[i]);
}
return flops;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment