합성데이터 기반 객체 탐지 AI 경진대회

안녕하세요. 점박이물범팀입니다.

저희 팀은 mmdetection을 사용하여 여러 모델(rtmdet, grid-rcnn 등)을 테스트 해봤습니다.

그 결과 YOLOv8이 충분히 성능이 좋다고 생각하여 사용하게 되었습니다.

저희 팀의 최종 제출 코드는  filot님께서 공유해주신 코드를 이용하여 학습했습니다. filot님의 코드 공유에 감사의 말씀드립니다!


저희 팀의 규칙은 다음과 같았습니다.
    1. YOLOv8n을 사용하여 학습하고 결과 확인
    2. 1번의 결과를 바탕으로 점수가 잘 나오는 방법을 택하여 YOLOv8x로 학습


1번을 진행했을 때 더 높은 점수가 나온 방법이 있었지만, YOLOv8x 모델을 학습하는 컴퓨터의 이슈로 인해 진행하지 못했습니다.

YOLOv8n 테스트 시 더 높은 점수가 나온 방법
- pre-train 모델을 사용하는 경우
- mix-up Augmentation의 prob을 높이는 경우

위의 두 가지 방법을 사용한다면 저희 모델보다 더 나은 모델을 학습할 수 있을 것이라 생각합니다.

감사합니다.

다시 한 번 filot님의 코드 공유에 감사의 말씀드립니다!