50 using namespace ModelFitting;
52 int main(
int argc,
char **argv) {
55 engine_impl = argv[1];
64 size_t image_cols = image.cols;
65 size_t image_rows = image.rows;
81 auto i0 = std::make_shared<EngineParameter>(50000., make_unique<ExpSigmoidConverter>(1, 1000000.));
82 auto n = std::make_shared<ManualParameter>(1.);
83 auto k = std::make_shared<ManualParameter>(1.);
86 auto reg_man = make_unique<OnlySmooth>();
87 auto exp = make_unique<SersicModelComponent>(
move(reg_man), i0, n, k);
93 auto x = std::make_shared<EngineParameter>(120, make_unique<NormalizedConverter>(1500.));
94 auto y = std::make_shared<EngineParameter>(140, make_unique<NormalizedConverter>(1500.));
95 auto x_scale = std::make_shared<EngineParameter>(1.0, make_unique<SigmoidConverter>(0, 10.));
96 auto y_scale = std::make_shared<EngineParameter>(1.0, make_unique<SigmoidConverter>(0, 10.));
97 auto rot_angle = std::make_shared<EngineParameter>(20.0 * M_PI/180.0, make_unique<SigmoidConverter>(0, 2*M_PI));
106 rot_angle, width, height,
x, y);
109 auto back = std::make_shared<EngineParameter>(100., make_unique<ExpSigmoidConverter>(1, 1000000.));
115 auto psf =
readPsf(psf_path[0].
string());
124 writeToFits(frame_model.getImage(),
"example_gal2_before.fits");
133 manager.registerParameter(
x);
134 manager.registerParameter(y);
135 manager.registerParameter(x_scale);
136 manager.registerParameter(y_scale);
137 manager.registerParameter(rot_angle);
138 manager.registerParameter(back);
144 cv::Mat weight = cv::Mat::ones(image.rows, image.cols, CV_64F);
155 [](
double x,
double y){
return x/
y;},
159 res_estimator.registerBlockProvider(make_unique<WorldValueResidual>(
166 cout <<
"I0 = " << i0->getValue() <<
'\n';
167 cout <<
"X = " << x->getValue() <<
'\n';
168 cout <<
"Y = " << y->getValue() <<
'\n';
169 cout <<
"X_SCALE = " << x_scale->getValue() <<
'\n';
170 cout <<
"Y_SCALE = " << y_scale->getValue() <<
'\n';
171 cout <<
"angle = " << rot_angle->getValue() <<
'\n';
172 cout <<
"Background = " << back->getValue() <<
'\n';
175 auto engine = LeastSquareEngineManager::create(engine_impl);
177 auto solution = engine->solveProblem(manager, res_estimator);
184 cout <<
"I0 = " << i0->getValue() <<
'\n';
185 cout <<
"X = " << x->getValue() <<
'\n';
186 cout <<
"Y = " << y->getValue() <<
'\n';
187 cout <<
"X_SCALE = " << x_scale->getValue() <<
'\n';
188 cout <<
"Y_SCALE = " << y_scale->getValue() <<
'\n';
189 cout <<
"angle = " << rot_angle->getValue() <<
'\n';
190 cout <<
"Background = " << back->getValue() <<
'\n';
195 reg_man = make_unique<OnlySmooth>();
196 exp = make_unique<SersicModelComponent>(
move(reg_man), i0, n, k);
197 component_list.clear();
198 component_list.emplace_back(
move(exp));
199 extended_models.clear();
200 extended_models.emplace_back(
move(component_list),x_scale, y_scale,
201 rot_angle, width, height, x, y);
202 constant_models.clear();
203 constant_models.emplace_back(back);
206 move(extended_models),
readPsf(psf_path[0].
string())
208 writeToFits(frame_model_after.getImage(),
"example_gal2_after.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< Parameters...> > createDependentParameter(typename DependentParameter< Parameters...>::ValueCalculator value_calculator, Parameters...parameters)
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.