[python] # 6 . Numpy이용한 코드로 픽셀 값 찾는 방법 두 가지! np.unique, np.where

안녕하세요. 욱찡입니다

 

이번에는 이미지에서 코드로 픽셀 값 찾는 방법을 소개해드리려고 합니다.

 

우선 이미지를 불러옵니다.

 

import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('rubi_512.jpg',0)
print(img.shape) #(512,512)

(왼쪽) 기본 이미지                                                                                     (오른쪽) graycale로 불러온 이미지

# np.unique

그리고 픽셀값들이 뭐가 있는지 확인하기 위해서 np.unique를 사용합니다.

 

import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('rubi_512.jpg',0)
print(img.shape) #(512,512)
print(np.unique(img))

#[  0   1   2   3   4   5   6   7   8   9  10  11  12  13  14  15  16  17
#  18  19  20  21  22  23  24  25  26  27  28  29  30  31  32  33  34  35
#  36  37  38  39  40  41  42  43  44  45  46  47  48  49  50  51  52  53
#  54  55  56  57  58  59  60  61  62  63  64  65  66  67  68  69  70  71
#  72  73  74  75  76  77  78  79  80  81  82  83  84  85  86  87  88  89
#  90  91  92  93  94  95  96  97  98  99 100 101 102 103 104 105 106 107
# 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
# 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143
# 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
# 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
# 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197
# 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215
# 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
# 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
# 252 253 254 255]

제 픽셀 값들입니다. 나중에 특정 상황에서 이미지 전처리 하실때에 np.unique를 활용해서 값들이 뭐가 있는지 확인해야 합니다.

 

# np.where

np.where은 픽셀 값들이 어디에 존재하는지를 알 수 있습니다.

만약 픽셀 값이 1일때 어디어디에 위치해 있는지 확인해보겠습니다.

 

import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('rubi_512.jpg',0)
print(img.shape) #(512,512)
print(np.where(img==1)) # (array([265], dtype=int64), array([308], dtype=int64))

308,265 에서 픽셀 값이 1입니다.

 

 

픽셀값이 1인 위치는 딱하나인데, x축 308, y축 265에서 픽셀 값이 1인걸 확인할수 있었습니다.

import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('rubi_512.jpg',0)
print(img.shape) #(512,512)
print(np.where(img==4)) # (array([267, 288, 305], dtype=int64), array([ 38, 324, 344], dtype=int64))

픽셀값이 4인 위치는 총 3개로 (38,267) , (324, 288), (344,305) 에서 확인이 가능합니다.

 

픽셀값 확인은 나중에 중요한 작업이므로 알아두면 편합니다 :)

 

감사합니다.