Function ImportDataSet

tools/quantization/quantize.cc:125–154

}

int ImportDataSet(DataSet& dataset, std::string folder_path) {
    dataset.file_list.clear();

    DIR* dp;
    struct dirent* dirp;
    if ((dp = opendir(folder_path.c_str())) == NULL) {
        printf("Can't open %s\n", folder_path.c_str());
        return -1;
    }
    while ((dirp = readdir(dp)) != NULL) {
        if (dirp->d_type == DT_REG) {
            std::string file_name = dirp->d_name;
            FileFormat format     = NOTSUPPORT;
            if (GetInputType(file_name, format)) {
                char full_name[256 + 1];
                snprintf(full_name, 256, "%s/%s", folder_path.c_str(), dirp->d_name);
                dataset.file_list.push_back(std::make_pair(full_name, format));
                printf("\timport: %s  type: %d\n", dirp->d_name, format);
            }
        }
    }
    closedir(dp);

    if (dataset.file_list.size() == 0) {
        printf("no valid input file found!\n");
        return -1;
    }
    printf("import total %lu files\n", dataset.file_list.size());
    return 0;
}

bool CheckNumberString(std::string num_str) {
    const char* num_char = num_str.c_str();