Updated model yamls

scaleoutsystems · Oct 30, 2024 · ec08a9f · ec08a9f
1 parent 0b96ee5
commit ec08a9f
Show file tree

Hide file tree

Showing 5 changed files with 84 additions and 84 deletions.
diff --git a/client/yolov8models/large.yaml b/client/yolov8models/large.yaml
@@ -4,33 +4,33 @@
 
 # YOLOv8l backbone
 backbone:
-  - [-1, 1, Conv, [64, 3, 2]]          # 0-P1/2
-  - [-1, 1, Conv, [128, 3, 2]]         # 1-P2/4
-  - [-1, 6, C2f, [128, True]]
-  - [-1, 1, Conv, [256, 3, 2]]         # 3-P3/8
-  - [-1, 12, C2f, [256, True]]
-  - [-1, 1, Conv, [512, 3, 2]]         # 5-P4/16
-  - [-1, 12, C2f, [512, True]]
-  - [-1, 1, Conv, [512, 3, 2]]         # 7-P5/32
-  - [-1, 6, C2f, [512, True]]
+  - [-1, 1, Conv, [64, 3, 2]]  
+  - [-1, 1, Conv, [128, 3, 2]] 
+  - [-1, 3, C2f, [128, True]]  
+  - [-1, 1, Conv, [256, 3, 2]] 
+  - [-1, 6, C2f, [256, True]]  
+  - [-1, 1, Conv, [384, 3, 2]]  
+  - [-1, 6, C2f, [384, True]]  
+  - [-1, 1, Conv, [512, 3, 2]]  
+  - [-1, 3, C2f, [512, True]]
   - [-1, 1, SPPF, [512, 5]]
 
 # YOLOv8l head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]]
-  - [-1, 6, C2f, [256]]
+  - [-1, 3, C2f, [384]] 
 
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 4], 1, Concat, [1]]
-  - [-1, 6, C2f, [128]]                # P3/8-small
-
-  - [-1, 1, Conv, [128, 3, 2]]
-  - [[-1, 11], 1, Concat, [1]]
-  - [-1, 6, C2f, [256]]                # P4/16-medium
+  - [-1, 3, C2f, [256]]  
 
   - [-1, 1, Conv, [256, 3, 2]]
+  - [[-1, 12], 1, Concat, [1]]
+  - [-1, 3, C2f, [384]] 
+
+  - [-1, 1, Conv, [384, 3, 2]]
   - [[-1, 9], 1, Concat, [1]]
-  - [-1, 6, C2f, [512]]                # P5/32-large
+  - [-1, 3, C2f, [512]]  
 
-  - [[14, 17, 20], 1, Detect, [nc]]
+  - [[15, 18, 21], 1, Detect, [nc]] 
diff --git a/client/yolov8models/medium.yaml b/client/yolov8models/medium.yaml
@@ -4,33 +4,33 @@
 
 # YOLOv8m backbone
 backbone:
-  - [-1, 1, Conv, [48, 3, 2]]          # 0-P1/2
-  - [-1, 1, Conv, [96, 3, 2]]          # 1-P2/4 
-  - [-1, 4, C2f, [96, True]]
-  - [-1, 1, Conv, [192, 3, 2]]         # 3-P3/8
-  - [-1, 6, C2f, [192, True]]
-  - [-1, 1, Conv, [384, 3, 2]]         # 5-P4/16 
-  - [-1, 6, C2f, [384, True]]
-  - [-1, 1, Conv, [768, 3, 2]]         # 7-P5/32 
-  - [-1, 4, C2f, [768, True]]
-  - [-1, 1, SPPF, [768, 5]]
+  - [-1, 1, Conv, [48, 3, 2]]  
+  - [-1, 1, Conv, [96, 3, 2]]  
+  - [-1, 2, C2f, [96, True]]  
+  - [-1, 1, Conv, [192, 3, 2]]  
+  - [-1, 4, C2f, [192, True]]  
+  - [-1, 1, Conv, [384, 3, 2]] 
+  - [-1, 4, C2f, [384, True]]  
+  - [-1, 1, Conv, [512, 3, 2]] 
+  - [-1, 2, C2f, [512, True]]
+  - [-1, 1, SPPF, [512, 5]]
 
 # YOLOv8m head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]]
