[FIXED] How to filter a elements of matrixes?

Issue

I want to get some pixel values from an image which are not equal with a specified value. But I want to get back in RGB format and not as a long vector. How can I do it?

import cv2
import numpy as np

image = cv2.imread('image.jpg')
sought = [36,255,12]
result = image[image!=sought]

import sys
np.set_printoptions(threshold=sys.maxsize)
print(result)

And I've got:

  [20 103  75  21  98  70  16 100  72  18 101  73  19  97  69  15  95  66
  15  95  67  13 101  73  19 104  77  21  96  69  13  94  65   8  99  69
  14  98  68  13  94  63  10  88  66  24  92  69  24  92  67  23  93  67
  13  93  67  13  93  67  13  97  72  16  96  70  16  93  66  15  96  68
  .....
  99  69  14  96  66  11  91  67  25  88  65  20  92  68  14  96  69  18
  96  70  16  91  64  13  95  67  13  92  64  10  90  63]

But I want something like this:

     [[[R,G,B], [R,G,B], [R,G,B], [R,G,B]],
       ......
      [[R,G,B], [R,G,B], [R,G,B], [R,G,B]]]

What did I miss here?

Solution

If the wanted output is a list of pixels, then after the component-wise comparison, you must check what pixels differ on any of the R,G or B with .any(axis = 2):

image[(image != sought).any(axis=2)]

output of the form:

array([[ 22, 136, 161],
       [197, 141, 153],
       [173, 122,  65],
       [137, 189,  67],
             ...
       [166, 205, 238],
       [207,  99, 129],
       [ 44,  76,  97]])

Answered By - PlainRavioli