Blame view

3rdparty/opencv_4_1/include/opencv2/ml/ml.inl.hpp 1.73 KB
09c2d08c   Hu Chunming   arm交付版
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
  // This file is part of OpenCV project.
  // It is subject to the license terms in the LICENSE file found in the top-level directory
  // of this distribution and at http://opencv.org/license.html.
  
  #ifndef OPENCV_ML_INL_HPP
  #define OPENCV_ML_INL_HPP
  
  namespace cv { namespace ml {
  
  // declared in ml.hpp
  template<class SimulatedAnnealingSolverSystem>
  int simulatedAnnealingSolver(SimulatedAnnealingSolverSystem& solverSystem,
       double initialTemperature, double finalTemperature, double coolingRatio,
       size_t iterationsPerStep,
       CV_OUT double* lastTemperature,
       cv::RNG& rngEnergy
  )
  {
      CV_Assert(finalTemperature > 0);
      CV_Assert(initialTemperature > finalTemperature);
      CV_Assert(iterationsPerStep > 0);
      CV_Assert(coolingRatio < 1.0f);
      double Ti = initialTemperature;
      double previousEnergy = solverSystem.energy();
      int exchange = 0;
      while (Ti > finalTemperature)
      {
          for (size_t i = 0; i < iterationsPerStep; i++)
          {
              solverSystem.changeState();
              double newEnergy = solverSystem.energy();
              if (newEnergy < previousEnergy)
              {
                  previousEnergy = newEnergy;
                  exchange++;
              }
              else
              {
                  double r = rngEnergy.uniform(0.0, 1.0);
                  if (r < std::exp(-(newEnergy - previousEnergy) / Ti))
                  {
                      previousEnergy = newEnergy;
                      exchange++;
                  }
                  else
                  {
                      solverSystem.reverseState();
                  }
              }
          }
          Ti *= coolingRatio;
      }
      if (lastTemperature)
          *lastTemperature = Ti;
      return exchange;
  }
  
  }} //namespace
  
  #endif // OPENCV_ML_INL_HPP