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

알고리즘 | 비전 | 객체 탐지 | mAP

상금 : 1,000 만원
1,670명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

private 24th 솔루션

ili0820

2023.06.25 12:04 1,470 조회 language

yolov8을 사용하여 3060으로 200 epoch 정도 학습시켰습니다.
학습 당 약 하루 정도 소요 되었습니다. (3060 사용)
주피터 노트북으로 하지 않고 .py 로 진행 한 내용을 옮긴 것이라 바로 정상작동 하지 않을 수 있을 듯합니다.
yolov8, yolov6 등의 4가지의 모델을 최종적으로 wbf 하여 제출하였습니다.
이미지 사이즈의 경우 gpu 여건상 640~1280 사이 값을 실험 하였는데, 640일때 오히려 속도도 빠르고 성능이 괜찮게 나와 이를 사용하였습니다.

코드