Commit 6f4012df authored by berak's avatar berak

xphoto: fix learning_based_color_balance.cpp

parent 53986aaa
......@@ -95,21 +95,19 @@ class LearningBasedWBImpl : public LearningBasedWB
palette_size = 300;
palette_bandwidth = 0.1f;
prediction_thresh = 0.025f;
if (path_to_model.empty())
/* try to load model from file */
FileStorage fs;
if (!path_to_model.empty() && fs.open(path_to_model, FileStorage::READ))
{
/* use the default model */
num_trees = _num_trees;
num_tree_nodes = _num_tree_nodes;
feature_idx = _feature_idx;
thresh_vals = _thresh_vals;
leaf_vals = _leaf_vals;
}
else
{
/* load model from file */
FileStorage fs(path_to_model, 0);
num_trees = fs["num_trees"];
num_tree_nodes = fs["num_tree_nodes"];
if (fs["num_trees"].isReal()) { //workaround for #10506
double nt = fs["num_trees"];
num_trees = int(nt);
double ntn = fs["num_tree_nodes"];
num_tree_nodes = int(ntn);
} else {
num_trees = fs["num_trees"];
num_tree_nodes = fs["num_tree_nodes"];
}
fs["feature_idx"] >> feature_idx_Mat;
fs["thresh_vals"] >> thresh_vals_Mat;
fs["leaf_vals"] >> leaf_vals_Mat;
......@@ -117,6 +115,15 @@ class LearningBasedWBImpl : public LearningBasedWB
thresh_vals = thresh_vals_Mat.ptr<float>();
leaf_vals = leaf_vals_Mat.ptr<float>();
}
else
{
/* use the default model */
num_trees = _num_trees;
num_tree_nodes = _num_tree_nodes;
feature_idx = _feature_idx;
thresh_vals = _thresh_vals;
leaf_vals = _leaf_vals;
}
}
int getRangeMaxVal() const { return range_max_val; }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment