• 周日. 10 月 6th, 2024

5G编程聚合网

5G时代下一个聚合的编程学习网

热门标签

opencv在python和c#中的对照应用-替换颜色

admin

11 月 28, 2021

先来python的,

# -*- coding:utf-8 -*-
import cv2 as cv
import numpy as np

def rgb2hsv(r, g, b):
    r, g, b = r/255.0, g/255.0, b/255.0
    mx = max(r, g, b)
    mn = min(r, g, b)
    m = mx-mn
    if mx == mn:
        h = 0
    elif mx == r:
        if g >= b:
            h = ((g-b)/m)*60
        else:
            h = ((g-b)/m)*60 + 360
    elif mx == g:
        h = ((b-r)/m)*60 + 120
    elif mx == b:
        h = ((r-g)/m)*60 + 240
    if mx == 0:
        s = 0
    else:
        s = m/mx
    v = mx
    H = h / 2
    S = s * 255.0
    V = v * 255.0
    return H, S, V

def fixHSVRange(h, s, v):
    # Normal H,S,V: (0-360,0-100%,0-100%)
    # OpenCV H,S,V: (0-180,0-255 ,0-255)
    return (180 * h / 360, 255 * s / 100, 255 * v / 100)


# Load the aerial image and convert to HSV colourspace
image = cv.imread("d:Screenshot1.png")
hsv=cv.cvtColor(image,cv.COLOR_BGR2HSV)


rgbs=[(254,214,105),(255,238,187),(151,196,252),(208,239,196),(232,157,224)
    ,(219,95,209),(13,95,152)]
for rgb in rgbs:
    pixel =rgb2hsv(rgb[0], rgb[1], rgb[2])
    brown_hi =  np.array([pixel[0] + 20, pixel[1] + 20, pixel[2] + 40])
    brown_lo =  np.array([pixel[0] - 20, pixel[1] - 20, pixel[2] - 40])
    mask=cv.inRange(hsv,brown_lo,brown_hi)
    image[mask>0]=(255,255,255)

rgbs=[(175,174,173),(170,173,167),(175,175,173),(237,236,229)]
for rgb in rgbs:
    pixel =rgb2hsv(rgb[0], rgb[1], rgb[2])
    brown_hi =  np.array([pixel[0] + 10, pixel[1] + 10, pixel[2] + 20])
    brown_lo =  np.array([pixel[0] - 10, pixel[1] - 10, pixel[2] - 20])
    mask=cv.inRange(hsv,brown_lo,brown_hi)
    image[mask>0]=(255,255,255)



cv.imshow('res',image)
cv.imwrite("d:Screenshot3.png", image)
# cv.imwrite("result.png",image)
cv.waitKey(0)
cv.destroyAllWindows()

View Code

c#的

        private double max(double x, double y, double z)
        {
            return (new double[] { x, y, z }).Max();
        }
        private double min(double x, double y, double z)
        {
            return (new double[] { x, y, z }).Min();
        }
        private double[] rgb2hsv(double r, double g, double b)
        {
            r = r / 255.0;
            g = g / 255.0;
            b = b / 255.0;
            double mx = max(r, g, b);
            double mn = min(r, g, b);
            double m = mx - mn;
            double h = 0;
            double s = 0;
            if (mx == mn)
                h = 0;
            else if (mx == r)
                if (g >= b)
                    h = ((g - b) / m) * 60;
                else
                    h = ((g - b) / m) * 60 + 360;
            else if (mx == g)
                h = ((b - r) / m) * 60 + 120;
            else if (mx == b)
                h = ((r - g) / m) * 60 + 240;
            if (mx == 0)
                s = 0;
            else
                s = m / mx;
            double v = mx;
            double H = h / 2;
            double S = s * 255.0;
            double V = v * 255.0;
            return new double[] { H, S, V };
        }


        private void btnReplaceColor_Click(object sender, EventArgs e)
        {
            var image = Cv2.ImRead(@"d:Screenshot1.png");
            new Window("DST", WindowMode.Normal, image);
            Cv2.WaitKey();

            byte[][] rgbs1 = { new byte[] { 254, 214, 105 },
                new byte[] { 255, 238, 187 }, 
                new byte[] { 151, 196, 252 },
                new byte[] { 208, 239, 196 }, 
                new byte[] { 232, 157, 224 } };

            byte[][] rgbs2 = {
                new byte[] { 175, 174, 173 },
                new byte[] { 170, 173, 167 },
                new byte[] { 175, 175, 173 },
                new byte[] { 237, 236, 229 } };

            using (Mat hsv = new Mat())
            {
                Cv2.CvtColor(image, hsv, ColorConversionCodes.BGR2HSV);

                foreach (var rgb in rgbs1)
                {
                    var pixel = rgb2hsv(rgb[0], rgb[1], rgb[2]);
                    Scalar brown_hi = new Scalar(pixel[0] + 20, pixel[1] + 20, pixel[2] + 40);
                    Scalar brown_lo = new Scalar(pixel[0] - 20, pixel[1] - 20, pixel[2] - 40);
                    Mat mask = new Mat();
                    Cv2.InRange(hsv, brown_lo, brown_hi, mask);
                    //image[mask > 0] = (255, 255, 255);
                    image.SetTo(new Scalar(255, 255, 255), mask);
                }

                foreach (var rgb in rgbs2)
                {
                    var pixel = rgb2hsv(rgb[0], rgb[1], rgb[2]);
                    Scalar brown_hi = new Scalar(pixel[0] + 10, pixel[1] + 10, pixel[2] + 20);
                    Scalar brown_lo = new Scalar(pixel[0] - 10, pixel[1] - 10, pixel[2] - 20);
                    Mat mask = new Mat();
                    Cv2.InRange(hsv, brown_lo, brown_hi, mask);
                    //image[mask > 0] = (255, 255, 255);
                    image.SetTo(new Scalar(255, 255, 255), mask);
                }

            }

            new Window("DST", WindowMode.Normal, image);
            Cv2.WaitKey();
        }

View Code

 

发表回复