48 using namespace ModelFitting;
60 int main(
int argc,
char **argv) {
63 engine_impl = argv[1];
72 size_t image_cols = image.cols;
73 size_t image_rows = image.rows;
89 auto i0 = std::make_shared<EngineParameter>(50000., make_unique<ExpSigmoidConverter>(1, 1000000.));
90 auto n = std::make_shared<ManualParameter>(1.);
91 auto k = std::make_shared<ManualParameter>(1.);
94 auto reg_man = make_unique<OnlySmooth>();
95 auto exp = make_unique<SersicModelComponent>(
move(reg_man), i0, n, k);
101 auto x = std::make_shared<EngineParameter>(120, make_unique<NormalizedConverter>(1500.));
102 auto y = std::make_shared<EngineParameter>(140, make_unique<NormalizedConverter>(1500.));
103 auto x_scale = std::make_shared<EngineParameter>(1.0, make_unique<SigmoidConverter>(0, 10.));
104 auto y_scale = std::make_shared<EngineParameter>(1.0, make_unique<SigmoidConverter>(0, 10.));
105 auto rot_angle = std::make_shared<EngineParameter>(20.0 * M_PI/180.0, make_unique<SigmoidConverter>(0, 2*M_PI));
114 rot_angle, width, height,
x, y);
117 auto back = std::make_shared<EngineParameter>(100., make_unique<ExpSigmoidConverter>(1, 1000000.));
123 auto psf =
readPsf(psf_path[0].
string());
132 writeToFits(frame_model.getImage(),
"example3b.fits");
141 manager.registerParameter(
x);
142 manager.registerParameter(y);
143 manager.registerParameter(x_scale);
144 manager.registerParameter(y_scale);
145 manager.registerParameter(rot_angle);
146 manager.registerParameter(back);
152 cv::Mat weight = cv::Mat::ones(image.rows, image.cols, CV_64F);
161 cout <<
"I0 = " << i0->getValue() <<
'\n';
162 cout <<
"X = " <<
x->getValue() <<
'\n';
163 cout <<
"Y = " << y->getValue() <<
'\n';
164 cout <<
"X_SCALE = " << x_scale->getValue() <<
'\n';
165 cout <<
"Y_SCALE = " << y_scale->getValue() <<
'\n';
166 cout <<
"angle = " << rot_angle->getValue() <<
'\n';
167 cout <<
"Background = " << back->getValue() <<
'\n';
170 auto engine = LeastSquareEngineManager::create(engine_impl);
172 auto solution = engine->solveProblem(manager, res_estimator);
179 cout <<
"I0 = " << i0->getValue() <<
'\n';
180 cout <<
"X = " <<
x->getValue() <<
'\n';
181 cout <<
"Y = " << y->getValue() <<
'\n';
182 cout <<
"X_SCALE = " << x_scale->getValue() <<
'\n';
183 cout <<
"Y_SCALE = " << y_scale->getValue() <<
'\n';
184 cout <<
"angle = " << rot_angle->getValue() <<
'\n';
185 cout <<
"Background = " << back->getValue() <<
'\n';
190 reg_man = make_unique<OnlySmooth>();
191 exp = make_unique<SersicModelComponent>(
move(reg_man), i0, n, k);
192 component_list.clear();
193 component_list.emplace_back(
move(exp));
194 extended_models.clear();
195 extended_models.emplace_back(
move(component_list),x_scale, y_scale,
196 rot_angle, width, height,
x, y);
197 constant_models.clear();
198 constant_models.emplace_back(back);
201 move(extended_models),
readPsf(psf_path[0].
string())
203 writeToFits(frame_model_after.getImage(),
"example3b2.fits");
std::pair< cv::Mat, double > readImage(const std::string &filename)
std::shared_ptr< DependentParameter< std::shared_ptr< EngineParameter > > > x
ModelFitting::OpenCvPsf readPsf(const std::string &filename)
std::shared_ptr< DependentParameter< std::shared_ptr< EngineParameter > > > y
void registerBlockProvider(std::unique_ptr< ResidualBlockProvider > provider)
Registers a ResidualBlockProvider to the ResidualEstimator.
void registerParameter(std::shared_ptr< EngineParameter > parameter)
Registers an EngineParameter to the EngineParameterManager.
void writeToFits(const cv::Mat &image, const std::string &filename)
ELEMENTS_API std::vector< boost::filesystem::path > pathSearchInEnvVariable(const std::string &file_name, const std::string &path_like_env_variable, SearchType search_type=SearchType::Recursive)
Class responsible for managing the parameters the least square engine minimizes.
std::unique_ptr< DataVsModelResiduals< typename std::remove_reference< DataType >::type, typename std::remove_reference< ModelType >::type, typename std::remove_reference< WeightType >::type, typename std::remove_reference< Comparator >::type > > createDataVsModelResiduals(DataType &&data, ModelType &&model, WeightType &&weight, Comparator &&comparator)
void printLevmarInfo(std::array< double, 10 > info)
Provides to the LeastSquareEngine the residual values.
Data vs model comparator which computes a modified residual.