LutzSegmentation.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_SEGMENTATION_LUTZ_SEGMENTATION_H
#define _SEIMPLEMENTATION_SEGMENTATION_LUTZ_SEGMENTATION_H
 
#include <cassert>
#include <memory>
#include "SEFramework/Frame/Frame.h"
#include "SEFramework/Source/SourceFactory.h"
#include "SEFramework/Pipeline/Segmentation.h"
 
namespace SourceXtractor {
 
class LutzSegmentation : public Segmentation::Labelling {
 
public:
 
  virtual ~LutzSegmentation() = default;
 
  explicit LutzSegmentation(std::shared_ptr<SourceFactory> source_factory, int window_size = 0)
      : m_source_factory(source_factory),
        m_window_size(window_size) {
    assert(source_factory != nullptr);
  }
 
  void labelImage(Segmentation::LabellingListener& listener, std::shared_ptr<const DetectionImageFrame> frame) override;
 
private:
  std::shared_ptr<SourceFactory> m_source_factory;
  int m_window_size;
};
 
} /* namespace SourceXtractor */
 
#endif