ImageDescriber_SIFT_OCV.hpp

// This file is part of the AliceVision project.
// Copyright (c) 2016 AliceVision contributors.
// Copyright (c) 2012 openMVG contributors.
// This Source Code Form is subject to the terms of the Mozilla Public License,
// v. 2.0. If a copy of the MPL was not distributed with this file,
// You can obtain one at https://mozilla.org/MPL/2.0/.
 
#pragma once
 
#include <aliceVision/feature/imageDescriberCommon.hpp>
#include <aliceVision/feature/ImageDescriber.hpp>
#include <aliceVision/feature/regionsFactory.hpp>
 
namespace aliceVision {
 
namespace image {
 
template<typename T>
class Image;
 
}  // namespace image
 
namespace feature {
 
class SIFT_openCV_Params
{
  public:
    void setConfigurationPreset(ConfigurationPreset preset);
 
    std::size_t gridSize = 4;
    std::size_t maxTotalKeypoints = 1000;
    int nOctaveLayers = 6;            //< default opencv value is 3
    double contrastThreshold = 0.04;  //< default opencv value is 0.04
    double edgeThreshold = 10;
    double sigma = 1.6;
    // bool rootSift = true;
};
 
class ImageDescriber_SIFT_openCV : public ImageDescriber
{
  public:
    bool useCuda() const override { return false; }
 
    bool useFloatImage() const override { return false; }
 
    EImageDescriberType getDescriberType() const override { return EImageDescriberType::SIFT_OCV; }
 
    std::size_t getMemoryConsumption(std::size_t width, std::size_t height) const override
    {
        return 3 * width * height * sizeof(unsigned char) + (_params.maxTotalKeypoints * 128 * sizeof(unsigned char));
    }
 
    void setConfigurationPreset(ConfigurationPreset preset) override { _params.setConfigurationPreset(preset); }
 
    bool describe(const image::Image<unsigned char>& image,
                  std::unique_ptr<Regions>& regions,
                  const image::Image<unsigned char>* mask = NULL) override;
 
    void allocate(std::unique_ptr<Regions>& regions) const override { regions.reset(new SIFT_Regions); }
 
  private:
    SIFT_openCV_Params _params;
};
 
}  // namespace feature
}  // namespace aliceVision