YOLO WanDB

https://wandb.ai/home

해당 사이트에 회원가입

로그인 후 Project new Create

Project name 입력 후 Create Project 버튼을 클릭하여 프로젝트 생성

pip를 이용해 종속성 라이브러리 설치

pip3 install wandb

API Key 복사

https://wandb.ai/quickstart/pytorch

train.py 이용해 학습 시작

python3 train.py --img 640 --batch 32 --epochs 20 --data main-directory경로/dataset/data.yaml --weights yolov7.pt --name yolov7test --project 생성한프로젝트명

train.py로 학습시작하면 wandb api key 입력하라고 뜰때 사전에 복사해둔 키 입력 후 엔터

프로젝트 생성, 선택할 수 있는데 2번을 눌러 프로젝트명 입력후 엔터

Leave a Comment