[python] 이미지에 도형과 선 그리기

이미지에 도형과 선을 그리는 것은 파이썬에서 매우 간단한 작업입니다. PIL (Python Imaging Library)이나 OpenCV 같은 이미지 처리 라이브러리를 사용하여 이미지에 도형과 선을 그릴 수 있습니다. 이번 예제에서는 PIL을 사용하여 이미지에 도형과 선을 그리는 방법을 소개하겠습니다.

PIL 설치하기

PIL은 파이썬 이미지 처리를 위한 강력한 라이브러리입니다. 아래 명령어를 사용하여 PIL을 설치할 수 있습니다.

pip install Pillow

이미지에 사각형 그리기

다음은 이미지에 사각형을 그리는 예제입니다. 사각형은 PIL.ImageDraw.Draw.rectangle() 함수를 사용하여 그릴 수 있습니다.

from PIL import Image, ImageDraw

# 이미지 열기
image = Image.open('image.jpg')

# 이미지에 그릴 도구 생성
draw = ImageDraw.Draw(image)

# 사각형 그리기
draw.rectangle(((100, 100), (200, 200)), outline='red')

# 이미지 저장
image.save('image_with_rectangle.jpg')

위 예제에서는 Image.open() 함수를 사용하여 이미지를 열고, ImageDraw.Draw() 함수를 사용하여 그릴 도구를 생성합니다. 그리고 rectangle() 함수를 호출하여 사각형을 그립니다. outline 매개변수를 사용하여 사각형의 테두리 색상을 지정할 수 있습니다. 마지막으로 이미지를 저장합니다.

이미지에 선 그리기

이미지에 선을 그리는 것도 PIL.ImageDraw.Draw.line() 함수를 사용하여 간단하게 할 수 있습니다.

from PIL import Image, ImageDraw

# 이미지 열기
image = Image.open('image.jpg')

# 이미지에 그릴 도구 생성
draw = ImageDraw.Draw(image)

# 선 그리기
draw.line(((100, 100), (200, 200)), fill='blue', width=5)

# 이미지 저장
image.save('image_with_line.jpg')

위 예제에서는 line() 함수를 호출하여 선을 그립니다. fill 매개변수를 사용하여 선의 색상을 지정하고, width 매개변수를 사용하여 선의 두께를 지정할 수 있습니다.

결론

이번 예제에서는 PIL을 사용하여 이미지에 도형과 선을 그리는 방법을 알아보았습니다. PIL은 파이썬에서 간단하고 편리하게 이미지 처리 작업을 수행할 수 있는 라이브러리입니다. 추가로 다른 도형을 그리거나 선의 스타일을 변경하는 등 다양한 작업을 할 수 있습니다. PIL 문서를 참조하여 더 많은 기능을 알아보세요.

참고 자료