-  - [-1, 4, C2f, [384]]
+  - [-1, 3, C2f, [384]] 
 
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 4], 1, Concat, [1]]
-  - [-1, 4, C2f, [192]]                # P3/8-small
+  - [-1, 3, C2f, [192]] 
 
   - [-1, 1, Conv, [192, 3, 2]]
-  - [[-1, 11], 1, Concat, [1]]
-  - [-1, 4, C2f, [384]]                # P4/16-medium
+  - [[-1, 12], 1, Concat, [1]]
+  - [-1, 3, C2f, [384]]
 
   - [-1, 1, Conv, [384, 3, 2]]
   - [[-1, 9], 1, Concat, [1]]
-  - [-1, 4, C2f, [768]]                # P5/32-large
+  - [-1, 3, C2f, [512]]  
 
-  - [[14, 17, 20], 1, Detect, [nc]]
+  - [[15, 18, 21], 1, Detect, [nc]]  
diff --git a/client/yolov8models/nano.yaml b/client/yolov8models/nano.yaml
@@ -4,33 +4,33 @@
 
 # YOLOv8n backbone
 backbone:
-  - [-1, 1, Conv, [16, 3, 2]]          # 0-P1/2
-  - [-1, 1, Conv, [32, 3, 2]]          # 1-P2/4 
-  - [-1, 2, C2f, [32, True]]           
-  - [-1, 1, Conv, [64, 3, 2]]          # 3-P3/8 
-  - [-1, 3, C2f, [64, True]]           
-  - [-1, 1, Conv, [128, 3, 2]]         # 5-P4/16 
-  - [-1, 3, C2f, [128, True]]          
-  - [-1, 1, Conv, [256, 3, 2]]         # 7-P5/32
-  - [-1, 2, C2f, [256, True]]          
-  - [-1, 1, SPPF, [256, 5]]            
+  - [-1, 1, Conv, [16, 3, 2]]  # 64 * 0.25 = 16
+  - [-1, 1, Conv, [32, 3, 2]]  # 128 * 0.25 = 32
+  - [-1, 1, C2f, [32, True]]  # min(128 * 0.25, 1024) = 32
+  - [-1, 1, Conv, [64, 3, 2]]  # 256 * 0.25 = 64
+  - [-1, 2, C2f, [64, True]]  # int(6 * 0.33) = 2; min(256 * 0.25, 1024) = 64
+  - [-1, 1, Conv, [128, 3, 2]]  # 512 * 0.25 = 128
+  - [-1, 2, C2f, [128, True]]  # min(512 * 0.25, 1024) = 128
+  - [-1, 1, Conv, [256, 3, 2]]  # min(1024 * 0.25, 1024) = 256
+  - [-1, 1, C2f, [256, True]]
+  - [-1, 1, SPPF, [256, 5]]
 
 # YOLOv8n head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]]          
-  - [-1, 2, C2f, [128]]                
+  - [-1, 1, C2f, [128]]                
 
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 4], 1, Concat, [1]]          
-  - [-1, 2, C2f, [64]]                 # P3/8-small
+  - [-1, 1, C2f, [64]]                 # P3/8-small
 
   - [-1, 1, Conv, [64, 3, 2]]
-  - [[-1, 11], 1, Concat, [1]]         
-  - [-1, 2, C2f, [128]]                # P4/16-medium
+  - [[-1, 10], 1, Concat, [1]]         
+  - [-1, 1, C2f, [128]]                # P4/16-medium
 
   - [-1, 1, Conv, [128, 3, 2]]
-  - [[-1, 9], 1, Concat, [1]]          
-  - [-1, 2, C2f, [256]]                # P5/32-large
+  - [[-1, 8], 1, Concat, [1]]          
+  - [-1, 1, C2f, [256]]                # P5/32-large
 
-  - [[14, 17, 20], 1, Detect, [nc]]    
+  - [[13, 16, 19], 1, Detect, [nc]]    
diff --git a/client/yolov8models/small.yaml b/client/yolov8models/small.yaml
@@ -4,33 +4,33 @@
 
 # YOLOv8s backbone
 backbone:
