실습 1 — 폴더 점검: 깨진·작은·채널이 다른 파일 찾기

import os
import pandas as pd

folder = "ch16_raw"
# 정답표(answer.csv)는 사진이 아니므로 뺀다
names = sorted(n for n in os.listdir(folder) if n != "answer.csv")

rows = []
for name in names:
    path = os.path.join(folder, name)
    rows.append({
        "파일": name,
        "클래스": name.split("_")[0],               # 이름의 앞부분이 라벨이다
        "KB": round(os.path.getsize(path) / 1024, 1),
    })
table = pd.DataFrame(rows)
print("파일 수:", len(table))
# 클래스마다 몇 개인지 센다
print(table["클래스"].value_counts().sort_index().to_dict())
# 가장 작은 파일 다섯 개: 크기가 이상하게 작으면 의심한다
print(table.sort_values("KB").head(5).to_string(index=False))

import cv2

def inspect(path):
    """파일 하나를 저장된 그대로 읽어 가로, 세로, 채널 수를 돌려준다. 못 읽으면 None."""
    # IMREAD_UNCHANGED: 흑백은 (세로, 가로), 투명 채널이 있으면 (세로, 가로, 4) 로 읽는다
    img = cv2.imread(path, cv2.IMREAD_UNCHANGED)
    if img is None:
        return None
    h, w = img.shape[:2]
    channels = 1 if img.ndim == 2 else img.shape[2]
    return w, h, channels

shapes = [inspect(os.path.join(folder, n)) for n in table["파일"]]
table["가로"] = [s[0] if s else "-" for s in shapes]
table["세로"] = [s[1] if s else "-" for s in shapes]
table["채널"] = [s[2] if s else "-" for s in shapes]
# 읽지 못했거나, 채널이 3 이 아니거나, 짧은 변이 64 보다 작은 줄만 본다
odd = [s is None or s[2] != 3 or min(s[0], s[1]) < 64 for s in shapes]
print(table[odd].to_string(index=False))

MIN_SIDE = 64   # 짧은 변이 이보다 작으면 너무 작다(이 장의 기준)

def check_file(path):
    """정제의 첫 관문: (문제, 처리) 를 돌려준다. 앞에서 걸리면 뒤는 보지 않는다."""
    if os.path.getsize(path) == 0:
        return "빈 파일", "버림"
    shape = inspect(path)
    if shape is None:
        return "깨짐", "버림"
    w, h, channels = shape
    if min(w, h) < MIN_SIDE:
        return "너무 작음", "버림"
    if channels == 1:
        return "흑백", "고침"
    if channels == 4:
        return "투명 채널", "고침"
    return "없음", "그대로"

results = [check_file(os.path.join(folder, n)) for n in table["파일"]]
table["문제"] = [r[0] for r in results]
table["처리"] = [r[1] for r in results]
print(table.loc[table["문제"] != "없음", ["파일", "KB", "문제", "처리"]].to_string(index=False))
print(table["처리"].value_counts().to_dict())

import matplotlib.pyplot as plt

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

def to_bgr(img):
    """채널 수가 1 이나 4 인 그림을 BGR 세 채널로 맞춘다."""
    if img.ndim == 2:
        return cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)     # 같은 밝기를 세 채널에
    if img.shape[2] == 4:
        return cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)     # 넷째(투명도) 채널을 뗀다
    return img

fixed = {}
for name in table.loc[table["처리"] == "고침", "파일"]:
    raw = cv2.imread(os.path.join(folder, name), cv2.IMREAD_UNCHANGED)
    fixed[name] = (raw, to_bgr(raw))
    print(f"{name}: {raw.shape} → {fixed[name][1].shape}")

fig, axes = plt.subplots(1, 4, figsize=(11, 2.8))
for k, (name, (raw, out)) in enumerate(fixed.items()):
    # 원본은 채널 수에 따라 띄우는 법이 다르다: 흑백은 cmap, 투명은 BGRA → RGBA
    if raw.ndim == 2:
        axes[2 * k].imshow(raw, cmap="gray", vmin=0, vmax=255)
    else:
        axes[2 * k].imshow(cv2.cvtColor(raw, cv2.COLOR_BGRA2RGBA))
    axes[2 * k].set_title(f"{name} 원본")
    axes[2 * k + 1].imshow(cv2.cvtColor(out, cv2.COLOR_BGR2RGB))
    axes[2 * k + 1].set_title("세 채널로 맞춤")
for ax in axes:
    ax.axis("off")
plt.tight_layout()
plt.show()