[Python 데이터 분석과 이미지 처리] OpenCV를 활용한 이미지 흑백 처리

뽕칠이·2023년 12월 21일
0

OpenCV를 활용한 이미지 흑백 처리

OpenCV란

이미지를 python으로 읽어와 특정 처리를 거쳐서 결과를 만들어내고 다시 파일 형태로 저장 할 수 있도록 하는 라이브러리

기본 함수

  • cv2.imread(file_name, flag)

    이미지를 읽어 Numpy 객체로 만드는 함수
    flag : 이미지를 읽는 방법 설정
    -- IMREAD_COLOR : 이미지를 color로 읽고 투명한 부분은 무시
    -- IMREAD_GRAYSCALE : 이미지를 흑백으로 읽기
    -- IMREAD_UNCHANGED : 이미지를 color로 읽고 투명한 부분도 읽기

주의할 점: opencv는 BGR형태로 색깔을 읽음

  • cv2.imshow(title. image)

    특정한 이미지를 화면에 출력

  • cv2.imwrite(file_name, image)

    특정한 이미지를 파일로 저장하는 함수

  • cv2.cvtColor(image, flag)

    이미지의 색상 형태를 변경하는 함수

예제

import cv2
import matplotlib.pyplot as plt

# color로 이미지를 읽어옴
img_basic = cv2.imread('cat.jpg', cv2.IMREAD_COLOR)
plt.imshow(cv2.cvtColor(img_basic, cv2.COLOR_BGR2RGB))
plt.show()

# 흑백으로 이미지를 읽어옴
# BGR을 GRAY로
img_basic = cv2.cvtColor(img_basic, cv2.COLOR_BGR2GRAY)
# GRAY를 RGB로
plt.imshow(cv2.cvtColor(img_basic, cv2.COLOR_GRAY2RGB))
plt.show()

결과

  • color로 출력한 결과

  • 흑백으로 출력한 결과

0개의 댓글