SourceXtractorPlusPlus  0.16
Please provide a description of the project.
SegmentationConfig.h
Go to the documentation of this file.
1 
23 #ifndef _SEIMPLEMENTATION_SEGMENTATIONCONFIG_H
24 #define _SEIMPLEMENTATION_SEGMENTATIONCONFIG_H
25 
26 #include <memory>
28 
29 namespace SourceXtractor {
30 
37 public:
38 
39  enum class Algorithm {
40  UNKNOWN,
41  LUTZ,
42  BFS,
43  ML
44  };
45 
49  virtual ~SegmentationConfig() = default;
50 
52  explicit SegmentationConfig(long manager_id);
53 
55  void preInitialize(const UserValues& args) override;
56  void initialize(const UserValues& args) override;
57 
59  return m_selected_algorithm;
60  }
61 
62  int getLutzWindowSize() const {
63  return m_lutz_window_size;
64  }
65 
66  int getBfsMaxDelta() const {
67  return m_bfs_max_delta;
68  }
69 
70  bool isFilteringEnabled() const {
71  return m_filter != nullptr;
72  }
73 
75  return m_filter;
76  }
77 
79  return m_onnx_model_path;
80  }
81 
82  double getMLThreashold() const {
83  return m_ml_threshold;
84  }
85 
86 
87 private:
92 
95 
100 }; /* End of SegmentationConfig class */
101 
102 } /* namespace SourceXtractor */
103 
104 
105 #endif
Used to select a Segmentation algorithm.
std::shared_ptr< DetectionImageFrame::ImageFilter > m_filter
SegmentationConfig(long manager_id)
Constructs a new SegmentationConfig object.
std::shared_ptr< DetectionImageFrame::ImageFilter > getFilter() const
std::map< std::string, Configuration::OptionDescriptionList > getProgramOptions() override
void preInitialize(const UserValues &args) override
std::shared_ptr< DetectionImageFrame::ImageFilter > getDefaultFilter() const
void initialize(const UserValues &args) override
std::shared_ptr< DetectionImageFrame::ImageFilter > loadFITSFilter(const std::string &filename) const
std::shared_ptr< DetectionImageFrame::ImageFilter > loadASCIIFilter(const std::string &filename) const
std::shared_ptr< DetectionImageFrame::ImageFilter > loadFilter(const std::string &filename) const
virtual ~SegmentationConfig()=default
Destructor.
string filename
Definition: conf.py:63