optimizer_main.cpp

#include "functions.h"
#include <iostream>
#include <vector>
#include <string>

int main(int argc, char* argv[]) {
    if (argc != 9) {
        std::cerr << "Usage: optimizer_main nfeatures scaleFactor nlevels edgeThreshold patchSize fastThreshold WTA_K scoreType" << std::endl;
        return 1;
    }

    int nfeatures = std::stoi(argv[1]);
    float scaleFactor = std::stof(argv[2]);
    int nlevels = std::stoi(argv[3]);
    int edgeThreshold = std::stoi(argv[4]);
    int patchSize = std::stoi(argv[5]);
    int fastThreshold = std::stoi(argv[6]);
    int wta_k = std::stoi(argv[7]);
    int score_type = std::stoi(argv[8]);

    std::vector<std::string> object_folders = {
        "004_sugar_box",
        "006_mustard_bottle",
        "035_power_drill"
    };

    double miou = process_all_orb(object_folders, nfeatures, scaleFactor, nlevels, edgeThreshold, patchSize, fastThreshold, wta_k, score_type);

    std::cout << miou << std::endl;

    return 0;
}