et al/AI Tech 6

mmdetection에 wandb 연결

코딩상륙작전 2024. 1. 11. 03:01

configs/_base_/default_runtime.py

위 파일 경로에서 vis_backends를 다음과 같이 고친다.

 

mm환경설정 : 

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
vis_backends = [dict(type='LocalVisBackend'),
                #dict(type='TensorboardVisBackend'), tensorboard가 default이다.
                dict(type='WandbVisBackend',
                    init_kwargs={
                        'project': 'recycle_detection',
                        'group': 'sparse-rcnn'  # wandibi에 그룹으로 저장된다.
                    })]

설치 후에 완디비 api를 넣어준다.

https://wandb.ai/authorize