포디블록 구조 추출 AI 경진대회

알고리즘 | 비전 | 분류 | Accuracy

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

 

[Private 42위] Efficientnet-b4 + cross-val + 배경마스크

2023.02.13 20:57 1,156 조회 language

안녕하세요 훌륭한 모델들이 많아서 많이 배웠습니다. 대부분 시도하셨을 것 같아 특이사항만 간략하게 말씀드리면

Efficientnet-b4 와 cross-validation 을 통한 앙상블 모델을 만들었습니다
배경을 제거하기 위해 cv2.inRange 를 사용하였고
Dataset을 불러올때 Augmentation된 배경과 이미지를 따로 불러와 학습하여 일반화 성능을 높이고자 하였습니다

코드