이전 글에서 저장한 .mat 파일을 Python 환경에서 호출하고 출력해보자.

Import Libraries

.mat 파일을 Python에서 사용하려면 scipy라이브러리를 사용해야한다.

import scipy.io
import numpy as np
import matplotlib.pyplot as plt

Load Data

scipy.io.loadmat 메소드를 통해서 저장된 .mat 파일을 Load한다.

data= scipy.io.loadmat('savetest.mat')

Data 확인

현재 필자가 저장한 데이터는 초음파 데이터가 저장된 mat 파일이다.
데이터가 잘 불러와졌는지 shape로 확인해보자.

#print shape 
print(imgs_P.shape) # PData
print(imgs_Q.shape) # QData
print(imgs_I.shape) # ImgDataI
print(imgs_R.shape) # RCVData
print(imgs.shape) # ImgData

데이터가 정상적으로 저장 및 호출된 것을 확인할 수 있다.

데이터 시각화

불러온 데이터를 plt 라이브러리를 통해 시각화해보자.

imgs_P_plt = imgs_P[:, : , :, 0]

# log compression
imgs_P_plt = 20* np.log(imgs_P_plt)
plt.imshow(imgs_P_plt, cmap='gray')

불러온 데이터가 시각화까지 잘되는 것을 확인했다.

애니메이션 출력

import matplotlib.animation as animation
from IPython.display import HTML

fig = plt.figure(figsize=(10,5))
ims = []
for i in range(0, 10):
    im = plt.imshow(20*np.log(imgs_P[:,:,:,i]), animated=True, cmap='gray')
    
    ims.append([im])
    
ani = animation.ArtistAnimation(fig, ims, interval=50, blit=True, repeat_delay=10)
HTML(ani.to_jshtml())

# gif 파일로 저장
ani.save('imgs_P.gif', writer='pillow', fps=10)