config.json

{
  "architectures": [
    "YolosForObjectDetection"
  ],
  "model_type": "yolos",
  "input_size": [640, 640],
  "nc": 2,
  "depth_multiple": 1.0,
  "width_multiple": 1.0,
  "anchors": [
    [12, 16, 19, 36, 40, 28],
    [36, 75, 76, 55, 72, 146],
    [142, 110, 192, 243, 459, 401]
  ],
  "backbone":[
    [-1, 1, "Conv", [32, 3, 1]],  
  
   [-1, 1, "Conv", [64, 3, 2]],      
   [-1, 1, "Conv", [64, 3, 1]],
   
   [-1, 1, "Conv", [128, 3, 2]],   
   [-1, 1, "Conv", [64, 1, 1]],
   [-2, 1, "Conv", [64, 1, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [[-1, -3, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [256, 1, 1]],  
         
   [-1, 1, "MP", []],
   [-1, 1, "Conv", [128, 1, 1]],
   [-3, 1, "Conv", [128, 1, 1]],
   [-1, 1, "Conv", [128, 3, 2]],
   [[-1, -3], 1, "Concat", [1]],   
   [-1, 1, "Conv", [128, 1, 1]],
   [-2, 1, "Conv", [128, 1, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [[-1, -3, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [512, 1, 1]],  
         
   [-1, 1, "MP", []],
   [-1, 1, "Conv", [256, 1, 1]],
   [-3, 1, "Conv", [256, 1, 1]],
   [-1, 1, "Conv", [256, 3, 2]],
   [[-1, -3], 1, "Concat", [1]],    
   [-1, 1, "Conv", [256, 1, 1]],
   [-2, 1, "Conv", [256, 1, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [[-1, -3, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [1024, 1, 1]], 
         
   [-1, 1, "MP", []],
   [-1, 1, "Conv", [512, 1, 1]],
   [-3, 1, "Conv", [512, 1, 1]],
   [-1, 1, "Conv", [512, 3, 2]],
   [[-1, -3], 1, "Concat", [1]],    
   [-1, 1, "Conv", [256, 1, 1]],
   [-2, 1, "Conv", [256, 1, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [[-1, -3, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [1024, 1, 1]]  
  ],
  "head":[
    [-1, 1, "SPPCSPC", [512]], 
  
   [-1, 1, "Conv", [256, 1, 1]],
   [-1, 1, "nn.Upsample", ["None", 2, "nearest"]],
   [37, 1, "Conv", [256, 1, 1]], 
   [[-1, -2], 1, "Concat", [1]],
   
   [-1, 1, "Conv", [256, 1, 1]],
   [-2, 1, "Conv", [256, 1, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [[-1, -2, -3, -4, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [256, 1, 1]],
   
   [-1, 1, "Conv", [128, 1, 1]],
   [-1, 1, "nn.Upsample", ["None", 2, "nearest"]],
   [24, 1, "Conv", [128, 1, 1]], 
   [[-1, -2], 1, "Concat", [1]],
   
   [-1, 1, "Conv", [128, 1, 1]],
   [-2, 1, "Conv", [128, 1, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [-1, 1, "Conv", [64, 3, 1]],
   [[-1, -2, -3, -4, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [128, 1, 1]], 
      
   [-1, 1, "MP", []],
   [-1, 1, "Conv", [128, 1, 1]],
   [-3, 1, "Conv", [128, 1, 1]],
   [-1, 1, "Conv", [128, 3, 2]],
   [[-1, -3, 63], 1, "Concat", [1]],
   
   [-1, 1, "Conv", [256, 1, 1]],
   [-2, 1, "Conv", [256, 1, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [-1, 1, "Conv", [128, 3, 1]],
   [[-1, -2, -3, -4, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [256, 1, 1]], 
      
   [-1, 1, "MP", []],
   [-1, 1, "Conv", [256, 1, 1]],
   [-3, 1, "Conv", [256, 1, 1]],
   [-1, 1, "Conv", [256, 3, 2]],
   [[-1, -3, 51], 1, "Concat", [1]],
   
   [-1, 1, "Conv", [512, 1, 1]],
   [-2, 1, "Conv", [512, 1, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [-1, 1, "Conv", [256, 3, 1]],
   [[-1, -2, -3, -4, -5, -6], 1, "Concat", [1]],
   [-1, 1, "Conv", [512, 1, 1]], 
   
   [75, 1, "RepConv", [256, 3, 1]],
   [88, 1, "RepConv", [512, 3, 1]],
   [101, 1, "RepConv", [1024, 3, 1]],

   [[102,103,104], 1, "IDetect", ["nc", "anchors"]]
  ],
  "names": ["0", "1"],
  "output_layers": [
    {"layer_size": [52, 52], "anchors": 3},
    {"layer_size": [26, 26], "anchors": 3},
    {"layer_size": [13, 13], "anchors": 3}
  ],
  "nms_threshold": 0.1
}