mmyolo wandb 연동

[nanpuhaha]

mmyolo 공식문서에는 WandbVisBackend가 있는데, mmdet에서처럼 MMDetWandbHook이 없다.

visualizer = dict(vis_backends=[dict(type='LocalVisBackend'), dict(type='WandbVisBackend')])

WandbVisBackend를 사용하면 각자 프로젝트가 새로 생성되면서 기록되어서, 하나의 프로젝트에 기록할 수가 없는 문제가 있다.

mmdet에서 WandbLoggerHook을 바탕으로 MMDetWandbHook를 만들었듯이, MMYoloWandbHook을 만들어야 할 수도

from mmcv.runner.hooks.logger.wandb import WandbLoggerHook

@HOOKS.register_module()
class MMDetWandbHook(WandbLoggerHook):

[Sungyu-Park]

MMEngine의 vis_backend.py를 확인해보니 init_kwargs를 그대로 적용할 수가 있습니다.

vis_backends = [
    dict(type="LocalVisBackend"),
    dict(
        type="WandbVisBackend",
        init_kwargs={
            "entity": "boostcamp-ai-tech-4-cv-17",
            "project": "Final Project",
            "name": "test",
        },
    ),
]

위와 같이 사용하면 되겠습니다.

[nanpuhaha]

mmengine WandbVisBackend

mmcv WandbLoggerHook

mmdetection MMDetWandbHook