SIFT-GPU关于SaveSIFT函数的改进

最新推荐文章于 2024-10-09 09:17:25 发布

lhanchao

最新推荐文章于 2024-10-09 09:17:25 发布

阅读量2.8k

点赞数

分类专栏：三维重建文章标签： sift SIFT-GPU

本文链接：https://blog.csdn.net/lhanchao/article/details/52222116

版权

本文介绍了SIFT-GPU在SIFT特征提取过程中，SaveSIFT函数导致运行速度较慢的问题。作者通过分析源码，发现C++的stream类保存方式效率较低，改为使用fprintf提高速度。在改进后的代码中，二进制格式保存部分未改动，但整体运行速度提升至原来的10倍。

摘要由CSDN通过智能技术生成

SIFT-GPU是一个高效的进行SIFT特征提取和匹配的工具，是目前我用过的速度最快的工具。网址如下：http://www.cs.unc.edu/~ccwu/siftgpu/#lowesift
我在用SIFT-GPU进行SIFT特征提取的过程中，发现运行的速度比较慢，通过对运行的代码的时间分析，发现主要元凶是保存SIFT特征的函数SaveSIFT占用了大量的时间（包括匹配的时间，save的时间大约在80%以上）。然后分析源码可以看出来它的源码是用C++的stream类进行保存的，我们都知道C语言的fprintf比前面的方法要快很多，因此我改进了其中的代码，是SiftGPU工程下的SiftPyramid.cpp文件中的SaveSIFT函数。
这里我只改了一部分，因为SaveSIFT是可以选择保存为二进制格式文件还是保存为普通的文件的，对于保存为二进制格式的部分我没有改动，有需要的请自己修改，改动的方法是相同的。调整后的代码运行速度大约是原有代码运行速度的10倍。
改进后的代码如下：

void SiftPyramid::SaveSIFT(const char * szFileName)
{
    if (_featureNum <=0) return;
    float * pk = &_keypoint_buffer[0];
    FILE *f = fopen(szFileName, "w");
    cout << szFileName << endl;
    if(GlobalUtil::_BinarySIFT)
    {
        std::ofstream out(szFileName, ios::binary);
        out.write((char* )(&_featureNum), sizeof(int));

        if(GlobalUtil::_DescriptorPPT)
        {
            int dim = 128;