P Stage 1 : Project Day 8
Day 8 list
- K fold 5개로 각 250 epoch으로 앙상블 진행
- mask, age, gender 각각 훈련 모델 코딩
1번은 큰 내용 없으므로 2번으로 넘어갑니다.
Mask, Age, Gender 각각 훈련 모델
이전에 Day 4에서 언급한 모델의 각각 분리 형태 분류를 시도해봤으나 성과가 그렇게 좋지는 못했습니다.
그러던 중 떠오른게 그냥 각 모델이 csv파일 내보내면 그걸 합치면 되지 않을까?라는 생각으로 각각으로 k-fold 학습을 하는 앙상블 모델을 코딩했습니다.
위 그림을 1개의 모델에서 처리를 했더니 모델 자체의 부담도 너무 크고 여러가지로 속도면에서도 메리트가 없어서 각각 결과물을 도출하게 코딩을 했습니다.
문제는 일요일에 사진을 wandb로 전달해서 제대로 맞추지 못하는 유형을 파악한 결과 연령을 잘 맞추지 못하는 경향이 강했습니다.
특히 연령대 구분인 30대 부근과 60대 부근의 이미지가 들어오면 정확도가 낮아졌으므로 나이를 조금 더 세분화할 필요가 있다고 생각했습니다.
새롭게 설계한 모델은 연령 구간을 총 10구간으로 세분화하여 모델이 학습하고 세분화된 구간을 다시 큰 구간의 원본 구간으로 변환하는 것입니다. 이렇게하면 분절된 구간의 값을 좀 더 세분화하게 분석이 가능할 것으로 보입니다.
현재 큰 학습을 진행중이라 현재 진행중인 학습 종료 후 테스트 가능할 것으로 보입니다.
Comments powered by Disqus.