Browse Source

clean code warnings

tags/v1.2.0
lihongkang 5 years ago
parent
commit
307ad95eb5
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      model_zoo/official/cv/yolov4/ascend310_infer/src/main.cc
  2. +1
    -1
      model_zoo/official/cv/yolov4/ascend310_infer/src/utils.cc

+ 1
- 1
model_zoo/official/cv/yolov4/ascend310_infer/src/main.cc View File

@@ -141,7 +141,7 @@ int main(int argc, char **argv) {
average += diff;
infer_cnt++;
}
average = average/infer_cnt;
average = average / infer_cnt;
std::stringstream timeCost;
timeCost << "NN inference cost average time: "<< average << "ms of infer_count " << infer_cnt << std::endl;
std::cout << "NN inference cost average time: "<< average << "ms of infer_count " << infer_cnt << std::endl;


+ 1
- 1
model_zoo/official/cv/yolov4/ascend310_infer/src/utils.cc View File

@@ -58,7 +58,7 @@ int WriteResult(const std::string& imageFile, const std::vector<MSTensor> &outpu
std::string fileName(imageFile, pos + 1);
fileName.replace(fileName.find('.'), fileName.size() - fileName.find('.'), '_' + std::to_string(i) + ".bin");
std::string outFileName = homePath + "/" + fileName;
FILE * outputFile = fopen(outFileName.c_str(), "wb");
FILE *outputFile = fopen(outFileName.c_str(), "wb");
fwrite(netOutput.get(), outputSize, sizeof(char), outputFile);

fclose(outputFile);


Loading…
Cancel
Save