서울 랜드마크 이미지 분류 AI 해커톤

[Private 14th] vgg16 전이학습 및 데이터 증강활용

2022.09.13 15:51 1,485 조회 language

배운 지식을 활용할 곳이 필요했는데 대회를 열어주셔서 분들께 감사드립니다.
efficient 와 Resnet, vgg를 사용해 전이학습을 해보았고 vgg16이 가장 좋은 결과가 나왔습니다.
다른 분들은 Resnet을 활용하셨던데 초기값이나 gradient가 어느 방향에서 시작하는지가 중요한 요소 중 하나인 것 같습니다.
또한 데이터 셋이 크지않아 기본적인 데이터 증강을 활용하여서 100회 정도 훈련시켰습니다.
(colab으로 진행시켜 중간에 다운 되는 것을 막기 위해 나누어서 훈련했습니다)

코드