Python OpenCV (1) 기본 이미지 다루기

OpenCV는 컴퓨터 비전을 목적으로 한 라이브러리이다.

파이썬은 pip install opencv-python 명령어로 설치할 수 있다.

 

import cv2로 패키지를 불러온다.

cv2는 BGR로 색상을 읽지만 RGB로 변환하려면 cv2.cvtColor(img, cv2.COLOR_BGR2RGB) 를 사용한다.

이미지는 y, x (height, width)로 선택하고 cv2 함수는 x, y (width, height)이고 반대이다.

기본적인 이미지 다루기

import cv2
import os

# 디렉터리 경로 변경
# os.chdir("ChangeDirectory_Path")
# 디렉터리 경로 확인
# print(os.getcwd())

# 이미지 경로
image_path = "TempImage.jpg"

# 이미지 읽기
img = cv2.imread(image_path)

# 이미지 구성 height, width, 픽셀 색상 채널
print(img.shape) # y, x, channel

# 이미지 복사
cv2.imwrite("copy_img.jpg", img)

# 이미지 보기
cv2.imshow("Title", img)  # 제목 표시
cv2.waitKey() # 키 입력 대기

# 그레이 변환 컨버터
# rgb_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # RGB로 변환

grey_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# cv2.imshow("", rgb_img)
cv2.imshow("g", grey_img)
cv2.waitKey()

# 색으로 보기
(B, G, R) = cv2.split(img)
color = R
cv2.imshow("", color)
cv2.waitKey()

zeros = np.zeros(img.shape[:2], dtype="uint8")
print(img.shape[:2])

cv2.imshow("Red", cv2.merge([zeros, zeros, R])) # R로 채움
cv2.imshow("Green", cv2.merge([zeros, G, zeros])) # G로 채움
cv2.imshow("Blue", cv2.merge([B, zeros, zeros])) # B로 채움
cv2.waitKey(0)

# 픽셀 값 접근
print(img[100, 200]) # (y, x) 크기 B G R
cv2.imshow("", img)
cv2.waitKey()
cv2.destroyAllWindows()

# 크기 조절
cv2.imshow("", img)

img = cv2.resize(img, (400, 300)) # x, y (width, height)
cv2.imshow("big", img)

img = cv2.resize(img, (100, 50))
cv2.imshow("small", img)
cv2.waitKey()

# 자르기
cv2.imshow("change", img[100:150, 50:100]) # ymin:ymax, xmin:xmax
h, w, c = img.shape

cv2.imshow("crop", img[int(h/2 - 50): int(h/2 + 50), int(w/2 - 50): int(w/2 + 50)])
print(int(h/2 - 50), int(h/2 + 50), int(w/2 - 50), int(w/2 + 50))
cv2.waitKey()