Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEAT] wandb 연결 & 시각화, Develop/Font_Classifier -> master #7

Merged
merged 4 commits into from
Jan 10, 2023

Conversation

jerry-ryu
Copy link
Contributor

Overview

  • 용이한 학습 및 결과 공유를 위해 font classifier wandb 연결
    image
  • validation set과 train set을 stratified하게 split
  • validation set 시각화
  • inf. 클래스 확률 추가 출력
    image

Change Log

model/font_classifier/dataset_font.py

전체 이미지에서 val_ratio 비율로 뽑아오던 val_set을 각 폰트별 이미지 파일에서 각각 val_ratio 비율로 뽑아오도록 변경하였음

model/font_classifier/train_font.py

  1. wandb 연결
  2. validation set이 wandb artifact에 나타나도록 시각화(이미지, label, pred, 각 클래스 확률 logging)
  3. val, train stratified split을 적용

model/font_classifier/inference_font.py

출력 결과에 예측 확률 추가


To Reviewer

  • train_font.py를 실행하고, wandb artifact에 들어가서 stratified split과 시각화가 잘 되었는지 확인하기
  • inference_font.py에 예측 확률이 추가되었는지 확인하기

Issue Tag

boostcampaitech4lv23cv2/final-project-level2-cv-11#2

@jerry-ryu jerry-ryu requested a review from jane79 January 9, 2023 18:15
@jane79 jane79 merged commit beb7f96 into master Jan 10, 2023
@jerry-ryu jerry-ryu changed the title [FEAT] wandb 연결 & 시각화, Develop/font classifier -> master [FEAT] wandb 연결 & 시각화, Develop/Font_Classifier -> master Jan 11, 2023
@jerry-ryu jerry-ryu deleted the Develop/Font_Classifier branch January 20, 2023 02:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants