OpenCV3.3中給出了邏輯回歸(logistic regression)的實(shí)現(xiàn)，即cv::ml::LogisticRegression類，類的聲明在include/opencv2/ml.hpp文件中，實(shí)現(xiàn)在modules/ml/src/lr.cpp文件中,它既支持兩分類，也支持多分類，其中：

(1)、cv::ml::LogisticRegression類繼承自cv::ml::StateModel，而cv::ml::StateModel又繼承自cv::Algorithm；

(2)、setLearningRate函數(shù)用來(lái)設(shè)置學(xué)習(xí)率，getLearningRate函數(shù)用來(lái)獲取學(xué)習(xí)率值；

(3)、setIterations函數(shù)用來(lái)設(shè)置迭代次數(shù)，getIterations函數(shù)用來(lái)獲取迭代次數(shù)值；

(4)、setRegularization函數(shù)用來(lái)設(shè)置采用哪種正則化方法，目前支持兩種L1 norm和L2 norm，正則化方法主要用來(lái)防止過擬合，getRegularization函數(shù)用來(lái)獲取采用哪種正則化方法；

(5)、setTrainMethod函數(shù)用來(lái)設(shè)置采用哪種訓(xùn)練方法，目前支持兩種Batch和Mini-Batch, getTrainMethod函數(shù)用來(lái)獲取采用哪種訓(xùn)練方法；

(6)、setMiniBatchSize函數(shù)用來(lái)設(shè)置在Mini-Batch梯度下降訓(xùn)練方法中每一個(gè)step采集的訓(xùn)練樣本數(shù)，getMiniBatchSize函數(shù)用來(lái)獲取每一個(gè)step采集的訓(xùn)練樣本數(shù)；

(7)、setTermCriteria函數(shù)用來(lái)設(shè)置終止訓(xùn)練的條件，包括迭代次數(shù)和期望的精度，getTermCriteria用來(lái)獲取終止訓(xùn)練的條件；

(8)、get_learnt_thetas函數(shù)用來(lái)獲取訓(xùn)練參數(shù)；

(9)、create函數(shù)為static, new一個(gè)LogisticRegressionImpl用來(lái)創(chuàng)建一個(gè)LogisticRegression對(duì)象;

(10)、train函數(shù)(使用基類StatModel中的)進(jìn)行訓(xùn)練；

(11)、predict函數(shù)用于預(yù)測(cè)；

(12)、save函數(shù)(使用基類Algorithm中的)保存已訓(xùn)練好的model，支持xml,yaml,json格式;

(13)、load函數(shù)用來(lái)load已訓(xùn)練好的model；

? 以下為兩分類測(cè)試代碼：訓(xùn)練數(shù)據(jù)集為從MNIST中train中隨機(jī)選取的0、1各10個(gè)圖像；測(cè)試數(shù)據(jù)集為從MNIST中test中隨機(jī)選取的0、1各10個(gè)圖像，如下圖，其中第一排前10個(gè)0用于訓(xùn)練，后10個(gè)0用于測(cè)試；第二排前10個(gè)1用于訓(xùn)練，后10個(gè)1用于測(cè)試：

#include "opencv.hpp"
#include <string>
#include <vector>
#include <memory>
#include <opencv2/opencv.hpp>
#include <opencv2/ml.hpp>
#include "common.hpp"// Logistic Regression ///
static void show_image(const cv::Mat& data, int columns, const std::string& name)
{cv::Mat big_image;for (int i = 0; i < data.rows; ++i) {big_image.push_back(data.row(i).reshape(0, columns));}cv::imshow(name, big_image);cv::waitKey(0);
}static float calculate_accuracy_percent(const cv::Mat& original, const cv::Mat& predicted)
{return 100 * (float)cv::countNonZero(original == predicted) / predicted.rows;
}int test_opencv_logistic_regression_train()
{const std::string image_path{ "E:/GitCode/NN_Test/data/images/digit/handwriting_0_and_1/" };cv::Mat data, labels, result;for (int i = 1; i < 11; ++i) {const std::vector<std::string> label{ "0_", "1_" };for (const auto& value : label) {std::string name = std::to_string(i);name = image_path + value + name + ".jpg";cv::Mat image = cv::imread(name, 0);if (image.empty()) {fprintf(stderr, "read image fail: %s\n", name.c_str());return -1;}data.push_back(image.reshape(0, 1));}}data.convertTo(data, CV_32F);//show_image(data, 28, "train data");std::unique_ptr<float[]> tmp(new float[20]);for (int i = 0; i < 20; ++i) {if (i % 2 == 0) tmp[i] = 0.f;else tmp[i] = 1.f;}labels = cv::Mat(20, 1, CV_32FC1, tmp.get());cv::Ptr<cv::ml::LogisticRegression> lr = cv::ml::LogisticRegression::create();lr->setLearningRate(0.00001);lr->setIterations(100);lr->setRegularization(cv::ml::LogisticRegression::REG_DISABLE);lr->setTrainMethod(cv::ml::LogisticRegression::MINI_BATCH);lr->setMiniBatchSize(1);CHECK(lr->train(data, cv::ml::ROW_SAMPLE, labels));const std::string save_file{ "E:/GitCode/NN_Test/data/logistic_regression_model.xml" }; // .xml, .yaml, .jsonslr->save(save_file);return 0;
}int test_opencv_logistic_regression_predict()
{const std::string image_path{ "E:/GitCode/NN_Test/data/images/digit/handwriting_0_and_1/" };cv::Mat data, labels, result;for (int i = 11; i < 21; ++i) {const std::vector<std::string> label{ "0_", "1_" };for (const auto& value : label) {std::string name = std::to_string(i);name = image_path + value + name + ".jpg";cv::Mat image = cv::imread(name, 0);if (image.empty()) {fprintf(stderr, "read image fail: %s\n", name.c_str());return -1;}data.push_back(image.reshape(0, 1));}}data.convertTo(data, CV_32F);//show_image(data, 28, "test data");std::unique_ptr<int[]> tmp(new int[20]);for (int i = 0; i < 20; ++i) {if (i % 2 == 0) tmp[i] = 0;else tmp[i] = 1;}labels = cv::Mat(20, 1, CV_32SC1, tmp.get());const std::string model_file{ "E:/GitCode/NN_Test/data/logistic_regression_model.xml" };cv::Ptr<cv::ml::LogisticRegression> lr = cv::ml::LogisticRegression::load(model_file);lr->predict(data, result);fprintf(stdout, "predict result: \n");std::cout << "actual: " << labels.t() << std::endl;std::cout << "target: " << result.t() << std::endl;fprintf(stdout, "accuracy: %.2f%%\n", calculate_accuracy_percent(labels, result));return 0;
}

測(cè)試代碼中，test_opencv_logistic_regression_train函數(shù)用于訓(xùn)練，訓(xùn)練結(jié)果會(huì)產(chǎn)生一個(gè)叫l(wèi)ogistic_regression_model.xml的model文件；test_opencv_logistic_regression_predict函數(shù)用于預(yù)測(cè)，預(yù)測(cè)結(jié)果如下，由結(jié)果可知，預(yù)測(cè)全部正確：

GitHub：?https://github.com/fengbingchun/NN_Test

總結(jié)

以上是生活随笔為你收集整理的OpenCV3.3中逻辑回归(Logistic Regression)使用举例的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。