80 lines
2.8 KiB
C++
80 lines
2.8 KiB
C++
/*
|
||
* ====================================================
|
||
|
||
* 功能：通道提取，又名除色。可提取BGR图像中的单个通道、两种通道的混合以及去除彩色像素的图像，目标图像均为灰度图
|
||
* 作者：刘丁维
|
||
* 生成时间：2020/04/21
|
||
* 最近修改时间：v1.0 2020/04/21
|
||
v1.1 2020/06/11 在除红时，增加对比度，提高除色效果。
|
||
v1.2 2020/07/21 修正之前增强红绿蓝效果的色彩配比。
|
||
v1.3 2021/05/24 替换红色增强算法方案。
|
||
v1.4 2023/03/01 增加Red_Plus、Green_Plus和Blue_Plus模式。取消使用CImageApplyAdjustColors增强通道。
|
||
v1.5 2023/03/16 可自定义Red_Plus、Green_Plus和Blue_Plus除色增强比例。
|
||
v1.5.1 2023/03/16 修复通道切换的BUG。
|
||
* 版本号：v1.5.1
|
||
|
||
* ====================================================
|
||
*/
|
||
|
||
#ifndef IMAGE_APPLY_CHANNEL_H
|
||
#define IMAGE_APPLY_CHANNEL_H
|
||
|
||
#include "ImageApply.h"
|
||
|
||
class GIMGPROC_LIBRARY_API CImageApplyChannel : public CImageApply
|
||
{
|
||
public:
|
||
|
||
typedef enum channel
|
||
{
|
||
Red_Plus, //增强红色通道
|
||
Green_Plus, //增强绿色通道
|
||
Blue_Plus, //增强蓝色通道
|
||
All, //去除所有HSV色彩结构中，S大于80的色彩
|
||
Invalid, //无效
|
||
Except_Red, //绿蓝色通道混合
|
||
Except_Green, //红蓝色通道混合
|
||
Except_Blue, //红绿色通道混合
|
||
Red, //红色通道
|
||
Green, //绿色通道
|
||
Blue, //蓝色通道
|
||
}Channel;
|
||
|
||
public:
|
||
|
||
CImageApplyChannel();
|
||
|
||
/// <summary>
|
||
/// 构造函数
|
||
/// 默认Red_Plus时，m_scale = 0.333333；Green_Plus时， m_scale = 1；Blue_Plus时 m_scale = 1
|
||
/// </summary>
|
||
/// <param name="channel">通道模式</param>
|
||
CImageApplyChannel(Channel channel);
|
||
|
||
/// <summary>
|
||
/// 构造函数
|
||
/// </summary>
|
||
/// <param name="channel">通道模式</param>
|
||
/// <param name="scale">仅在Red_Plus、 Green_Plus、 Blue_Plus模式下生效。增强除色效果，取值范围[0, +∞)，值越大增强效果越明显</param>
|
||
CImageApplyChannel(Channel channel, double scale);
|
||
|
||
virtual ~CImageApplyChannel(void);
|
||
|
||
virtual void apply(cv::Mat& pDib, int side);
|
||
|
||
virtual void apply(std::vector<cv::Mat>& mats, bool isTwoSide);
|
||
private:
|
||
|
||
void except_channel(const cv::Mat& src, cv::Mat& dst, int channel);
|
||
|
||
void colourless(const cv::Mat& src, cv::Mat& dst, uchar threshold = 80);
|
||
|
||
void channel_plus(const cv::Mat& src, cv::Mat& dst, int channel, double scale);
|
||
|
||
private:
|
||
Channel m_channel;
|
||
double m_scale;
|
||
};
|
||
|
||
#endif // !IMAGE_APPLY_CHANNEL_H
|