python-3.x - 计算机视觉图像遮蔽问题

Question

谁能帮我解决这个问题。当我使用此功能时，它适用于红色和黄色信号图像，但不适用于绿色。绿色信号图像全黑。知道出了什么问题。但令人惊讶的是，如果我让它 BGR2HSV 它显示绿色信号图像，但其他 2 个是黑色的。我正在使用 Matplotlib 导入图像，所以我猜默认情况下它是 RGB。

def 掩码(rgb_image) :

hsv_image = cv2.cvtColor(rgb_image, cv2.COLOR_RGB2HSV)

## mask of red color range 1
red_mask1 = cv2.inRange(hsv_image, (0,20,0), (10,255,255))

## mask of red color range 2
red_mask2 = cv2.inRange(hsv_image, (170,20,0), (180,255,255))

## mask of green
green_mask = cv2.inRange(hsv_image, (40,0,0), (80,255,255))

## mask of yellow
yellow_mask = cv2.inRange(hsv_image, (10,30,100), (30,255,255))

## final mask
mask1 = cv2.bitwise_or(red_mask1, red_mask2)
mask2 = cv2.bitwise_or(mask1, yellow_mask)
mask3 = cv2.bitwise_or(mask2, green_mask)

target = cv2.bitwise_and(rgb_image,rgb_image, mask=mask3)

plt.imshow(target)

用于读取图片的代码:

def load_dataset(image_dir):

    im_list = []
    image_types = ["red", "yellow", "green"]

    for im_type in image_types:
      for file in glob.glob(os.path.join(image_dir, im_type, "*")):
        im = mpimg.imread(file) 
        if not im is None:
            im_list.append((im, im_type))
    return im_list

Answer 1

它起作用了:我的绿色范围不正确。它应该是下限:(80,20,20)，上限:(170,255,255)。