在OpenCV Python中设置像素值

时间:2018-07-04 07:37:48

标签: python opencv

如何快速更改像素值?在C#中,我只需要使用GetPixel()来获取像素值,并使用SetPixel()来更改它(它很容易使用,但速度慢,MarshallCopy和Lock / UnlockBits更快)。

在此代码中,我将黑色像素标记为1,将白色像素标记为0

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

path = tkFileDialog.askopenfilename()
bmp = cv2.imread(path) #reading image
height, width, channels = bmp.shape

if channels == 3:
    bmp = cv2.cvtColor(bmp, cv2.COLOR_BGR2GRAY) #if image have 3 channels, convert to BW
    bmp = bmp.astype('uint8')

bmp = cv2.adaptiveThreshold(bmp,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C,\
            cv2.THRESH_BINARY,11,2) #Otsu thresholding

imageData = np.asarray(bmp) #get pixels values

pixelArray = [[0 for y in range(height)] for x in range(width)] #set size of array for pixels

for y in range(len(imageData)):
    for x in range(len(imageData[0])):
        if imageData[y][x] == 0:
            pixelArray[y][x] = 1 #if black pixels = 1
        else:
            pixelArray[y][x] = 0 #if white pixels = 0

在c#中,它看起来可能像这样:

for (y = 0; y < bmp.Height-1; y++)
            {
                for (x = 0; x < bmp.Width-1; x++)
                {
                    if (pixelArray[y, x] == 1)
                        newImage.SetPixel(x, y, Color.Black); //printing new bitmap
                    else
                        newImage.SetPixel(x, y, Color.White);
                }
            }
            image2.Source = Bitmap2BitmapImage(newImage);

在下一步中,我将countour像素标记为“ 2”,但是现在我想问你,如何根据我的特定值在python中设置新图像,然后显示它?出于实验目的,我只想按字节值反转图像(从黑白到黑白)。你能帮我怎么做吗?

EDIT1

我想我找到了一个解决方案,但是我有一个通道带有GREYSCALE图像(我认为这是当我使用cv2.cvtColor将3个通道图像转换为灰度图像时的工作方式)。像这样的功能:

im[np.where((im == [0,0,0]).all(axis = 2))] = [0,33,166]

可以很好地工作,但是如何使该功能与灰度图像一起工作?我想将一些黑色像素(0)设置为白色(255)

1 个答案:

答案 0 :(得分:0)

对于单通道图像(灰度图像),请使用以下内容:

首先创建灰色图像的副本:

gray_2 = gray.copy()

现在将黑色像素分配为白色:

gray_2[np.where(gray == 0)] = 255