闭合曲线内涂色code(辅助制作Ground truth)

import cv2
import numpy as np
from matplotlib import pyplot as plt

path = 'C:\\Users\\Ashraf\\Desktop\\ground\\'

def red_bw2d(image):
    img = image
    r, c, _ = img.shape
    for i in range(r):
        for j in range(c):
            if img[i, j][2] > img[i, j][0] and img[i, j][2] > img[i, j][1]:
                img[i, j][0] = 255
                img[i,j][1] = 255
                img[i,j][2] = 255
            else:
                img[i, j][0] = 0
                img[i, j][1] = 0
                img[i, j][2] = 0
    img = img[:, :, 1]
    _, bin = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY)

    return bin

def green_bw2d(image):
    img = image
    r, c, _ = img.shape
    for i in range(r):
        for j in range(c):
            if img[i, j][1] > img[i, j][0] and img[i, j][1] > img[i, j][2]:
                img[i, j][0] = 255
                img[i,j][1] = 255
                img[i,j][2] = 255
            else:
                img[i, j][0] = 0
                img[i, j][1] = 0
                img[i, j][2] = 0
    img = img[:, :, 1]
    _, bin = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY)

    return bin

def blue_bw2d(image):
    img = image
    r, c, _ = img.shape
    for i in range(r):
        for j in range(c):
            if img[i, j][0] > img[i, j][1] and img[i, j][0] > img[i, j][2]:
                img[i, j][0] = 255
                img[i,j][1] = 255
                img[i,j][2] = 255
            else:
                img[i, j][0] = 0
                img[i, j][1] = 0
                img[i, j][2] = 0
    img = img[:, :, 1]
    _, bin = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY)

    return bin

def denoise(image):
    img = image
    img = cv2.medianBlur(img, 3)
    return img

def fill(image, name):
    point = (0,0)
    src = image.copy()
    src = np.invert(src)
    connectivity = 4
    flags = connectivity
    flags |= cv2.FLOODFILL_FIXED_RANGE
    cv2.floodFill(src, None, point, (0, 0, 0), (100,) * 3, (100,) * 3, flags)
    cv2.imwrite(name, src)

def function(img, name):
    img = denoise(img)
    img = red_bw2d(img)
    img = denoise(img)
    fill(img, name)

# Main
img = cv2.imread('./3.png', cv2.IMREAD_COLOR)
name = './test.jpg'
function(img, name)