해당 포스팅은 네이버 부스트캠프 P-stage를 위해 개인 학습 내용을 정리했습니다.
1. 목표
- Dataset.py, train.py 수정하기
- Resnet50 사용해보기
- 데이터 증강해서 돌려보기
2. 진행 내용
1) Dataset.py, train.py 수정하기
Dataset에 클래스 부족한 것들의 데이터를 추가하기 위한 함수를 추가했습니다. train.py의 training 과정을 일부 수정했습니다.
2) Resnet50 사용해보기
Resnet을 수렴하게 하기 위해 StepLR을 적용해봤습니다. 생각보다 3 Epoch마다 변경해주도록 하니, 잘 수렴했습니다. 하지만, 결과는 생각보다 좋지 않았습니다.
3) 데이터 증강해서 돌려보기
데이터 증강을 위해 Dataset에 클래스 별로 이름을 list에 추가하는 코드를 넣었습니다. resnet50으로 돌렸고, 결과는 내일 나올 볼 수 있습니다.
3. 회고
처음에 seed값을 정해놓지 않아서 관리가 잘 안되서 아쉽습니다. 데이터 증강한 결과가 어떻게 나올지 기대가 됩니다.
