실습 4 — 갤러리 폴더 한 번에 보기

import os
import cv2

folder = "ch02_gallery"
gallery = []                                     # (이름, 그림) 짝을 모은다
for file in sorted(os.listdir(folder)):          # 이름순으로 정렬해 차례를 고정한다
    img = cv2.imread(os.path.join(folder, file))
    if img is None:                              # 읽지 못한 파일은 건너뛴다
        continue
    name = os.path.splitext(file)[0]             # splitext: "bus.png" → ("bus", ".png")
    gallery.append((name, img))
    print(f"{name:9} {img.shape}")
print("읽은 그림:", len(gallery), "장")

import matplotlib.pyplot as plt

plt.rcParams["font.family"] = "Malgun Gothic"
plt.rcParams["axes.unicode_minus"] = False

# plt.subplots(줄, 칸): 칸 배열 axes 의 모양은 (2, 3)
fig, axes = plt.subplots(2, 3, figsize=(8, 4.6), layout="constrained")
print("axes 의 모양:", axes.shape)
# axes.flat 으로 칸을 차례로 꺼내 그림과 짝짓는다
for ax, (name, img) in zip(axes.flat, gallery):
    ax.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    ax.set_title(name)
    ax.axis("off")
fig.suptitle("ch02_gallery 의 그림 여섯 장")
plt.show()

import math

cols = 4                                         # 한 줄에 놓을 칸 수
rows = math.ceil(len(gallery) / cols)            # 6 ÷ 4 = 1.5 → 올림해서 2줄
print(f"그림 {len(gallery)}장 → {rows}줄 × {cols}칸 = {rows * cols}칸, 남는 칸 {rows * cols - len(gallery)}개")

fig, axes = plt.subplots(rows, cols, figsize=(9, 3.6), layout="constrained")
for k, ax in enumerate(axes.flat):
    if k < len(gallery):
        name, img = gallery[k]
        ax.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
        ax.set_title(f"{k + 1}. {name}")
    ax.axis("off")                               # 그림이 있는 칸도 없는 칸도 눈금을 지운다
plt.show()