[ML] matplotlib로 이미지 보여주기

한장

Image.open으로 Image 객체로 받아오고,

np.array로 array로 만들고

plt.imshow로 보여주기.

img_path = f'어쩌구.jpg'
img = np.array(Image.open(img_path))
 
plt.figure(figsize=(16,8))
plt.imshow(img)

 

4장 묶어서

 

plt.subplots함수로 만들어주고, 

각각에 imshow와 title 달아주기

tight_layout으로 완성

n_rows, n_cols = 2, 2

fig, axes = plt.subplots(n_rows, n_cols, sharex=True, sharey=True, figsize=(8, 8))

axes[0][0].imshow(img)
axes[0][0].set_title(f'Original Image')

trfm = Compose([
            HorizontalFlip(p=1.0),
        ], p=1.0)
axes[0][1].imshow(trfm(image=img)['image'])
axes[0][1].set_title(f'Horizontal Flip')

trfm = Compose([
            VerticalFlip(p=1.0),
        ], p=1.0)
axes[1][0].imshow(trfm(image=img)['image'])
axes[1][0].set_title(f'Vertical Flip')

trfm = Compose([
            HorizontalFlip(p=1.0),
            VerticalFlip(p=1.0),
        ], p=1.0)
axes[1][1].imshow(trfm(image=img)['image'])
axes[1][1].set_title(f'Horizontal Flip & Vertical Flip')

plt.tight_layout()