按vector＜string＞中指定位置的数字大小重新排序，解决使用视频生成一堆图像，然后重新读取这堆图像时，图像的顺序不对的问题

问题描述：

本人根据一个视频生成了一堆图像，图像名中只有数字，如0.jpg。然后本人想要按照视频顺序重新读取图像，计算两张图像中的相机运动。

重新读取图像时，使用vector<string>存储图像名，但是vector<string>中的顺序类似为：

本文解决这种类似的排序问题。

经常出现的场景就是使用一个视频在一个文件夹下生成一堆图像，然后重新读取这堆图像，此时图像的顺序并不是按照生成图像的顺序

方法：

步骤1:

//获取文件夹下的全部文件名
void getImageNameList(std::string folder_name, std::vector<std::string> &imagenamelist)
{
    struct __finddata64_t filefind;
    std::string curr = folder_name + "\\*.*";

    __int64 handle = _findfirst64(curr.c_str(), &filefind);
    if (handle == -1) return;

    //成功返回0，否则返回-1
    while (0 == _findnext64(handle, &filefind))
    {
        if (0 == std::strcmp(filefind.name, ".."))
        {
            continue;
        }

        std::string subfolder = folder_name + "\\" + filefind.name;
        imagenamelist.push_back(subfolder);
    }

    _findclose(handle);
}

注意：如果文件夹下还有文件夹则本段代码并不适用

步骤2：

//按指定字符拆分字符串
std::vector<std::string> splitString(std::string str, std::string pattern)
{
    std::string::size_type pos;
    std::vector<std::string> result;
    str += pattern;        //扩展字符串以方便操作
    int size = str.size();
    for (int i = 0; i < size; i++)
    {
        pos = str.find(pattern, i);
        if (pos < size)
        {
            std::string s = str.substr(i, pos - i);
            result.push_back(s);
            i = pos + pattern.size() - 1;
        }
    }
    return result;
}

注意：这个是从网上搜的，按照指定字符拆分字符串，有类似需求的可以拿去用

步骤3：

//排除掉图像路径和图像后缀，只保留纯图像名
std::string getImagePureName(std::string imagename, std::string pattern1, int index1, std::string pattern2, int index2)
{
    std::vector<std::string> result1 = splitString(imagename, pattern1);
    std::vector<std::string> result2 = splitString(result1[index1], pattern2);
    return result2[index2];
}

注意：将字符串拆分，第一次把图像路径拿掉只保留图像名“0.jpg”，第二次把图像后缀去掉“0”

步骤4：

//排序全部文件名
std::vector<std::string> sortImageNameList(std::vector<std::string> imagenamelist) 
{
    std::vector<std::string> newlist;
    for (int i = 0; i < imagenamelist.size(); i++) 
    {
        if (i == 0) 
        {
            newlist.push_back(imagenamelist[i]);
        }
        else 
        {
            bool insertInEnd = true;

            for (int j = 0; j < newlist.size(); j++) 
            {
                std::string srcname = getImagePureName(imagenamelist[i], "\\", 3, ".",0);
                int srcnumber = std::atoi(srcname.c_str());
                std::string dstName = getImagePureName(newlist[j], "\\", 3, ".", 0);
                int dstnumber = std::atoi(dstName.c_str());
                if (srcnumber < dstnumber) 
                {
                    insertInEnd = false;
                    newlist.insert(newlist.begin() + j, imagenamelist[i]);
                    break;
                }
            }
            if (insertInEnd) newlist.push_back(imagenamelist[i]);

        }
    }

    return newlist;
}

注意：按照图像名的数字重新排序确定好你的图像路径合理地使用参数

 

完整代码：

#include <stdio.h>
#include <io.h>
#include <iostream>
#include <opencv2/core/core.hpp>

