image.h

浏览该文件的文档.

/*
 * @Description: 图片类 头文件
 * @Version:
 * @Autor: dreamy-xay
 * @date: 2023-12-17
 * @LastEditors: dreamy-xay
 * @LastEditTime: 2024-05-28
 */
 
#ifndef COMMON_IMAGE_H
#define COMMON_IMAGE_H
 
#include <opencv2/core.hpp>
#include <vector>
 
#if defined(__linux__)
#include <opencv2/imgproc/types_c.h>
#endif
 
#include "common/base/interval.hpp"
#include "common/debug/debug.hpp"
#include "common/enum.h"
#include "common/list/lines.hpp"
#include "common/list/rects.hpp"
#include "common/macro.h"
#include "common/type.h"
#include "common/feature/image_preprocess.hpp"
 
namespace cm {
 
class Image : public cv::Mat {
   public:
    using cv::Mat::Mat;
 
    Image();
    Image(const Image& image);
    Image(const cv::Mat& image);
    ~Image();
 
    cv::Size Size() const;
 
    Image& Resize(size_t height, size_t width, int interpolation = cv::INTER_LINEAR);
    Image& Resize(size_t max_size, ResizeType resize_type = RT_AUTO);
    Image& Scale(double factor, int interpolation = cv::INTER_LINEAR);
 
    Image& RmRedBlueStamp(uchar bg_gray_value = 255);
    Image& RmRedStamp(uchar bg_gray_value = 255);
    static double AdjustImageByGrayImage(cv::Mat& image, const cv::Mat& gray_image, uchar bg_gray_value, const std::vector<uchar>& thresholds);
 
    Image& HorizontalLengthSmooth(int threshold);
 
    Rects GetTextAreaOutline(size_t dilate_size, double max_aspect_ratio = 30) const;
 
    Lines GetLines(LineType line_type, const ImagePreprocessor& imp, int threshold, double min_line_length, double max_line_gap = 10) const;
 
    Image& ModifyLinesAreaValue(LineType line_type, const Lines& lines, int pixel_margin = 2, uchar pixel_value = 0);
 
    Image InvertColorRegion(double min_area = 7000) const;
 
    Image SeparateColor(SeparatedColorType type) const;
 
    Image GetGrayHistogram() const;
};
 
}  // namespace cm
 
#endif