Integration of v8 segmentation (#107)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Laughing 2022-12-28 23:01:38 +08:00 committed by GitHub
parent 384f0ef1c6
commit 8406b49b49
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
16 changed files with 422 additions and 224 deletions

View file

@ -4,9 +4,8 @@
nc: 80 # number of classes
depth_multiple: 0.33 # model depth multiple
width_multiple: 0.25 # layer channel multiple
anchors: [[16,19], [55,65], [178,192]]
# YOLOv8n v0.0 backbone
# YOLOv8.0n backbone
backbone:
# [from, number, module, args]
[[-1, 1, Conv, [64, 3, 2]], # 0-P1/2
@ -21,7 +20,7 @@ backbone:
[-1, 1, SPPF, [1024, 5]], # 9
]
# YOLOv8n v0.0 head
# YOLOv8.0n head
head:
[[-1, 1, nn.Upsample, [None, 2, 'nearest']],
[[-1, 6], 1, Concat, [1]], # cat backbone P4