You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
3.1 KiB
77 lines
3.1 KiB
// 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. |
|
// |
|
// Tencent is pleased to support the open source community by making WeChat QRCode available. |
|
// Copyright (C) 2020 THL A29 Limited, a Tencent company. All rights reserved. |
|
|
|
#ifndef __OPENCV_WECHAT_QRCODE_HPP__ |
|
#define __OPENCV_WECHAT_QRCODE_HPP__ |
|
#include "opencv2/core.hpp" |
|
/** @defgroup wechat_qrcode WeChat QR code detector for detecting and parsing QR code. |
|
*/ |
|
namespace cv { |
|
namespace wechat_qrcode { |
|
//! @addtogroup wechat_qrcode |
|
//! @{ |
|
/** |
|
* @brief WeChat QRCode includes two CNN-based models: |
|
* A object detection model and a super resolution model. |
|
* Object detection model is applied to detect QRCode with the bounding box. |
|
* super resolution model is applied to zoom in QRCode when it is small. |
|
* |
|
*/ |
|
class CV_EXPORTS_W WeChatQRCode { |
|
public: |
|
/** |
|
* @brief Initialize the WeChatQRCode. |
|
* It includes two models, which are packaged with caffe format. |
|
* Therefore, there are prototxt and caffe models (In total, four paramenters). |
|
* |
|
* @param detector_prototxt_path prototxt file path for the detector |
|
* @param detector_caffe_model_path caffe model file path for the detector |
|
* @param super_resolution_prototxt_path prototxt file path for the super resolution model |
|
* @param super_resolution_caffe_model_path caffe file path for the super resolution model |
|
*/ |
|
CV_WRAP WeChatQRCode(const std::string& detector_prototxt_path = "", |
|
const std::string& detector_caffe_model_path = "", |
|
const std::string& super_resolution_prototxt_path = "", |
|
const std::string& super_resolution_caffe_model_path = ""); |
|
~WeChatQRCode(){}; |
|
|
|
/** |
|
* @brief Both detects and decodes QR code. |
|
* To simplify the usage, there is a only API: detectAndDecode |
|
* |
|
* @param img supports grayscale or color (BGR) image. |
|
* @param points optional output array of vertices of the found QR code quadrangle. Will be |
|
* empty if not found. |
|
* @return list of decoded string. |
|
*/ |
|
CV_WRAP std::vector<std::string> detectAndDecode(InputArray img, OutputArrayOfArrays points = noArray()); |
|
|
|
/** |
|
* @brief set scale factor |
|
* QR code detector use neural network to detect QR. |
|
* Before running the neural network, the input image is pre-processed by scaling. |
|
* By default, the input image is scaled to an image with an area of 160000 pixels. |
|
* The scale factor allows to use custom scale the input image: |
|
* width = scaleFactor*width |
|
* height = scaleFactor*width |
|
* |
|
* scaleFactor valuse must be > 0 and <= 1, otherwise the scaleFactor value is set to -1 |
|
* and use default scaled to an image with an area of 160000 pixels. |
|
*/ |
|
CV_WRAP void setScaleFactor(float _scalingFactor); |
|
|
|
CV_WRAP float getScaleFactor(); |
|
|
|
protected: |
|
class Impl; |
|
Ptr<Impl> p; |
|
}; |
|
|
|
//! @} |
|
} // namespace wechat_qrcode |
|
} // namespace cv |
|
#endif // __OPENCV_WECHAT_QRCODE_HPP__
|
|
|