프로젝트 1 — 수업 전 실습 PC 점검 스크립트

import os
import cv2
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

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

# 요구사항 1: 세 패키지의 버전
print("[환경 점검]")
for name, module in [("OpenCV", cv2), ("NumPy", np), ("Matplotlib", matplotlib)]:
    print(f"  {name:11}{module.__version__}")

folder = "ch01_samples"
rows, images, failed = [], [], []

for name in sorted(os.listdir(folder)):
    path = os.path.join(folder, name)            # "ch01_samples/파일 이름" 꼴의 경로
    size_kb = round(os.path.getsize(path) / 1024, 1)
    # IMREAD_UNCHANGED: 저장된 그대로 읽는다. 흑백 그림은 채널 없이 (세로, 가로) 로 읽힌다
    img = cv2.imread(path, cv2.IMREAD_UNCHANGED)
    if img is None:
        # 요구사항 4: 실패한 파일은 표에 적고 다음 파일로 넘어간다
        rows.append({"파일": name, "가로": "-", "세로": "-", "채널": "-", "KB": size_kb, "상태": "읽기 실패"})
        failed.append(name)
        continue
    h, w = img.shape[:2]
    channels = 1 if img.ndim == 2 else img.shape[2]   # 요구사항 3: 2차원이면 흑백(채널 1)
    rows.append({"파일": name, "가로": w, "세로": h, "채널": channels, "KB": size_kb, "상태": "정상"})
    images.append((name, img))

# 요구사항 2: 사전 목록을 표로 바꿔 찍는다
print("[파일 점검표]")
print(pd.DataFrame(rows).to_string(index=False))

# 요구사항 4: 실패 개수와 이름
print(f"정상 {len(images)}개 · 읽기 실패 {len(failed)}개")
for name in failed:
    print("  확인할 파일:", name)

target_h = 160   # 미리보기에서 맞출 높이(칸)
tiles = []
for name, img in images:
    if img.ndim == 2:
        # 흑백(채널 1)을 컬러(채널 3)로: 세 채널에 같은 밝기를 넣어 이어 붙일 수 있게 한다
        img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
    h, w = img.shape[:2]
    new_w = round(w * target_h / h)   # 가로세로 비율을 지키며 높이를 160 에 맞춘다
    # cv2.resize(그림, (새 가로, 새 세로)): 크기를 바꾼다. 크기는 (가로, 세로) 순서로 준다
    tiles.append(cv2.resize(img, (new_w, target_h)))

# cv2.hconcat: horizontal concatenate. 높이와 채널이 같은 그림들을 가로로 붙인다
sheet = cv2.hconcat(tiles)
print("미리보기 모양:", sheet.shape)
print("저장:", cv2.imwrite("out_ch01_contact.png", sheet))

plt.figure(figsize=(9, 2.8))
plt.imshow(cv2.cvtColor(sheet, cv2.COLOR_BGR2RGB))   # 띄울 때는 RGB 로 바꾼다
plt.title("정상으로 읽힌 그림 " + " · ".join(name for name, _ in images))
plt.axis("off")
plt.show()