SourceXtractorPlusPlus  0.16
Please provide a description of the project.
MLSegmentation.h
Go to the documentation of this file.
1 
19 #ifndef _SEIMPLEMENTATION_SEGMENTATION_MLSEGMENTATION_H_
20 #define _SEIMPLEMENTATION_SEGMENTATION_MLSEGMENTATION_H_
21 
24 
25 namespace SourceXtractor {
26 
27 
33 public:
34 
35  virtual ~MLSegmentation() = default;
36 
37  MLSegmentation(std::shared_ptr<SourceFactory> source_factory, std::string model_path, double ml_threshold)
38  : m_source_factory(source_factory), m_model_path(model_path), m_ml_threshold(ml_threshold) {
39  assert(source_factory != nullptr);
40  }
41 
43 
44 private:
45  struct Tile {
47  int width, height;
48  };
49 
51 
52 
54 
57 
58 };
59 
60 }
61 
62 #endif /* _SEIMPLEMENTATION_SEGMENTATION_MLSEGMENTATION_H_ */
Interface representing an image.
Definition: Image.h:43
Implements a Segmentation based on CNN.
virtual ~MLSegmentation()=default
std::vector< MLSegmentation::Tile > getTiles(const DetectionImage &image) const
void labelImage(Segmentation::LabellingListener &listener, std::shared_ptr< const DetectionImageFrame > frame) override
std::shared_ptr< SourceFactory > m_source_factory
MLSegmentation(std::shared_ptr< SourceFactory > source_factory, std::string model_path, double ml_threshold)
A pixel coordinate made of two integers m_x and m_y.