파이썬 이미지에 도형 그리기

import cv2
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

필요한 기본 모듈 임포트

blank_img = np.zeros(shape=(512,512,3),dtype=np.int16)

색상채널3개로 놓고 512*512 프레임에 0으로 가득채우니 검은색이 나올것이다

plt.imshow(blank_img)

cv2.rectangle(blank_img,pt1=(284,0),pt2=(500,150),color = (0,255,0),thickness=10)

꼭짓점 x=284,y=0, 두번째 꼭짓점 512면 경계 넘어갈수도 있음

도형을 밑으로 조금 내려주기 위해 150을 입력한다. 색은 초록색을 넣어주고 선은 기본단색 두꼐 10 이미지에 직접 적용할것임

plt.imshow(blank_img)

왼쪽 상위 꼭짓점이 (284,0) 오른쪽 하위 꼭짓점이 (500,150)에 있다 

숫자를 다르게 변형해서 정사각형, 직사각형을 만든다. 

cv2.rectangle(blank_img,pt1=(200,200),pt2= (300,300),color=(0,0,255),thickness=10)

삼각형을 하나 더그려줍니다. 색깔은 파란색으로 설정

plt.imshow(blank_img)

이렇게 사각형이 두개 그려짐

cv2.circle(img = blank_img,center=(100,100),radius=50,color=(255,0,0),thickness=8)

중심 x100 y 100 원의 반지름 radius=50, -1은 채워진 빨간색이다

plt.imshow(blank_img)

cv2.circle(img = blank_img,center=(400,400),radius=50,color=(255,0,0),thickness=-1)

빨간원을 (400,400) 꼭짓점을 기준으로 지름 50의 채워진 빨간색원을 그려줌

plt.imshow(blank_img)

cv2.line(blank_img,pt1=(0,0),pt2=(512,512),color=(102,255,255),thickness=5)
plt.imshow(blank_img)

꼭짓점 (0,0)에서 (512,512) 까지 선을 그려줌

font = cv2.FONT_HERSHEY_SIMPLEX
cv2.putText(blank_img,text='Hello',org=(10,500),fontFace=font,fontScale=4,color=(255,255,255),thickness=3,lineType = cv2.LINE_AA)
plt.imshow(blank_img)

글씨를 하단에 그려보기 위해 폰트 설정을 하고, 안에 쓸 문자도 설정해준다.

org은 문자열의 왼쪽 아래 모서리 어딘가에 있게한다. color =(255,255,255)는 흰색이다.

출처: udemy x KMOOC openCV 딥러닝을 이용한 computer vision  파이썬