Image Classification Hackathon: Guess the Type of Icons in the Data!

Dacon Hackathon | Algorithm | Begin | Vision | Classification | Accuracy

  • moneyIcon Prize : DASCHOOL Pro Subscription
  • 2025.03.04 ~ 2025.04.30 09:59 + Google Calendar
  • 744 Users Completed

 

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

2025.04.30 15:44 704 Views 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 을 평범하게 적용했습니다.

Code
Login Required
0 / 1000
EISLab_이희원
2025.05.01 01:29

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