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

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

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

 

[Private 19th] YOLOv8 + 5Fold WBF + Post process

공동작성자

stroke
2023.07.16 18:16 1,966 조회 language

# 학습방법
https://github.com/six-million/object-detection-for-synthetic-data
MMDetection과 YOLOv8을 모두 사용하려했으나 MMDetection(dino, cascade)은 0.5정도로 낮은 성능을 보여(epoch이 많이 부족했던 것으로 생각됩니다.) YOLOv8(YOLOv8x)만 사용했고 Recall이 1로 매우 높아 FP의 confidence를 줄이는 것을 목표로 진행했습니다.(자세한 내용은 README에 있습니다!)
- (512,1024)로 약 300epoch 정도 학습
- copyandpaste 사용
- StratifiedGroupKFold로 5개의 Fold로 split 해서 각각 학습
- WBF로 앙상블 진행
-  Post process 사용(bbox 넓이와 가로세로비 차이가 10% 이상인 예측의 confidence 2로 나누기)

# References
filot 님의 YOLOv8 코드를 참고했습니다.
https://dacon.io/competitions/official/236107/codeshare/8414?page=1&dtype=recent

코드
이전 글
이전 글이 존재하지 않습니다.
현재 글
[Private 19th] YOLOv8 + 5Fold WBF + Post process
대회 - 합성데이터 기반 객체 탐지 AI 경진대회
좋아요 6
조회 1,966
댓글 0
9달 전
다음 글
yolo8 코드입니다
대회 - 합성데이터 기반 객체 탐지 AI 경진대회
좋아요 42
조회 7,168
댓글 33
일 년 전