matplotlib 3D에 이미지 띄우기

matplotlib에 이미지를 띄우고 싶다면? 이미지를 imread 등으로 로드한 다음 imshow(image)를 호출하면 된다.

1
2
3
4
5
6
7
8


import matplotlib.pyplot as plt

fig, ax = plt.subplots()
ax.set_xlim([0, 1036])
ax.set_ylim([0, 1036])

img = plt.imread("image.png")
ax.imshow(img, extent=[0, 1036, 0, 1036])

2d 결과

extend=는 이미지맵의 범위를 설정해준다.

하지만 3D 프로젝션에서는 다음과 같은 오류가 발생한다.

1
2
3
4
5
6


import matplotlib.pyplot as plt

fig, ax = plt.subplots(subplot_kw={"projection": "3d"})

img = plt.imread("image.png")
ax.imshow(img, extent=[0, 1036, 0, 1036])

NotImplementedError: Axes3D currently only supports the aspect argument 'auto'. You passed in 'equal'.

에러 메시지대로 aspect=가 문제인가 싶어 auto를 넘겨도 보았지만 원하는대로 작동하지 않는다.

해결책

해결책은 imshow 대신 numpy의 ogrid와 plot_surface의 facecolors= 인자를 사용하는 것.

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21


import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots(subplot_kw={"projection": "3d"})

# 축 제한
ax.set_xlim([0, 1036])
ax.set_ylim([0, 1036])
ax.set_zlim([0, 500])

img = plt.imread("image.png")
X, Y = np.ogrid[0:img.shape[0], 0:img.shape[1]]

img = img[-Y,X] # 이미지 뒤집기

Z = np.zeros_like(X)

# rstride=와 cstride=로 화질을 조정할 수 있다. 
ax.plot_surface(X, Y, Z, rstride=5, cstride=5, facecolors=img)

plt.show()

3d 결과

matplotlib 3D에 이미지 띄우기

해결책

관련 글: