Browse Source

fix alerts

tags/20210507
nihuini 5 years ago
parent
commit
97903bbe85
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      tools/modelwriter.h

+ 5
- 5
tools/modelwriter.h View File

@@ -794,7 +794,7 @@ int ModelWriter::save(const char* parampath, const char* binpath)
int outh = blobs[layer->tops[0]].shape.h;
int outc = blobs[layer->tops[0]].shape.c;

mac += op->kernel_h * op->kernel_w * outw * outh * outc * inc;
mac += (uint64_t)op->kernel_h * op->kernel_w * outw * outh * outc * inc;
}
}
else if (layer->type == "ConvolutionDepthWise")
@@ -855,7 +855,7 @@ int ModelWriter::save(const char* parampath, const char* binpath)
int outh = blobs[layer->tops[0]].shape.h;
int outc = blobs[layer->tops[0]].shape.c;

mac += op->kernel_h * op->kernel_w * outw * outh * (outc / op->group) * (inc / op->group) * op->group;
mac += (uint64_t)op->kernel_h * op->kernel_w * outw * outh * (outc / op->group) * (inc / op->group) * op->group;
}
}
else if (layer->type == "Crop")
@@ -935,7 +935,7 @@ int ModelWriter::save(const char* parampath, const char* binpath)
int inc = blobs[layer->bottoms[0]].shape.c;
int outc = blobs[layer->tops[0]].shape.c;

mac += op->kernel_h * op->kernel_w * inw * inh * outc * inc;
mac += (uint64_t)op->kernel_h * op->kernel_w * inw * inh * outc * inc;
}
}
else if (layer->type == "DeconvolutionDepthWise")
@@ -992,7 +992,7 @@ int ModelWriter::save(const char* parampath, const char* binpath)
int inc = blobs[layer->bottoms[0]].shape.c;
int outc = blobs[layer->tops[0]].shape.c;

mac += op->kernel_h * op->kernel_w * inw * inh * (outc / op->group) * (inc / op->group) * op->group;
mac += (uint64_t)op->kernel_h * op->kernel_w * inw * inh * (outc / op->group) * (inc / op->group) * op->group;
}
}
else if (layer->type == "DetectionOutput")
@@ -1140,7 +1140,7 @@ int ModelWriter::save(const char* parampath, const char* binpath)
int inc = blobs[layer->bottoms[0]].shape.c;
int outw = blobs[layer->tops[0]].shape.w;

mac += inw * inh * inc * outw;
mac += (uint64_t)inw * inh * inc * outw;
}
}
else if (layer->type == "Input")


Loading…
Cancel
Save