YOLOX

YOLOX 는 오브젝트 검출을 위한 강력한 성능을 가진 앵커 프리 버전의 YOLO입니다. YOLOX W&B 인테그레이션을 사용하면 트레이닝, 검증, 시스템 관련 메트릭의 로깅을 켤 수 있으며, 단일 커맨드라인 인수로 예측값을 대화형으로 검증할 수 있습니다.

가입하고 API 키 만들기

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

1. 오른쪽 상단 모서리에 있는 사용자 프로필 아이콘을 클릭합니다.
2. User Settings 를 선택한 다음 API Keys 섹션으로 스크롤합니다.
3. Reveal 을 클릭합니다. 표시된 API 키를 복사합니다. API 키를 숨기려면 페이지를 새로 고침하십시오.

wandb 라이브러리 설치 및 로그인

wandb 라이브러리를 로컬에 설치하고 로그인하려면 다음을 수행합니다.

• Command Line
• Python
• Python notebook

pip install wandb

import wandb
wandb.login()

!pip install wandb

import wandb
wandb.login()

메트릭 로그

--logger wandb 커맨드 라인 인수를 사용하여 wandb로 로깅을 켭니다. 선택적으로 wandb.init 가 예상하는 모든 인수를 전달할 수도 있습니다. 각 인수 앞에 wandb- 를 붙입니다.

num_eval_imges 는 모델 평가를 위해 W&B 테이블에 기록되는 검증 세트 이미지 및 예측값의 수를 제어합니다.

# wandb에 로그인
wandb login

# `wandb` 로거 인수로 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 repository 에 이슈를 여세요.