농업 환경 변화에 따른 작물 병해 진단 AI 경진대회

AI hub 작물데이터 사용해서 pretrain 모델 만들기

2022.01.22 15:02 1,996 조회

AI hub 에 올라온 시설작물데이터 중 저희대회에 나오는 작물 6가지를 골라 300GB 정도의 data 를 받았습니다.

이미지넷 pretrain이 아니라 작물데이터로 pretrain 된 모델을 만들고자 한건데, 1 epoch 만에 f1 score 가 거의 만점에 가깝게 됐었습니다. 하지만 downstream task 인 저희 데이터로 fine-tuning 해보니 결과가 엉망이네요..

아직 살펴볼건 많긴 하지만, 혹시 외부데이터 활용해서 모델 훈련시도해보신 분 계신가요? 해보셨다면 성과를 얻으셨나요?