Last active
December 8, 2022 21:36
-
-
Save baughmann/dd69bffa809b6efbe3f5ccaba7283fee to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# parameters | |
nc: 2 # number of classes | |
depth_multiple: 0.33 # model depth multiple | |
width_multiple: 0.50 # layer channel multiple | |
# anchors | |
anchors: | |
- [10,13, 16,30, 33,23] # P3/8 | |
- [30,61, 62,45, 59,119] # P4/16 | |
- [116,90, 156,198, 373,326] # P5/32 | |
# YOLOv5 backbone | |
backbone: | |
# [from, number, module, args] | |
[[-1, 1, Focus, [64, 3]], # 0-P1/2 | |
[-1, 1, Conv, [128, 3, 2]], # 1-P2/4 | |
[-1, 3, BottleneckCSP, [128]], | |
[-1, 1, Conv, [256, 3, 2]], # 3-P3/8 | |
[-1, 9, BottleneckCSP, [256]], | |
[-1, 1, Conv, [512, 3, 2]], # 5-P4/16 | |
[-1, 9, BottleneckCSP, [512]], | |
[-1, 1, Conv, [1024, 3, 2]], # 7-P5/32 | |
[-1, 1, SPP, [1024, [5, 9, 13]]], | |
[-1, 3, BottleneckCSP, [1024, False]], # 9 | |
] | |
# YOLOv5 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, BottleneckCSP, [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, BottleneckCSP, [256, False]], # 17 (P3/8-small) | |
[-1, 1, Conv, [256, 3, 2]], | |
[[-1, 14], 1, Concat, [1]], # cat head P4 | |
[-1, 3, BottleneckCSP, [512, False]], # 20 (P4/16-medium) | |
[-1, 1, Conv, [512, 3, 2]], | |
[[-1, 10], 1, Concat, [1]], # cat head P5 | |
[-1, 3, BottleneckCSP, [1024, False]], # 23 (P5/32-large) | |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5) | |
] |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment