如何在OpenCV(Python)中将灰度图像转换为RGB,以便在处理二进制图像后可视化轮廓?

2024-04-16 19:04:10 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在学习使用OpenCV进行实时应用的图像处理。我对图像做了一些阈值处理,想用绿色标记轮廓,但它们没有用绿色显示,因为我的图像是黑白的。

在程序的早期,我使用了gray=cv2.cvtColor(frame,cv2.COLOR戋BGR2GRAY)将RGB转换为grayscale,但是返回时我感到困惑,函数backtorgb=cv2.cvtColor(gray,cv2.CV戋GRAY2RGB)给出了AttributeError:“module”对象没有属性“CV戋GRAY2RGB”。

下面的代码似乎没有用绿色绘制轮廓-这是因为它是灰度图像吗?如果是,我可以将灰度图像转换回RGB,使轮廓显示为绿色吗?

import numpy as np
import cv2
import time

cap = cv2.VideoCapture(0)
while(cap.isOpened()):

    ret, frame = cap.read()

    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    ret, gb = cv2.threshold(gray,128,255,cv2.THRESH_BINARY)

    gb = cv2.bitwise_not(gb)

    contour,hier = cv2.findContours(gb,cv2.RETR_CCOMP,cv2.CHAIN_APPROX_SIMPLE)

    for cnt in contour:
        cv2.drawContours(gb,[cnt],0,255,-1)
    gray = cv2.bitwise_not(gb)

    cv2.drawContours(gray,contour,-1,(0,255,0),3)

    cv2.imshow('test', gray)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

Tags: 图像importrgbcv2framecvcolor轮廓
3条回答

如果你把你的图像转换成灰度,你就回不来了。你已经从三个频道转到一个频道,当你试着回去时,三个号码都是一样的。所以简短的回答是不你不能回去。引发此错误的原因是backtorgb函数需要采用以下格式:

CvtColor(input, output, CV_GRAY2BGR)

OpenCV使用BGR而不是RGB,因此如果您修复了顺序,它应该可以工作,尽管您的图像仍然是灰色的。

我将我的评论提升为一个答案:

简单的方法是:

You could draw in the original 'frame' itself instead of using gray image.

困难的方法(你试图实现的方法):

backtorgb = cv2.cvtColor(gray,cv2.COLOR_GRAY2RGB)是正确的语法。

试试这个:

import cv2
import cv

color_img = cv2.cvtColor(gray_img, cv.CV_GRAY2RGB)

我在使用opencv时发现,有些常量是在cv2模块中定义的,而另一些则是在cv模块中定义的。

相关问题 更多 >