Yolov5 BackBone about yolov5

[glenn-jocher]

@moahaimen 👋 Hello! Thanks for asking about YOLOv5 🚀 architecture visualization. We've made visualizing YOLOv5 🚀 architectures super easy. There are 3 main ways:

model.yaml

Each model has a corresponding yaml file that displays the model architecture. Here is YOLOv5s, defined by yolov5s.yaml:

yolov5/models/yolov5s.yaml

Lines 12 to 48 in 1a3ecb8

	# YOLOv5 v6.0 backbone
	backbone:
	# [from, number, module, args]
	[[-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
	[-1, 1, Conv, [128, 3, 2]], # 1-P2/4
	[-1, 3, C3, [128]],
	[-1, 1, Conv, [256, 3, 2]], # 3-P3/8
	[-1, 6, C3, [256]],
	[-1, 1, Conv, [512, 3, 2]], # 5-P4/16
	[-1, 9, C3, [512]],
	[-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
	[-1, 3, C3, [1024]],
	[-1, 1, SPPF, [1024, 5]], # 9
	]
	# YOLOv5 v6.0 head
	head:
	[[-1, 1, Conv, [512, 1, 1]],
	[-1, 1, nn.Upsample, [None, 2, 'nearest']],
	[[-1, 6], 1, Concat, [1]], # cat backbone P4
	[-1, 3, C3, [512, False]], # 13
	[-1, 1, Conv, [256, 1, 1]],
	[-1, 1, nn.Upsample, [None, 2, 'nearest']],
	[[-1, 4], 1, Concat, [1]], # cat backbone P3
	[-1, 3, C3, [256, False]], # 17 (P3/8-small)
	[-1, 1, Conv, [256, 3, 2]],
	[[-1, 14], 1, Concat, [1]], # cat head P4
	[-1, 3, C3, [512, False]], # 20 (P4/16-medium)
	[-1, 1, Conv, [512, 3, 2]],
	[[-1, 10], 1, Concat, [1]], # cat head P5
	[-1, 3, C3, [1024, False]], # 23 (P5/32-large)
	[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
	]

TensorBoard Graph

Simply start training a model, and then view the TensorBoard Graph for an interactive view of the model architecture. This example shows YOLOv5s viewed in our Notebook –

# Tensorboard
%load_ext tensorboard
%tensorboard --logdir runs/train

# Train YOLOv5s on COCO128 for 3 epochs
python train.py --weights yolov5s.pt --epochs 3

Netron viewer

Use https://netron.app to view exported ONNX models:

python export.py --weights yolov5s.pt --include onnx --simplify

Good luck 🍀 and let us know if you have any other questions!

[github-actions]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://docs.ultralytics.com/yolov5
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!