Function load_seg_data_from_folder

lesson6-Segmentation/util.cpp:28–63

遍历该目录下的.xml文件，并且找到对应的

//遍历该目录下的.xml文件，并且找到对应的
void load_seg_data_from_folder(std::string folder, std::string image_type,
                               std::vector<std::string> &list_images, std::vector<std::string> &list_labels)
{
    long long hFile = 0; //句柄
    struct _finddata_t fileInfo;
    std::string pathName;
    if ((hFile = _findfirst(pathName.assign(folder).append("\\*.*").c_str(), &fileInfo)) == -1)
    {
        return;
    }
    do
    {
        const char* s = fileInfo.name;

        if (fileInfo.attrib&_A_SUBDIR) //是子文件夹
        {
            //遍历子文件夹中的文件(夹)，查看是否有txt文件
            if (strcmp(s, ".") == 0 || strcmp(s, "..") == 0) //子文件夹目录是.或者..
                continue;
            std::string sub_path = folder + "\\" + fileInfo.name;
            load_seg_data_from_folder(sub_path, image_type, list_images, list_labels);
        }
        else //判断是不是txt文件
        {

            if (strstr(s, ".json"))
            {
                std::string label_path = folder + "\\" + fileInfo.name;
                list_labels.push_back(label_path);
                std::string image_path = replace_all_distinct(label_path, ".json", image_type);
                list_images.push_back(image_path);
            }
        }
    } while (_findnext(hFile, &fileInfo) == 0);
    return;
}