[Python] OpenCV 기초2 - 이미지 위에 그림그리기

1. 빈 스케치북 그리기

import cv2
import numpy as np

# 세로 46 * 가로 640, 3 Channel(RGB)에 해당하는 스케치북
img = np.zeros((460,640,3), dtype = np.uint8)
# img[:] = (255,255,255) # 전체 공간 흰색으로 채우기
# print(img)

cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

numpy를 활용하면 빈 스케치북의 사이즈를 설정할 수 있고, 

지정 영역의 색상을 지정할 수도 있다. 

 

2. 일부 영역 색칠

import cv2
import numpy as np

# 세로 46 * 가로 640, 3 Channel(RGB)에 해당하는 스케치북
img = np.zeros((460,640,3), dtype = np.uint8)
img[100:200,200:300] = (255,255,255)
#세로 영역, 가로영역

cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

특정 영역을 지정하여 색상을 변경할 수 있음

 

3. 직선 그리기

cv2.line([이미지], [시작점], [끝점], [색상], [직선종류])

OpenCV에서 지원하는 직선 종류

1. cv2.LINE_4 : 상하좌우 4방향으로 연결된 선
2. cv2.LINE_8 : 대각선 포함한 8방향으로 연결된 선(기본값)
3. cv2.LINE_AA :  부드러운 선(anti-aliasing)

import cv2
import numpy as np

# 세로 46 * 가로 640, 3 Channel(RGB)에 해당하는 스케치북
img = np.zeros((460,640,3), dtype = np.uint8)

COLOR = (0,255,255) #BGR : yellow
THICKNESS = 3 #두께

cv2.line(img,(50,100),(400,50),COLOR,THICKNESS,cv2.LINE_8)
cv2.line(img,(50,200),(400,150),COLOR,THICKNESS,cv2.LINE_4)
cv2.line(img,(50,300),(400,250),COLOR,THICKNESS,cv2.LINE_AA)
# 그림 위치, 시작점, 끝 점, 색깔 두께, 선 종류


cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

LINE_4와 LINE_8이 눈에 띄게 차이나지는 않지만,

LINE_4는 상하 좌우 4방향으로 연결되어 있고, 

LINE_8은 대각선을 포함한 8방향으로 연결되어져 있다. (default값)

 

4. 원 그리기

cv2.circle([이미지], [중심점], [반지름], [색상], [두께], [선 종류])

import cv2
import numpy as np

# 세로 46 * 가로 640, 3 Channel(RGB)에 해당하는 스케치북
img = np.zeros((460,640,3), dtype = np.uint8)

COLOR = (255,255,0) #BGR : 옥색
RADIUS = 50 # 반지름
THICKNESS = 10 #두께

cv2.circle(img,(200,100),RADIUS,COLOR,THICKNESS,cv2.LINE_AA) # 속이 빈 원
# 그릴 위치, 원의 중심, 반지름, 색깔, 두께, 선 종류

cv2.circle(img,(400,100),RADIUS,COLOR,cv2.FILLED,cv2.LINE_AA) # 속이 꽉 찬 원


cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

위치, 중심, 반지름, 채움 여부 설정

 

5. 사각형 그리기

cv2.rectangle([이미지], [왼쪽 위], [오른쪽 아래], [색상], [두께])

import cv2
import numpy as np

# 세로 46 * 가로 640, 3 Channel(RGB)에 해당하는 스케치북
img = np.zeros((460,640,3), dtype = np.uint8)

COLOR = (0,255,0) #BGR : 초록색
THICKNESS = 3 #두께

cv2.rectangle(img,(100,100),(200,200),COLOR,THICKNESS) # 속이 빈 사각형
# 그릴 위치, 왼쪽 위, 오른쪽 아래, 색깔, 두께
cv2.rectangle(img,(300,100),(400,300),COLOR,cv2.FILLED) # 속이 빈 사각형



cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

 

6. 다각형 그리기

cv2.polylines([이미지], [그릴 좌표들], [닫힘 여부], [두께], [선 종류])

import cv2
import numpy as np

# 세로 46 * 가로 640, 3 Channel(RGB)에 해당하는 스케치북
img = np.zeros((460,640,3), dtype = np.uint8)

COLOR = (0,0,255) #BGR : 빨간색
THICKNESS = 3 #두께

pts1 = np.array([[100,100] , [200,100], [100,200]])
pts2 = np.array([[200,100], [300,100], [300,200]])

#cv2.polylines(img, [pts1], True, COLOR,THICKNESS, cv2.LINE_AA)
#cv2.polylines(img, [pts2], True, COLOR,THICKNESS, cv2.LINE_AA)
cv2.polylines(img, [pts1,pts2], True, COLOR,THICKNESS, cv2.LINE_AA) # 속이 빈 다각형
# 그릴 위치, 그릴 좌표들, 닫힘 여부, 색깔, 두께, 선 종류

pts3 = np.array([[[100,300], [200,300], [100,400]], [[200,300], [300,300], [300,400]]])
cv2.fillPoly(img, pts3, COLOR, cv2.LINE_AA) # 꽉 찬 다각형
# 그릴 위치, 그릴 다각형, 색깔, 선 종류


cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()