YOLOX

W&B를 YOLOX와 연동하는 방법

YOLOX는 객체 탐지에서 뛰어난 성능을 보이는 앵커-프리(anchor-free) 버전의 YOLO입니다. YOLOX W&B 인테그레이션을 사용하면 트레이닝, 검증, 시스템 관련 메트릭 로깅을 활성화할 수 있고, 커맨드라인 인자 하나만으로 예측 결과를 대화형으로 검증할 수 있습니다.

​

가입 및 API 키 생성하기

API 키는 사용자의 머신을 W&B에 인증하는 데 사용됩니다. 사용자 프로필에서 API 키를 생성할 수 있습니다.

더 간편한 방법을 원한다면 User Settings에서 직접 API 키를 생성하세요. 새로 생성된 API 키를 즉시 복사하여 비밀번호 관리자와 같은 안전한 위치에 저장하세요.

1. 오른쪽 상단의 사용자 프로필 아이콘을 클릭합니다.
2. User Settings를 선택한 다음, 아래로 스크롤하여 API Keys 섹션을 찾습니다.

​

wandb 라이브러리를 설치하고 로그인하기

로컬 환경에서 wandb 라이브러리를 설치하고 로그인하려면:

명령줄

1. WANDB_API_KEY 환경 변수를 자신의 API 키로 설정합니다.

   export WANDB_API_KEY=<your_api_key>
   

2. wandb 라이브러리를 설치하고 로그인합니다.

   pip install wandb
   
   wandb login
   

Python

pip install wandb

import wandb
wandb.login()

Python 노트북

!pip install wandb

import wandb
wandb.login()

​

메트릭 기록

--logger wandb 명령줄 인자를 사용해 wandb 로깅을 활성화하세요. 선택적으로 wandb.init()이 기대하는 모든 인자를 동일하게 전달할 수 있으며, 각 인자 이름 앞에 wandb-를 붙이면 됩니다.

num_eval_imges는 모델 평가를 위해 W&B Tables에 로깅되는 검증 세트 이미지와 예측의 개수를 제어합니다.

# wandb에 로그인
wandb login

# `wandb` logger 인수를 사용하여 yolox 트레이닝 스크립트 호출
python tools/train.py .... --logger wandb \
                wandb-project <project-name> \
                wandb-entity <entity>
                wandb-name <run-name> \
                wandb-id <run-id> \
                wandb-save_dir <save-dir> \
                wandb-num_eval_imges <num-images> \
                wandb-log_checkpoints <bool>

​

예시

YOLOX 트레이닝 및 검증 메트릭이 포함된 예시 대시보드 ->

이 W&B 인테그레이션에 대해 질문이나 문제가 있으신가요? YOLOX 저장소에 이슈를 생성하세요.