YOLOv5

Ultralytics’ YOLOv5 (“You Only Look Once”) 모델 제품군은 고통스러운 과정 없이 컨볼루션 신경망을 통해 실시간 오브젝트 검출을 가능하게 합니다.

Weights & Biases는 YOLOv5에 직접 통합되어 실험 메트릭 추적, 모델 및 데이터셋 버전 관리, 풍부한 모델 예측 시각화 등을 제공합니다. YOLO 실험을 실행하기 전에 간단히 pip install 한 번만 실행하면 됩니다.

핵심 Experiments 추적

wandb를 설치하는 것만으로도 W&B 기본 제공 로깅 기능을 활성화할 수 있습니다. 시스템 메트릭, 모델 메트릭 및 대화형 대시보드에 기록된 미디어를 확인할 수 있습니다.

pip install wandb
git clone https://github.com/ultralytics/yolov5.git
python yolov5/train.py  # 작은 데이터셋에서 작은 네트워크를 트레이닝합니다.

wandb에서 표준 출력으로 출력된 링크를 따라가기만 하면 됩니다.

통합 사용자 정의

YOLO에 몇 가지 간단한 커맨드라인 인수를 전달하여 더 많은 W&B 기능을 활용할 수 있습니다.

• --save_period에 숫자를 전달하면 W&B는 모든 save_period 에포크가 끝날 때마다 모델 버전을 저장합니다. 모델 버전에는 모델 가중치가 포함되어 있으며 검증 세트에서 가장 성능이 좋은 모델에 태그를 지정합니다.
• --upload_dataset 플래그를 켜면 데이터 버전 관리를 위해 데이터셋도 업로드됩니다.
• --bbox_interval에 숫자를 전달하면 데이터 시각화이 켜집니다. 모든 bbox_interval 에포크가 끝나면 검증 세트에서 모델의 출력이 W&B에 업로드됩니다.

• 모델 버전 관리만 해당
• 모델 버전 관리 및 Data Visualization

python yolov5/train.py --epochs 20 --save_period 1

python yolov5/train.py --epochs 20 --save_period 1 \
  --upload_dataset --bbox_interval 1

다음은 그 모습입니다.