-  - [-1, 1, Conv, [32, 3, 2]]          # 0-P1/2 
-  - [-1, 1, Conv, [64, 3, 2]]          # 1-P2/4 
-  - [-1, 2, C2f, [64, True]]           
-  - [-1, 1, Conv, [128, 3, 2]]         # 3-P3/8
-  - [-1, 3, C2f, [128, True]]          
-  - [-1, 1, Conv, [256, 3, 2]]         # 5-P4/16
-  - [-1, 3, C2f, [256, True]]          
-  - [-1, 1, Conv, [512, 3, 2]]         # 7-P5/32
-  - [-1, 2, C2f, [512, True]]          
-  - [-1, 1, SPPF, [512, 5]]            
+  - [-1, 1, Conv, [32, 3, 2]]
+  - [-1, 1, Conv, [64, 3, 2]] 
+  - [-1, 1, C2f, [64, True]]  
+  - [-1, 1, Conv, [128, 3, 2]]
+  - [-1, 2, C2f, [128, True]] 
+  - [-1, 1, Conv, [256, 3, 2]]
+  - [-1, 2, C2f, [256, True]] 
+  - [-1, 1, Conv, [512, 3, 2]] 
+  - [-1, 1, C2f, [512, True]]
+  - [-1, 1, SPPF, [512, 5]]
 
 # YOLOv8s head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]]
-  - [-1, 2, C2f, [256]]
+  - [-1, 3, C2f, [256]]  
 
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 4], 1, Concat, [1]]
-  - [-1, 2, C2f, [128]]                # P3/8-small
+  - [-1, 3, C2f, [128]] 
 
   - [-1, 1, Conv, [128, 3, 2]]
-  - [[-1, 11], 1, Concat, [1]]
-  - [-1, 2, C2f, [256]]                # P4/16-medium
+  - [[-1, 12], 1, Concat, [1]]
+  - [-1, 3, C2f, [256]] 
 
   - [-1, 1, Conv, [256, 3, 2]]
   - [[-1, 9], 1, Concat, [1]]
-  - [-1, 2, C2f, [512]]                # P5/32-large
+  - [-1, 3, C2f, [512]] 
 
-  - [[14, 17, 20], 1, Detect, [nc]]
+  - [[15, 18, 21], 1, Detect, [nc]]
diff --git a/client/yolov8models/x-large.yaml b/client/yolov8models/x-large.yaml
@@ -4,33 +4,33 @@
 
 # YOLOv8x backbone
 backbone:
-  - [-1, 1, Conv, [80, 3, 2]]          # 0-P1/2
-  - [-1, 1, Conv, [160, 3, 2]]         # 1-P2/4
-  - [-1, 6, C2f, [160, True]]
-  - [-1, 1, Conv, [320, 3, 2]]         # 3-P3/8 
-  - [-1, 12, C2f, [320, True]]
-  - [-1, 1, Conv, [512, 3, 2]]         # 5-P4/16 
-  - [-1, 12, C2f, [512, True]]
-  - [-1, 1, Conv, [512, 3, 2]]         # 7-P5/32 
-  - [-1, 9, C2f, [512, True]]
-  - [-1, 1, SPPF, [512, 5]]
+  - [-1, 1, Conv, [80, 3, 2]] 
+  - [-1, 1, Conv, [160, 3, 2]]
+  - [-1, 3, C2f, [160, True]] 
+  - [-1, 1, Conv, [320, 3, 2]]
+  - [-1, 6, C2f, [320, True]] 
+  - [-1, 1, Conv, [512, 3, 2]]
+  - [-1, 6, C2f, [512, True]] 
+  - [-1, 1, Conv, [512, 3, 2]]
+  - [-1, 3, C2f, [512, True]] 
+  - [-1, 1, SPPF, [512, 5]]   
 
 # YOLOv8x head
 head:
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 6], 1, Concat, [1]]
-  - [-1, 9, C2f, [320]]
+  - [-1, 3, C2f, [384]] 
 
   - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
   - [[-1, 4], 1, Concat, [1]]
-  - [-1, 9, C2f, [160]]                # P3/8-small
-
-  - [-1, 1, Conv, [160, 3, 2]]
-  - [[-1, 11], 1, Concat, [1]]
-  - [-1, 9, C2f, [320]]                # P4/16-medium
+  - [-1, 3, C2f, [320]]  
 
   - [-1, 1, Conv, [320, 3, 2]]
+  - [[-1, 12], 1, Concat, [1]]
+  - [-1, 3, C2f, [384]] 
+
+  - [-1, 1, Conv, [384, 3, 2]]
   - [[-1, 9], 1, Concat, [1]]
-  - [-1, 9, C2f, [512]]                # P5/32-large
+  - [-1, 3, C2f, [512]] 
 
-  - [[14, 17, 20], 1, Detect, [nc]]
+  - [[15, 18, 21], 1, Detect, [nc]]