//获取文件夹下的全部文件名
void getImageNameList(std::string folder_name, std::vector<std::string> &imagenamelist)
{
    struct __finddata64_t filefind;
    std::string curr = folder_name + "\\*.*";

    __int64 handle = _findfirst64(curr.c_str(), &filefind);
    if (handle == -1) return;

    //成功返回0，否则返回-1
    while (0 == _findnext64(handle, &filefind))
    {
        if (0 == std::strcmp(filefind.name, ".."))
        {
            continue;
        }

        std::string subfolder = folder_name + "\\" + filefind.name;
        imagenamelist.push_back(subfolder);
    }

    _findclose(handle);
}

//按指定字符拆分字符串
std::vector<std::string> splitString(std::string str, std::string pattern)
{
    std::string::size_type pos;
    std::vector<std::string> result;
    str += pattern;        //扩展字符串以方便操作
    int size = str.size();
    for (int i = 0; i < size; i++)
    {
        pos = str.find(pattern, i);
        if (pos < size)
        {
            std::string s = str.substr(i, pos - i);
            result.push_back(s);
            i = pos + pattern.size() - 1;
        }
    }
    return result;
}

//排除掉图像路径和图像后缀，只保留纯图像名
std::string getImagePureName(std::string imagename, std::string pattern1, int index1, std::string pattern2, int index2)
{
    std::vector<std::string> result1 = splitString(imagename, pattern1);
    std::vector<std::string> result2 = splitString(result1[index1], pattern2);
    return result2[index2];
}

//排序全部文件名
std::vector<std::string> sortImageNameList(std::vector<std::string> imagenamelist) 
{
    std::vector<std::string> newlist;
    for (int i = 0; i < imagenamelist.size(); i++) 
    {
        if (i == 0) 
        {
            newlist.push_back(imagenamelist[i]);
        }
        else 
        {
            bool insertInEnd = true;

            for (int j = 0; j < newlist.size(); j++) 
            {
                std::string srcname = getImagePureName(imagenamelist[i], "\\", 3, ".",0);
                int srcnumber = std::atoi(srcname.c_str());
                std::string dstName = getImagePureName(newlist[j], "\\", 3, ".", 0);
                int dstnumber = std::atoi(dstName.c_str());
                if (srcnumber < dstnumber) 
                {
                    insertInEnd = false;
                    newlist.insert(newlist.begin() + j, imagenamelist[i]);
                    break;
                }
            }
            if (insertInEnd) newlist.push_back(imagenamelist[i]);

        }
    }

    return newlist;
}

int main()
{
    std::string folder_name = "F:\\VIDEO_MP4_20201203110959274_001\\temp";
    std::vector<std::string> imagenamelist_beforesort;
    getImageNameList(folder_name, imagenamelist_beforesort);
    for (int i = 0; i < imagenamelist_beforesort.size(); i++)
    {
        std::cout << imagenamelist_beforesort[i] << std::endl;
    }
    std::cout << "**************************" << std::endl;
    std::vector<std::string> imagenamelist = sortImageNameList(imagenamelist_beforesort);
    for (int i = 0; i < imagenamelist.size(); i++)
    {
        std::cout << imagenamelist[i] << std::endl;
    }

    return 0;
}

结果：

 

疑问：

我去，std::atoi("0.jpg")这么用也行嘛？？从程序上看，是出来结果了，也不知道有没有什么危险，如果有路过的大牛，希望不吝赐教，如果个人什么时候有时间尝试了，也会贴出结果

 

python 编程实现：

src=[]

for item in os.listdir(path):
    src.append(item)

bcontrol = 1
dst = []
for i, imagename in enumerate(src):
    if i is 0:
        dst.append(imagename)
    else:
        bcontrol = 1
        num, end = imagename.split(".")
        for j, sub in enumerate(dst):
            num1, end1 = sub.split(".")
            if int(num1) > int(num):
                dst.insert(j, imagename)
                bcontrol = 0
                break
        if bcontrol:
            dst.append(imagename)