MMDetection + WandB로 학습상황 확인하기 (+ 버전 3 방법 추가)

 

 

WandB

= Weights and Bias

실험 추적과 시각화, 모델 체크포인트 저장, 협업 등을 쉽게 할 수 있는 라이브러리 입니다.

 

 

 

 

👆시작 방법 

1. wandb 회원가입

https://kr.wandb.ai/

Weights & Biases – Developer tools for ML

 

 

 

2. wandb 설치 및 로그인

pip install wandb

wandb login

 

wandb: Paste an API key from your profile and hit enter, or press ctrl+c to quit: 

라고 나오면 

 

아래 그림처럼 API key를 복사해서 붙여넣기 합니다.

 

 

 

기본 세팅 완료!🎉

 

 

 

 

3. mmdetection에서 wandb 추가하기

 

Config파일에 아래의 코드를 추가합니다.

라면,

 

그 파일에 들어가서 아래의 코드를 추가해줍니다.

# 기존 코드

log_config = dict(
    hooks=[
        dict(type='WandbLoggerHook')
    ]
)

 

(추가설명)

이는 /mmdet/core/hook/wandblogger_hook.py에서 wandb 코드가 구현되어 있기 때문에
log_config에 추가만 해주면 간단하게 사용할 수 있는 원리입니다.

 

 

 

 

 

이제 파일을 다시 실행하면 실행부분에서

 

이렇게 링크가 형성됩니다!

 

 

 

 

결과

 

이렇게 실험 과정을 볼 수 있습니다.

 

 

 

---

MMDetection 3.x 버전

https://mmdetection.readthedocs.io/en/latest/migration/config_migration.html?highlight=wandb#logging-configuration

 

사용 방법이 오른쪽 그림처럼 바뀌었습니다.

 

 

 

config 파일에 아래 코드를 입력해주니 잘 되었습니다!

vis_backends = [
    dict(type='LocalVisBackend'),
    dict(type='WandbVisBackend',
         init_kwargs={
            'project': 'mmdetection',
            'group': 'faster-rcnn_r50-caffe_fpn_ms-1x_coco'
         })
]
visualizer = dict(
    type='DetLocalVisualizer',
    vis_backends=vis_backends,
    name='visualizer')

 

 

 

 

 

 

 

앞으로 wandb를 더 잘 활용하는 방법은 차차 공부해야겠네요😁