diff --git a/models/v5Lite-n.yaml b/models/v5Lite-n.yaml deleted file mode 100644 index 6886b7e2..00000000 --- a/models/v5Lite-n.yaml +++ /dev/null @@ -1,48 +0,0 @@ -# YOLOv5 🚀 by Ultralytics, GPL-3.0 license - -# Parameters -nc: 80 # number of classes -depth_multiple: 1.0 # model depth multiple -width_multiple: 1.0 # layer channel multiple -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 v6.0 backbone -backbone: - # [from, number, module, args] - [[-1, 1, Conv, [24, 6, 2, 2]], # 0-P1/2 - [-1, 1, RepVGGBlock, [32, 3, 2]], # 1-P2/4 - [-1, 1, C3, [32]], - [-1, 1, RepVGGBlock, [64, 3, 2]], # 3-P3/8 - [-1, 2, C3, [64]], - [-1, 1, RepVGGBlock, [128, 3, 2]], # 5-P4/16 - [-1, 3, C3, [128]], - [-1, 1, RepVGGBlock, [256, 3, 2]], # 7-P5/32 - [-1, 1, C3, [256]], - [-1, 1, SPPF, [256, 5]], # 9 - ] - -# YOLOv5 v6.0 head -head: - [[-1, 1, Conv, [128, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [[-1, 6], 1, Concat, [1]], # cat backbone P4 - [-1, 1, C3, [128, False]], # 13 - - [-1, 1, Conv, [64, 1, 1]], - [-1, 1, nn.Upsample, [None, 2, 'nearest']], - [[-1, 4], 1, Concat, [1]], # cat backbone P3 - [-1, 1, C3, [64, False]], # 17 (P3/8-small) - - [-1, 1, Conv, [64, 3, 2]], - [[-1, 14], 1, Concat, [1]], # cat head P4 - [-1, 1, C3, [128, False]], # 20 (P4/16-medium) - - [-1, 1, Conv, [128, 3, 2]], - [[-1, 10], 1, Concat, [1]], # cat head P5 - [-1, 1, C3, [256, False]], # 23 (P5/32-large) - - [[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5) - ] \ No newline at end of file