61 int main(
int argc,
char **argv) {
64 engine_impl = argv[1];
73 size_t image_cols = image.cols;
74 size_t image_rows = image.rows;
90 auto i0 = std::make_shared<EngineParameter>(50000., make_unique<ExpSigmoidConverter>(1, 1000000.));
91 auto n = std::make_shared<ManualParameter>(1.);
92 auto k = std::make_shared<ManualParameter>(1.);
95 auto reg_man = make_unique<OnlySmooth>();
96 auto exp = make_unique<SersicModelComponent>(
move(reg_man), i0, n, k);
102 auto x = std::make_shared<EngineParameter>(120, make_unique<NormalizedConverter>(1500.));
103 auto y = std::make_shared<EngineParameter>(140, make_unique<NormalizedConverter>(1500.));
104 auto x_scale = std::make_shared<EngineParameter>(1.0, make_unique<SigmoidConverter>(0, 10.));
105 auto y_scale = std::make_shared<EngineParameter>(1.0, make_unique<SigmoidConverter>(0, 10.));
106 auto rot_angle = std::make_shared<EngineParameter>(20.0 * M_PI/180.0, make_unique<SigmoidConverter>(0, 2*M_PI));
115 rot_angle, width, height,
x,
y));
118 auto back = std::make_shared<EngineParameter>(100., make_unique<ExpSigmoidConverter>(1, 1000000.));
124 auto psf =
readPsf(psf_path[0].
string());
133 writeToFits(frame_model.getImage(),
"example3b.fits");
142 manager.registerParameter(
x);
143 manager.registerParameter(
y);
144 manager.registerParameter(x_scale);
145 manager.registerParameter(y_scale);
146 manager.registerParameter(rot_angle);
147 manager.registerParameter(back);
153 cv::Mat weight = cv::Mat::ones(image.rows, image.cols, CV_64F);
162 cout <<
"I0 = " << i0->getValue() <<
'\n';
163 cout <<
"X = " <<
x->getValue() <<
'\n';
164 cout <<
"Y = " <<
y->getValue() <<
'\n';
165 cout <<
"X_SCALE = " << x_scale->getValue() <<
'\n';
166 cout <<
"Y_SCALE = " << y_scale->getValue() <<
'\n';
167 cout <<
"angle = " << rot_angle->getValue() <<
'\n';
168 cout <<
"Background = " << back->getValue() <<
'\n';
171 auto engine = LeastSquareEngineManager::create(engine_impl);
173 auto solution = engine->solveProblem(manager, res_estimator);
180 cout <<
"I0 = " << i0->getValue() <<
'\n';
181 cout <<
"X = " <<
x->getValue() <<
'\n';
182 cout <<
"Y = " <<
y->getValue() <<
'\n';
183 cout <<
"X_SCALE = " << x_scale->getValue() <<
'\n';
184 cout <<
"Y_SCALE = " << y_scale->getValue() <<
'\n';
185 cout <<
"angle = " << rot_angle->getValue() <<
'\n';
186 cout <<
"Background = " << back->getValue() <<
'\n';
191 reg_man = make_unique<OnlySmooth>();
192 exp = make_unique<SersicModelComponent>(
move(reg_man), i0, n, k);
193 component_list.clear();
194 component_list.emplace_back(
move(
exp));
195 extended_models.clear();
197 rot_angle, width, height,
x,
y));
198 constant_models.clear();
199 constant_models.emplace_back(back);
202 move(extended_models),
readPsf(psf_path[0].
string())
204 writeToFits(frame_model_after.getImage(),
"example3b2.fits");
void printLevmarInfo(std::array< double, 10 > info)
void writeToFits(const cv::Mat &image, const std::string &filename)
std::shared_ptr< DependentParameter< std::shared_ptr< EngineParameter > > > x
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.
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)
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)
Class responsible for managing the parameters the least square engine minimizes.
ModelFitting::OpenCvPsf readPsf(const std::string &filename)
Provides to the LeastSquareEngine the residual values.
std::pair< cv::Mat, double > readImage(const std::string &filename)
Data vs model comparator which computes a modified residual.
std::unique_ptr< T > make_unique(Args &&... args)
T emplace_back(T... args)
int main(int argc, char **argv)