Replacing pixels in matplot image

Question

I have two images as NDArray's

The original image
A prediction mask.

I would like to 'white out' the areas where the mask is not 6 for example

To keep things simple I've paste them below as tiny 3x3 images with each cell being the the RGB value of the pixel

Original

[
    [1,1,1], [1,5,1], [1,1,1]
    [3,3,3], [3,3,3], [3,3,3]
    [1,1,1], [5,2,1], [1,1,1]
]

The Prediction

[
    [0, 0, 0]
    [6, 6, 6]
    [1, 2, 3]
]

To do this I am just looping through the prediction and replacing cells in the original with [0,0,0] to white out the ones I don't want

for rowIndex, predictedPointRow in enumerate(predict):
    for colIndex, predPoint in enumerate(predictedPointRow):
        if predPoint is not 6:
            img[rowIndex][colIndex] = [0, 0, 0]

This is painfully slow however. Is there a better way to do it?

Thanks,

sehigle · Accepted Answer · 2018-12-17 16:41:19Z

1

You could do something like

img = np.array([[[1,1,1], [1,5,1], [1,1,1]],[[3,3,3], [3,3,3], [3,3,3]],[[1,1,1], [5,2,1], [1,1,1]]])
predict = np.array([[0,0,0],[6,6,6],[1,2,3]])
img[predict!=6] = [0,0,0]

answered Dec 17, 2018 at 16:41

sehigle

3321 silver badge6 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Jan Kuiken · Accepted Answer · 2018-12-17 16:44:22Z

1

You can make use of Boolean or “mask” index arrays :

mask = (predict != 6)   # create a 2D boolean array, which can be used for indexing
img[mask] = [0,0,0]

answered Dec 17, 2018 at 16:44

Jan Kuiken

2,02018 silver badges20 bronze badges

Collectives™ on Stack Overflow

Replacing pixels in matplot image

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related