실습 4 — 주석 검사기로 실수 찾기

from pathlib import Path
import cv2
import pandas as pd

ann = pd.read_csv("ch17_raw/annotations.csv")
# 그림마다 (세로, 가로) 를 사전에 담는다. 범위 검사에 그림 크기가 필요하다
sizes = {p.name: cv2.imread(str(p)).shape[:2] for p in sorted(Path("ch17_raw").glob("*.png"))}

print("주석 줄:", len(ann), "· 그림 파일:", len(sizes), "장")
print("클래스 이름:", ann["class"].value_counts().to_dict())
print("그림 크기 종류:", set(sizes.values()))

CLASSES = ["bolt", "nut", "washer"]

def check_row(r):
    """주석 한 줄(사전)을 검사해 실수 갈래의 목록을 돌려준다."""
    if r["filename"] not in sizes:                     # 짝 검사: 그림 파일이 없다
        return ["그림 없는 주석"]
    h, w = sizes[r["filename"]]
    x1, y1, x2, y2 = r["x1"], r["y1"], r["x2"], r["y2"]
    found = []
    if r["class"] not in CLASSES:                      # 이름 검사
        found.append("모르는 클래스")
    if x1 > x2 or y1 > y2:
        found.append("뒤집힌 상자")
    if x1 == x2 or y1 == y2:
        found.append("크기 0 상자")
    if min(x1, x2) < 0 or min(y1, y2) < 0 or max(x1, x2) > w or max(y1, y2) > h:
        found.append("범위 밖")
    return found

problems = []
# to_dict("records"): 표를 줄마다 사전 하나인 목록으로 바꾼다(열 이름 class 를 그대로 쓰려고)
for row_no, r in enumerate(ann.to_dict("records")):
    for kind in check_row(r):
        problems.append({"kind": kind, "row": row_no, "filename": r["filename"]})
        print(f"{row_no:3}번 줄 {r['filename']} {r['class']:6} ({r['x1']}, {r['y1']}, {r['x2']}, {r['y2']}) → {kind}")

# duplicated(): 앞에 모든 칸이 같은 줄이 있으면 True. 둘째부터 중복으로 친다
for row_no in ann.index[ann.duplicated()]:
    problems.append({"kind": "중복 줄", "row": int(row_no), "filename": ann.loc[row_no, "filename"]})

# 폴더에는 있는데 주석에 한 번도 나오지 않는 그림
for name in sorted(set(sizes) - set(ann["filename"])):
    problems.append({"kind": "주석 없는 그림", "row": "", "filename": name})

found = pd.DataFrame(problems)
print(found["kind"].value_counts().to_dict())
found.to_csv("out_ch17_problems.csv", index=False, encoding="utf-8-sig")
print("찾은 실수", len(found), "건을 out_ch17_problems.csv 에 저장")

answer = pd.read_csv("ch17_raw/mistakes.csv", dtype={"row": str}).fillna("")

# (갈래, 줄 번호, 파일) 세 칸을 묶어 집합으로 만들어 견준다
def key_set(df):
    return {(k, str(r), f) for k, r, f in zip(df["kind"], df["row"], df["filename"])}

mine, truth = key_set(found), key_set(answer)
print("정답의 실수:", len(truth), "건")
print("맞힘:", len(mine & truth), "· 놓침:", len(truth - mine), "· 헛짚음:", len(mine - truth))
for item in sorted(truth - mine):
    print("  놓친 것:", item)
for item in sorted(mine - truth):
    print("  헛짚은 것:", item)