图像转换函数使用python并打开

Question

最近我开始学习 opencv 和 python 进行图像处理。我在编写函数时遇到了问题。

给我的任务如下：

用python写一个函数，打开一张彩色图片，把图片转换成灰度图。

你需要编写一个带有两个参数的函数 color_grayscale(filename,g)：

  a. filename: a color image (Test color image is in folder “Task1_Practice/test_images”. Pick first image to perform the experiment.)

  b. g: an integer

如果 g = 1，程序的输出应该是灰度图像，否则应该是彩色图像。

我写的代码如下：

import cv2

def color_grayscale(filename,g):

    filename = cv2.imread("a15.jpg")
    " Enter Value of g:"
    if g == 1:
        gray = cv2.cvtColor(filename, cv2.COLOR_BGR2GRAY)
        img = cv2.imshow("gray",gray)
    else:
        img = cv2.imshow("original",filename)
    return(img)

color_grayscale("a15.jpg",1)

运行时的代码没有任何输出。

Answer 1

cv2.imshow 后面应该跟 waitKey 函数，它会在指定的毫秒内显示图像。否则，它不会显示图像。例如waitKey(0) 将无限显示窗口，直到任何按键（适用于图像显示）。 waitKey(25) 将显示一帧 25 毫秒，之后显示将自动关闭。 （如果你把它放在循环中读取视频，它会逐帧显示视频）

在返回img之前添加cv2.waitKey(0)就可以显示灰度图了