xiaying
69dba73dc7
Main Feature: 1. Add OpenCV API and Numpy API Support 2. Protobuf move into MNN 3. Add more op for torchscript convert 4. Add recompute to speed up geometry compute 5. Add ModuleBasic Test
130 lines
3.3 KiB
C++
130 lines
3.3 KiB
C++
//
|
|
// checkDir.cpp
|
|
// MNN
|
|
//
|
|
// Created by MNN on 2019/01/22.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <cmath>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#if defined(_MSC_VER)
|
|
#include <Windows.h>
|
|
#undef min
|
|
#undef max
|
|
#else
|
|
#include <dirent.h>
|
|
#include <sys/stat.h>
|
|
#endif
|
|
|
|
using namespace std;
|
|
|
|
#define NONE "\e[0m"
|
|
#define RED "\e[0;31m"
|
|
#define GREEN "\e[0;32m"
|
|
#define L_GREEN "\e[1;32m"
|
|
#define BLUE "\e[0;34m"
|
|
#define L_BLUE "\e[1;34m"
|
|
#define BOLD "\e[1m"
|
|
|
|
int main(int argc, char* argv[]) {
|
|
if (argc < 3) {
|
|
printf("Usage: ./checkDir.out outputDir1 outputDir2 thredhold [order]\n");
|
|
return 0;
|
|
}
|
|
|
|
// read args
|
|
printf("Compare:\n");
|
|
printf("%s\n", argv[1]);
|
|
printf("%s\n", argv[2]);
|
|
float tolerance = 0.001;
|
|
if (argc > 3) {
|
|
tolerance = stof(string(argv[3]));
|
|
}
|
|
printf("tolerance=%f\n", tolerance);
|
|
|
|
std::vector<std::string> compareFiles;
|
|
if (argc > 4) {
|
|
printf("Order file is %s\n", argv[4]);
|
|
std::ifstream orderOs(argv[4]);
|
|
std::string stringLine;
|
|
while (getline(orderOs, stringLine, '\n')) {
|
|
compareFiles.emplace_back(stringLine);
|
|
}
|
|
} else {
|
|
#if defined(_MSC_VER)
|
|
WIN32_FIND_DATA ffd;
|
|
HANDLE hFind = INVALID_HANDLE_VALUE;
|
|
hFind = FindFirstFile(argv[1], &ffd);
|
|
if (INVALID_HANDLE_VALUE == hFind) {
|
|
printf("Error to open %s\n", argv[1]);
|
|
return 0;
|
|
}
|
|
do {
|
|
if(INVALID_FILE_ATTRIBUTES != GetFileAttributes(ffd.cFileName) && GetLastError() != ERROR_FILE_NOT_FOUND) {
|
|
compareFiles.push_back(ffd.cFileName);
|
|
}
|
|
} while (FindNextFile(hFind, &ffd) != 0);
|
|
FindClose(hFind);
|
|
#else
|
|
// open dir
|
|
DIR* root = opendir(argv[1]);
|
|
if (NULL == root) {
|
|
printf("Error to open %s\n", argv[1]);
|
|
return 0;
|
|
}
|
|
struct dirent* ent;
|
|
while ((ent = readdir(root)) != NULL) {
|
|
compareFiles.push_back(ent->d_name);
|
|
}
|
|
closedir(root);
|
|
#endif
|
|
}
|
|
|
|
// compare files
|
|
for (int i=0; i<compareFiles.size(); ++i) {
|
|
auto& s = compareFiles[i];
|
|
if (s.size() <= 2) {
|
|
continue;
|
|
}
|
|
std::string empty = "";
|
|
std::string oldFile = empty + argv[1] + "/" + s;
|
|
std::string newFile = empty + argv[2] + "/" + s;
|
|
std::ifstream oldOs(oldFile.c_str());
|
|
if (oldOs.fail()) {
|
|
// printf("Can's open %s\n", oldFile.c_str());
|
|
continue;
|
|
}
|
|
std::ifstream newOs(newFile.c_str());
|
|
if (newOs.fail()) {
|
|
// printf("Can's open %s\n", newFile.c_str());
|
|
continue;
|
|
}
|
|
|
|
int pos = 0;
|
|
bool correct = true;
|
|
float v1, v2;
|
|
while (oldOs >> v1) {
|
|
auto& valid = newOs >> v2;
|
|
auto absError = fabsf(v1 - v2);
|
|
if (absError <= tolerance && !isnan(absError)) {
|
|
pos++;
|
|
continue;
|
|
}
|
|
printf("Error for %d, %s, %d, v1=%.6f, v2=%.6f\n", i, s.c_str(), pos, v1, v2);
|
|
correct = false;
|
|
break;
|
|
}
|
|
if (correct) {
|
|
printf("Correct %d: %s\n", i, s.c_str());
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|