third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #10398 from alalek:ml_simplify_simulated_annealing

Browse Source
This commit is contained in:
Alexander Alekhin
2017-12-22 22:43:59 +03:00
committed by GitHub
parent 9148a37624 00e43a9022
commit eba176c299
4 changed files with 100 additions and 209 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samples/cpp/travelsalesman.cpp
+9 -16
View File
@@ -2,7 +2,7 @@
using namespace cv;
class TravelSalesman : public ml::SimulatedAnnealingSolverSystem
class TravelSalesman
{
private :
const std::vector<Point>& posCity;
@@ -17,11 +17,11 @@ public:
rng = theRNG();
}
/** Give energy value for a state of system.*/
/*virtual*/ double energy() const;
double energy() const;
/** Function which change the state of system (random pertubation).*/
/*virtual*/ void changeState();
void changeState();
/** Function to reverse to the previous state.*/
/*virtual*/ void reverseState();
void reverseState();
};
@@ -81,31 +81,24 @@ int main(void)
posCity[i].y = static_cast<int>(radius*sin(theta)) + center.y;
next[i]=(i+1)%nbCity;
}
Ptr<TravelSalesman> ts_system(new TravelSalesman(posCity, next));
ml::SimulatedAnnealingSolver ts(ts_system);
TravelSalesman ts_system(posCity, next);
ts.setIterPerStep(10000*nbCity);
ts.setCoolingRatio(0.99);
ts.setInitialTemperature(100);
ts.setFinalTemperature(100*0.97);
DrawTravelMap(img,posCity,next);
imshow("Map",img);
waitKey(10);
double currentTemperature = 100.0;
for (int i = 0, zeroChanges = 0; zeroChanges < 10; i++)
{
int changesApplied = ts.run();
img = Mat::zeros(img.size(),CV_8UC3);
int changesApplied = ml::simulatedAnnealingSolver(ts_system, currentTemperature, currentTemperature*0.97, 0.99, 10000*nbCity, &currentTemperature, rng);
img.setTo(Scalar::all(0));
DrawTravelMap(img, posCity, next);
imshow("Map", img);
int k = waitKey(10);
double ti=ts.getFinalTemperature();
std::cout << "i=" << i << " changesApplied=" << changesApplied << " temp=" << ti << " result=" << ts_system->energy() << std::endl;
std::cout << "i=" << i << " changesApplied=" << changesApplied << " temp=" << currentTemperature << " result=" << ts_system.energy() << std::endl;
if (k == 27 || k == 'q' || k == 'Q')
return 0;
if (changesApplied == 0)
zeroChanges++;
ts.setInitialTemperature(ti);
ts.setFinalTemperature(ti*0.97);
}
std::cout << "Done" << std::endl;
waitKey(0);