generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

extended MinProblemSolver::Function to 1) report the space dimensionality, 2) compute gradient if needed

Browse Source
This commit is contained in:
Vadim Pisarevsky
2015-05-05 15:56:06 +03:00
parent 5b9182ba43
commit 63a63e3eaa
5 changed files with 32 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules/core/test/test_conjugate_gradient.cpp
View File

@@ -60,16 +60,19 @@ static void mytest(cv::Ptr<cv::ConjGradSolver> solver,cv::Ptr<cv::MinProblemSolv
class SphereF_CG:public cv::MinProblemSolver::Function{
public:
int getDims() const { return 4; }
double calc(const double* x)const{
return x[0]*x[0]+x[1]*x[1]+x[2]*x[2]+x[3]*x[3];
}
void getGradient(const double* x,double* grad){
// use automatically computed gradient
/*void getGradient(const double* x,double* grad){
for(int i=0;i<4;i++){
grad[i]=2*x[i];
}
}
}*/
};
class RosenbrockF_CG:public cv::MinProblemSolver::Function{
int getDims() const { return 2; }
double calc(const double* x)const{
return 100*(x[1]-x[0]*x[0])*(x[1]-x[0]*x[0])+(1-x[0])*(1-x[0]);
}