[카테고리:] <span>OpenCV in Colab</span>

안녕하세요. 언제나휴일에 언휴예요.

영상 처리를 위한 기술로 OpenCV는 잘 알려져 있죠.

여기에서는 Colab 환경에서 Python  언어로 OpenCV 기술 사용하는 것을 알아볼 거예요.

Colab은 구글에서 제공하는 가상 환경입니다. 별도의 Python 개발 환경을 구축하지 않고 사용할 수 있습니다.

1. 버전 확인

먼저 버전을 확인합시다.

OpenCV를 사용하기 위해 cv2를 import합니다.

cv2의 __version__ 속성을 통해 버전 정보를 확인할 수 있어요.

import cv2
cv2.__version__

실행 결과

4.1.2

2. 구글 드라이브 마운트 시키기

Colab에서 작업하다보니 이미지 파일 등을 로딩하여 사용할 때 자신의 컴퓨터 폴더에 파일을 사용하는 것에 제약이 있습니다.

이러한 제약은 구글 드라이브를 마운트하여 사용하여 극복할 수 있습니다.

Colab 에서 왼쪽 메뉴에서 파일을 선택하세요.

Colab의 파일 메뉴를 선택한 후에 드라이브 마운트 메뉴를 선택하면 구글 드라이브와 연동할 수 있어요.

다음은 드라이브 마운트 선택 후에 자신의 구글 드라이브와 경로를 연동하고 있는 모습입니다.

3. 이미지 로드

OpenCV를 사용하기 위해 cv2를 import합니다.

google의  colab에서 이미지를 시각화하기 위해 cv2_imshow를 import해야 합니다.

import cv2
from google.colab.patches import cv2_imshow

이미지를 로딩을 위해 드라이브 마운트한 파일을 하나 선택하기로 할게요.

경로는 마운트한 파일을 선택하여 왼쪽 마우스 버튼을 클릭하면 경로를 복사할 수 있어요.

path = '/content/drive/MyDrive/files/eh.jpg'

이미지를 로딩할 때 cv2의 imread 메서드를 사용합니다.

cv2.imread(path, flag)

flag –

cv2.IMREAD_COLOR – COLOR로 읽어 옮

cv2.IMREAD_GRAYSCALE – 흑백으로 읽어 옮

cv2.IMREAD_UNCHANGED – 원래 이미지에 맞게 읽어 옮

image = cv2.imread(path,cv2.IMREAD_COLOR)

로딩한 이미지를 시각화할 때 cv2_imshow 함수를 사용합니다.

cv2_imshow(image)

전체 소스 코드

import cv2
from google.colab.patches import cv2_imshow
path = '/content/drive/MyDrive/files/eh.jpg'

image = cv2.imread(path,cv2.IMREAD_COLOR)
cv2_imshow(image)

4. 웹 이미지 로딩

이번에는 웹 상에 이미지를 시각화 해 봅시다.

웹 서버에 요청하기 위해 urllib.request를 import 합니다.

OpenCV에서 이미지는 배열로 표현합니다. numpy 모듈도 import 할게요.

import cv2
import numpy as np
from google.colab.patches import cv2_imshow
import urllib.request

웹 상에 있는 이미지 주소를 입력 인자로 로딩하는 메서드를 정의합시다.

def load_online_image(url):

먼저 urlib.request의 urlopen 메서드로 웹 서버에 미지를 요청합니다.

  resp = urllib.request.urlopen(url)

요청 결과를 읽어와서 bytearray로 변환 후에 이를 입력 인자로 numpy 배열을 생성합니다. 배열의 데이터는 0~255값을 표현할 수 있는 unit8로 지정할게요.

  image = np.asarray(bytearray(resp.read()), dtype='uint8')

이미지를 decoding한 후에 반환합니다.

  image = cv2.imdecode(image, cv2.IMREAD_COLOR)
  return image

다음은 load_online_image 함수를 정의한 코드입니다.

def load_online_image(url):
  resp = urllib.request.urlopen(url)
  image = np.asarray(bytearray(resp.read()), dtype='uint8')
  image = cv2.imdecode(image, cv2.IMREAD_COLOR)
  return image

이제 온라인 상에 있는 이미지를 로딩하는 코드를 작성하여 확인합시다.

표시하고자 하는 온라인 상의 이미지 경로를 복사하여 이를 로딩하여 시각화해 봅시다.

image_url ='https://t1.daumcdn.net/cfile/tistory/996D34345BBEAD5117'
image = load_online_image(image_url)
cv2_imshow(image)

전체 코드

import cv2
import numpy as np
from google.colab.patches import cv2_imshow
import urllib.request

def load_online_image(url):
  resp = urllib.request.urlopen(url)
  image = np.asarray(bytearray(resp.read()), dtype='uint8')
  image = cv2.imdecode(image, cv2.IMREAD_COLOR)
  return image
image_url ='https://t1.daumcdn.net/cfile/tistory/996D34345BBEAD5117'
image = load_online_image(image_url)

cv2_imshow(image)