본문 바로가기
코딩공부/Machine Learning

딥러닝을 위한 기초적인 이미지 다루기

by Dong_Devlog 2022. 11. 16.

본격적으로 딥러닝을 배우기에 앞서 이미지 다루는 방법에 대해 학습하자.

프로그래밍 언어는 Python을 사용하고 딥러닝 라이브러리는 OpenCV 패키지를 사용할 것 이다.

 

패키지 불러오기

OpenCV 패키지를 설치한 후에,

pip install opencv-python

 import cv2를 통해 패키지를 불러온다. (cv2는 OpenCV의 패키지 이름)

import cv2

 

이미지 처리 기초

  • 이미지 불러오기

cv2.imread로 이미지를 불러와 img에 저장(test.jpg 경로 설정 주의!!)

import cv2

img = cv2.imread('test.jpg')

 

  • 이미지 형태 보기

shape로 이미지에 대한 형태를 확인

print(img.shape) # (400, 600, 3) = (높이, 너비, 채널)

 

  • 이미지 미리보기

cv2.imshow('띄우는 창의 이름', 불러올 이미지)로 이미지 미리보기

waitKey로 1을 눌렀을 때 창이 닫힘

cv2.imshow('img_show', img)
cv2.waitKey(1)

 

  • 이미지에 도형 그리기

cv2.rectangle(img, 왼쪽상단좌표, 오른쪽하단좌표, 선 색상, 선 두께)

cv2.rectangle(img, pt1=(330, 5), pt2=(495, 260), color=(255, 0, 0), thickness=1)

 

  • 이미지 자르기

img[y축으로 5~260, x축으로 330~495]으로 자름 (y,x 순이니 주의!!)

cropped_img = img[5:260, 330:495]

 

  • 이미지 크기 변경

cv2.resize(불러올 이미지, (변경 할 사이즈)) 를 통해 이미지 크기를 변경

resized_img = cv2.resize(img, (512, 256))

 

댓글