写一手好的C++代码 ,利用Strategy模式写的

9 篇文章 0 订阅
3 篇文章 0 订阅
<span style="font-size:32px;">策略设计模式就是将算法封装在类中,比如我们见到的进行一个二值化算法;</span>

#include <opencv2/core/core.hpp>
#include<opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>

using namespace cv;
using namespace std;

class ColorDetector{

    private:

        int minDist;
        Vec3b target;
        Mat result;

    public:

    ColorDetector():minDist(100){
        target[0] = target[1] = target[2] = 0;
    }
    //设置色彩距离阈值
    void setColorDistanceThreshold(int distance){
        if (distance < 0)
            distance = 0;
        minDist = distance;
    }
    //get set方法设置Private里面的变量
    int getColorDistanceThreshold()const{
        return minDist;
    }
    //设置需检测的颜色
    void setTargetColor(unsigned char red, unsigned char green, unsigned char blue){
        target[2] = red;
        target[1] = green;
        target[0] = blue;
    }
    void setTargetColor(Vec3b color){
        target = color;
    }
    Vec3b getTargetColor()const{
        return target;
    }

    //计算与目标颜色的距离,采用的是街区距离
    int getDistance(const Vec3b& color) const{
        return abs(color[0] - target[0]) +
            abs(color[1] - target[1]) +
            abs(color[2] - target[2]);
    }

    Mat ColorDetector::process(const Mat &image){
        result.create(image.rows,image.cols, CV_8U);
        Mat_<Vec3b>::const_iterator it = image.begin<Vec3b>();
        Mat_<Vec3b>::const_iterator itend = image.end<Vec3b>();
        Mat_<uchar>::iterator itout = result.begin<uchar>();
        for (; it != itend; ++it, ++itout){
            if (getDistance(*it) < minDist){
                *itout = 255;
            }
            else{
                *itout = 0;
            }
        }
        return result;
    }
};

int main(){
    //创建对象
    ColorDetector cdector;
    Mat image = imread("boldt.jpg");
    if (!image.data){ return 0; }
    cdector.setTargetColor(130, 190, 230);
    namedWindow("result");
    imshow("result", cdector.process(image));
    waitKey(0);
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值