Skip to content

Latest commit

 

History

History
106 lines (77 loc) · 3.09 KB

README.md

File metadata and controls

106 lines (77 loc) · 3.09 KB

SuperPoint_FBOW

本项目是通过SuerPoint与fbow来训练一个基于深度学习词袋

使用的教学视频

B站教学视频

第三方库依赖

opencv3.2.0(opencv4我还没测试过)

libtorch1.0.1

fbow(参考官方仓库:https://github.com/rmsalinas/fbow)

步骤说明

如果图片名称都转换完毕了,那就开始如下步骤

  1. 下载仓库

    git clone https://github.com/luohongk/SuperPoint_FBOW.git
    cd SuperPoint_FBOW
    
  2. 安装libtorch库

    项目根目录执如下命令
    chmod +x libtorch_download.sh
    ./libtorch_download.sh
    
  3. 编译superpoint 在编译superpoint之前,请你先将目录SuperPoint_FBOW/superpoint/src/SPextractor.cc文件中的第152行的权重路径换乘superpoint.pt在你电脑上的绝对路径。

    项目根目录执如下命令
    chmod +x build_superpoint.sh
    ./build_superpoint.sh
    

    编译完成后项目根目录superpoint下有一个build文件夹。build文件夹下有一个libsuperpoint.so。把libsuperpoint.so复制到(根目录/utils/superpoint)。

  4. 编译main入口文件

    在编译之前,请你将SuperPoint_FBOW/utils/fbow_create_voc_step0.cpp文件中的一部分代码设置成你想要的路径。 string outputfilename ="/home/lhk/data/output/superpoint.fbow";这里注意一定要有superpoint.fbow!!!

    第168行 std::stringfolderPath="/home/lhk/workspace/SuperPoint_FBOW/utils/renamed_images"; // 替换为你的训练图片的文件夹路径

       string des="descriptor";
        string outputfilename ="/home/lhk/data/output/superpoint.fbow";
        // 将特征保存到文件
        std::cerr << "saving to " <<outputfilename<< std::endl;
        saveToFile(outputfilename,descriptors,des);
        return 0;
    
    项目根目录执如下命令
    chmod +x run.sh
    ./run.sh
    
  5. 运行输出深度学习描述子(当然,这里也可以输出深度学习特征点,具体就看看代码)

    项目根目录执如下命令
    cd build/utils
    ./fbow_create_voc_step0
    
  6. 输出词袋

    ./fbow_create_voc_step1 /home/lhk/workspace/SuperPoint_FBOW/output/superpoint.fbow /home/lhk/workspace/SuperPoint_FBOW/output/superpointVoc.txt
    

系统说明

系统:ubuntu18.04(其他系统没有测试过,可能会出现一些bug)

图片名称转换说明(如果想要训练自己的图片的话)

需要训练的图片放置在 imges文件夹。在进行训练之前需要用 python先对 images文件夹中的图片重命名。

首先得电脑中得能运行python代码。(pyhton环境配置)

安装os,shutil这两个库(一般都自带)

图片名称转换,运行如下代码

git clone https://github.com/luohongk/SuperPoint_FBOW.git
cd SuperPoint_FBOW
python ./imagesProcess.py