feat: add visualization for incorrect predictions - Githubissues

boostcampaitech6 / level1-imageclassification-cv-10

level1-imageclassification-cv-10 created by GitHub Classroom

1 stars 2 forks source link

feat: add visualization for incorrect predictions #15

Closed lukehanjun closed 8 months ago

lukehanjun commented 8 months ago

Explanation

Validation dataset 중 모델이 틀린 경우(예측값과 실제 라벨이 다른 경우)의 이미지들을 wandb를 이용해 로깅하는 코드입니다.
metrics.py, train.py, logger.py 파일을 수정했습니다.
우선 각 epoch마다 10개씩 이미지를 랜덤하게 뽑아 이미지와 모델의 예측값, 그리고 실제 값을 보여주도록 구현했습니다.

woohee-yang commented 8 months ago

깔끔하게 짜주셨네요!! 감사합니다!