ImageDescriber_SIFT_vlfeat.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/Descriptor.hpp>
#include <aliceVision/feature/ImageDescriber.hpp>
#include <aliceVision/feature/regionsFactory.hpp>
#include <aliceVision/feature/sift/SIFT.hpp>
 
extern "C"
{
#include <nonFree/sift/vl/sift.h>
}
 
#include <iostream>
#include <numeric>
 
namespace aliceVision {
namespace feature {
 
class ImageDescriber_SIFT_vlfeat : public ImageDescriber
{
  public:
    explicit ImageDescriber_SIFT_vlfeat(const SiftParams& params = SiftParams(), bool isOriented = true)
      : ImageDescriber(),
        _params(params),
        _isOriented(isOriented)
    {
        // Configure VLFeat
        VLFeatInstance::initialize();
    }
 
    ~ImageDescriber_SIFT_vlfeat() override { VLFeatInstance::destroy(); }
 
    bool useCuda() const override { return false; }
 
    bool useFloatImage() const override { return true; }
 
    EImageDescriberType getDescriberType() const override
    {
        if (!_isOriented)
            return EImageDescriberType::SIFT_UPRIGHT;
        return EImageDescriberType::SIFT;
    }
 
    std::size_t getMemoryConsumption(std::size_t width, std::size_t height) const override
    {
        return getMemoryConsumptionVLFeat(width, height, _params);
    }
 
    void setUpRight(bool upRight) override { _isOriented = !upRight; }
 
    void setConfigurationPreset(ConfigurationPreset preset) override { _params.setPreset(preset); }
 
    bool describe(const image::Image<float>& image, std::unique_ptr<Regions>& regions, const image::Image<unsigned char>* mask = nullptr) override
    {
        return extractSIFT<unsigned char>(image, regions, _params, _isOriented, mask);
    }
 
    void allocate(std::unique_ptr<Regions>& regions) const override { regions.reset(new SIFT_Regions); }
 
  private:
    SiftParams _params;
    bool _isOriented;
};
 
}  // namespace feature
}  // namespace aliceVision