Skip to content

Instantly share code, notes, and snippets.

@bala-codes

bala-codes/Yolov5-model_train.py

Created August 12, 2020 16:55
Show Gist options
  • Save bala-codes/3862fcfed55d2c641b0967320797513a to your computer and use it in GitHub Desktop.
Save bala-codes/3862fcfed55d2c641b0967320797513a to your computer and use it in GitHub Desktop.
Download ZIP
BCC-model_build
Raw
Yolov5-model_train.py
# Start tensorboard (optional)
%load_ext tensorboard
%tensorboard --logdir runs/
!python yolov5/train.py --img 640 --batch 8 --epochs 100 \
--data bcc.yaml --cfg models/yolov5s.yaml --name BCCM
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment