potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 15:23:05 +04:00
Code Packages Releases Wiki Activity

Fix conv/deconv/fc layers FLOPS computation

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/layers/convolution_layer.cpp
+2 -2
View File
@@ -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;