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

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

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

연습

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

제출

[Private 41위] efficientnetb3 + 배경 제거 전처리 + 앙상블

JEONMINJU

공동작성자

2023.02.15 14:55 1,271 조회 language

Remove background library 를 활용해서 배경 제거, 흑백, 그린, 노이즈, 블러 처리 5가지 데이터셋에 대해 efficientnetb3 모델을 활용해 학습시킨 후 , 각 데이터셋 별 모델에 대해 확률값을 도출, 5,3,1,1,1 가중치를 곱해 최종 분류를 진행했습니다.
코드는 학습한 데이터셋 중 하나인 그린에 대한 학습 과정입니다. 감사합니다.

코드