개·고양이 얼굴 탐지

무엇을 하는 미션인가

반려동물 사진에서 개·고양이 얼굴 위치를 찾는 객체 검출 미션입니다. 어노테이션은 XML(bndbox)로 주어져, 파싱해서 학습 포맷으로 바꾸는 일부터 시작합니다.

두 가지 검출기

• SSD300 + VGG16 — ssd300_vgg16을 COCO 사전학습 가중치로 불러오고, SSDClassificationHead를 클래스 수에 맞게 교체했습니다. 입력은 300×300으로 리사이즈합니다
• YOLOv8 — Ultralytics로 학습했습니다

평가용으로 IoU와 mAP 계산 함수를 직접 작성한 코드도 노트북에 있습니다.

결과

노트북에 최종 성능 요약으로 남은 것은 YOLOv8 쪽입니다.

SSD300은 학습 코드까지 있지만 최종 성능 요약이 출력으로 남아 있지 않습니다. 그래서 두 검출기를 비교한 표는 만들지 않았습니다.

한계

• 위 수치는 YOLOv8 단독 결과이며, SSD와의 비교는 이 저장소로 확인되지 않습니다
• 단일 실행 결과입니다