Skip to main content
. 2023 Mar 6;9:e1167. doi: 10.7717/peerj-cs.1167

Table 6. Summary of YO-CNN.

S.no From n Parameters Module Arguments
  0    −1  1    3,520  models.common.Conv      [3, 32, 6, 2, 2]  
  1    −1  1   18,560  models.common.Conv      [32, 64, 3, 2]    
  2    −1  1   18,816  models.common.C3        [64, 64, 1]      
  3    −1  1   73,984  models.common.Conv      [64, 128, 3, 2]  
  4    −1  2  115,712  models.common.C3        [128, 128, 2]    
  5    −1  1  295,424  models.common.Conv      [128, 256, 3, 2]  
  6    −1  3  625,152  models.common.C3        [256, 256, 3]    
  7    −1  1 1,180,672  models.common.Conv      [256, 512, 3, 2]  
  8    −1  1 1,182,720  models.common.C3        [512, 512, 1]    
  9    −1  1  656,896  models.common.SPPF      [512, 512, 5]    
 10    −1  1  131,584  models.common.Conv      [512, 256, 1, 1]  
 11    −1  1       0  torch.nn.modules.upsampling.Upsample  [None, 2, ’nearest’] 
 12   [−1, 6]  1       0  models.common.Concat    [1]  
 13    −1  1  361,984  models.common.C3        [512, 256, 1, False]  
 14    −1  1   33,024  models.common.Conv      [256, 128, 1, 1]  
 15    −1  1       0  torch.nn.modules.upsampling.Upsample  [None, 2, ’nearest’]
 16   [−1, 4]  1       0  models.common.Concat    [1]  
 17    −1  1   90,880  models.common.C3        [256, 128, 1, False]
 18    -1  1  147,712  models.common.Conv      [128, 128, 3, 2]  
 19   [−1, 14]  1       0  models.common.Concat    [1]  
 20    −1  1  296,448  models.common.C3        [256, 256, 1, False] 
 21    −1  1  590,336  models.common.Conv      [256, 256, 3, 2]  
 22   [−1, 10]  1       0  models.common.Concat    [1]  
 23    −1  1 1,182,720  models.common.C3        [512, 512, 1, False]  
 24   [17, 20, 23]  1   59,334  models.yolo.Detect      [17, [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], [128, 256, 512]]

Notes.

Model Summary: 270 layers, 7,065,478 parameters, 7,065,478 gradients, 16.0 GFLOPs.