25.10.29 개발일지(MFC 프로젝트 진행내용)

 

25.10.29 개발일지

[MFC 프로젝트 진행내용]

 

 

pt 파일 학습

상단찌그러짐 : top_dent

뚜껑없음 : top_no_tap

측면찌그러짐 : side_dent

스크래치 : side_foreign

 

AI 서버 코드 수정

LABEL_MAP = {
    # 불량 클래스
    "top_dent": "상단찌그러짐",
    "top_no_tap": "뚜껑없음",
    "side_dent": "측면찌그러짐",
    "side_foreign": "스크래치",

    # 정상 클래스
    "top_normal": "정상(상단)",
    "side_normal": "정상(측면)"
}

# 카메라별 불량 라벨 기준
DEFECT_LABELS_TOP = {"상단찌그러짐", "뚜껑없음"}
DEFECT_LABELS_SIDE = {"측면찌그러짐", "스크래치"}

 

: pt 파일 class 이름과 맞게 코드 수정

# 불량 판정 로직
    defect_detected = False
    THRESHOLD = 0.8  # 신뢰도 기준

    if camera_type == "top":
        defect_detected = any(b["label"] in DEFECT_LABELS_TOP and b["score"] >= THRESHOLD for b in bboxes)
    elif camera_type == "side":
        defect_detected = any(b["label"] in DEFECT_LABELS_SIDE and b["score"] >= THRESHOLD for b in bboxes)

 

"result": "defective" if defect_detected else "normal",

 

: 단순히 불량을 판단하는 코드에서

TOP: 뚜껑없음 or 상단찌그러짐이 탐지되면 불량

SIDE: 측면찌그러짐 or 스크래치가 탐지되면 불량 으로 판별

results = model.predict(
    source=img_rgb,
    imgsz=640,
    conf=0.4,
    iou=0.45,
    device=DEVICE,
    verbose=False
)

 

: conf 부분 수정해서 불량 감지 민감도를 조절