[Computer Vision] 그리기 함수

👀 본 예제는 Window10의 VSCode, Python3.11.0로 작성되었습니다.

 

"직선"을 그리는 함수는 "line" 메서드를 사용하면 된다.

import cv2
import numpy as np

# 흰배경의 빈 이미지 생성(512 x 512)

white_background = np.ones((512,512),dtype=np.uint8) * 255

# 직선
cv2.line(
    img=white_background,
    pt1=(0,0),
    pt2=(256,256),
    color=100,
    thickness=3,
    lineType=cv2.LINE_4,
    shift=0)
cv2.imshow("line",white_background)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• pt1 : 시작 좌표
• pt2 : 끝 좌표
• color : 색
• thickness : 선 굵기
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)
• shift : 그리기 좌표값의 축소 비율

 

"화살표 직선"은 "arrowedLine" 메서드를 사용하면 된다.

import cv2
import numpy as np

# 흰배경의 빈 이미지 생성(512 x 512)

white_background = np.ones((512,512),dtype=np.uint8) * 255

# 화살표
cv2.arrowedLine(
    img=white_background,
    pt1=(0,0),
    pt2=(256,256),
    color=100,
    thickness=3,
    line_type=cv2.LINE_AA,
    shift=0,
    tipLength=0.2
)
cv2.imshow("arrow",white_background)

cv2.waitKey(0)
cv2.destroyAllWindows()

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• pt1 : 시작 좌표
• pt2 : 끝 좌표
• color : 색
• thickness : 선 굵기
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)
• shift : 그리기 좌표값의 축소 비율
• tipLength : 화살표 모양의 길이

 

 

"마커"은 "drawMarker" 메서드를 사용하면 된다.

import cv2
import numpy as np

# 흰배경의 빈 이미지 생성(512 x 512)

white_background = np.ones((512,512),dtype=np.uint8) * 255

# 마커
cv2.drawMarker(
    img=white_background,
    position=(50,50),
    color=0,
    markerType=cv2.MARKER_DIAMOND,
    markerSize=20,
    thickness=3,
    line_type=cv2.LINE_8
)
cv2.imshow("Marker",white_background)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• position : 그릴 좌표
• color : 색
• markerType : 마커의 종류(cv2.MARKER_DIAMONE, ...)
• markerSize : 마커의 크기
• thickness : 선 굵기
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)

 

 

 

"사각형"은 "rectangle"을 사용한다.

import cv2
import numpy as np

background = np.ones((512,512,3),dtype=np.uint8) * 255 # BGR 채널

# 사각형
cv2.rectangle(
    img=background,
    pt1 = (50,50),
    pt2 = (250,250),
    color=(255,0,0),
    thickness=1,
    lineType=cv2.LINE_AA,
    shift=0
)
cv2.imshow("rectangle",background)

cv2.waitKey(0)
cv2.destroyAllWindows()

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• pt1 : 시작 좌표
• pt2 : 끝 좌표
• color : 색
• thickness : 선 굵기
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)
• shift : 그리기 좌표값의 축소 비율

 

 

"원"은 "circle"을 사용한다.

import cv2
import numpy as np

background = np.ones((512,512,3),dtype=np.uint8) * 255 # RGB 채널

cv2.circle(
    img=background,
    center=(256,256),
    radius=50,
    color=(0,255,0),
    thickness=3,
    lineType=cv2.LINE_8,
    shift=0
)
cv2.imshow("circle",background)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• center : 중심점 좌표
• radius : 반지름
• color : 색
• thickness : 선 굵기
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)
• shift : 그리기 좌표값의 축소 비율

 

 

"타원"은 "ellipse"를 사용한다.

import cv2
import numpy as np

background = np.ones((512,512,3),dtype=np.uint8) * 255 # RGB 채널

cv2.ellipse(
    img=background,
    center=(256,256),
    axes=(170,100),
    angle=0,
    startAngle=0,
    endAngle=360,
    color=(0,255,100),
    thickness=3,
    lineType=cv2.LINE_AA,
    shift=0
)
cv2.imshow("ellipse",background)

cv2.waitKey(0)
cv2.destroyAllWindows()

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• center : 중심점 좌표
• axes : (x반지름, y반지름)
• angle : 타원 회전 각도(x축 기준, 시계 방향)
• startAngle : 타원 호의 시작 각도(x축기준, 시계방향)
• endAngle : 타원 호의 끝 각도(y축 기준, 시계 방향)
• color : 색
• thickness : 선 굵기
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)
• shift : 그리기 좌표값의 축소 비율

 

 

"다각형"을 그릴때는 "polylines"를 사용한다.

import cv2
import numpy as np

background = np.zeros((512,512,3),dtype=np.uint8) # RGB 채널
points = np.array([[[10,10],[170,10],[200,230],[70,70],[50,150]]],dtype=np.int32)

cv2.polylines(
    img=background,
    pts=points,
    isClosed=True,
    color=(255,255,0),
    thickness=2,
    lineType=cv2.LINE_AA,
    shift=0
)
cv2.imshow("polylines",background)
cv2.waitKey(0)
cv2.destroyAllWindows()

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• pts : 폴리곤 좌표 배열
• isClosed : True 시 닫힌 도형
• color : 색
• thickness : 선 굵기
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)
• shift : 그리기 좌표값의 축소 비율

 

"볼록 다각형"을 그릴때는 "fillConvexPoly"를 사용한다.

import cv2
import numpy as np

background = np.zeros((512,512,3),dtype=np.uint8) # RGB 채널

points = np.array([[[10,10],[170,10],[200,230],[70,70],[50,150]]],dtype=np.int32)

# 볼록 다각형
cv2.fillConvexPoly(
    img=background,
    points=points,
    color=(102,100,24),
    lineType=cv2.LINE_8,
    shift=0
)
cv2.imshow("fill Poly",background)

cv2.waitKey(0)
cv2.destroyAllWindows()

 

매개변수는 다음과 같다.

• img : 그릴 np.array
• points : 폴리곤 좌표 배열
• axes : (x반지름, y반지름)
• color : 색
• lineType : 선 타입(cv2.LINE_4, cv2.LINE_8, cv2.LINE_AA)
• shift : 그리기 좌표값의 축소 비율