이미지 분류 해커톤: 데이터 속 아이콘의 종류를 맞혀라!

데이콘 해커톤 | 알고리즘 | 입문 | 비전 | 분류 | Accuracy

  • moneyIcon 상금 : 데이스쿨 프로 구독권
  • 2025.03.04 ~ 2025.04.30 09:59 + Google Calendar
  • 744명 마감

 

Private 2위, resnet50 모델, auto augment 기법

2025.04.30 15:44 568 조회 language

https://github.com/DeepVoltaire/AutoAugment

TIMM 에서 pretrained 되지 않은 resnet 50 을 사용했고, augmentation 기법으로 Autoaugment 논문의 cifar-10 코드를 수정해서 사용했습니다. AutoML 이 금지되어 있는 것으로 아는데, 제가 사용한 코드는 cifar-10 augmentation 범위를 지정해서 랜덤으로 augment 해주는 코드지, AutoML 은 아닌 거 같더군요 (논문은 이러한 범위 (policy) 를 AutoML 처럼 찾는 거지만).
 
Autoaugment 제외하고는 5 fold 에 TTA, soft-voting 을 평범하게 적용했습니다.

코드
로그인이 필요합니다
0 / 1000
EISLab_이희원
2025.05.01 01:29

(참고) AutoAugment는 다음 코드로 간소화할 수 있습니다.
transforms.AutoAugment(
        policy=transforms.AutoAugmentPolicy.CIFAR10
    ),