분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
계층적 분류와 사전 증강 기반 resnet, resnext 분류 모델 LB 69
이미지는 전체 학습 데이터셋의 색상 통계를 추적하여 배경 제거 후 컬러 증강을 하는 사전 증강을 적용했습니다.
수동으로 클래스 당 이미지 직접 확인하면서 브랜드와 국가 등으로 분류하여 1차 분류기를 지나
이 예측 결과와 이미지를 피쳐로 계층적 분류하는 아키텍쳐로 구성하였습니다.
두 모델 모두 acc는 검증 acc가 0.97 초 중반까지는 상승하나 log loss의 갭이 너무 커 높은 순위를 차지하지는 못한 것 같습니다.
추론 과정에서 폴드별 체크포인트를 SWA로 가중치를 평균 내어 온도 스케일링 최적값 튜닝을 한 과정이 가장 좋은 log loss를 추출할 수 있었습니다.
코드는 깃허브에 공유하였습니다. read me와 코드는 정리 중 입니다.
첨부는 증강 예시 본 입니다.
https://github.com/zzunnii/Hecto-AI-Challenge-CarClassification
데이콘(주) | 대표 김국진 | 699-81-01021
통신판매업 신고번호: 제 2021-서울영등포-1704호
직업정보제공사업 신고번호: J1204020250004
서울특별시 영등포구 은행로 3 익스콘벤처타워 901호
이메일 dacon@dacon.io |
전화번호: 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved