SRD_dseg.yaml

# This is a custom yaml file for the yolov9 architecture with segmentation.

#------------------------------ Params ------------------------#
nc: 80
depth_multiple: 1.0
width_multiple: 1.0
anchors: 3

#---------------------------Blocks & Params-----------------------------#

# Silence()
# Conv(ch_out,k,s)
# ADown(ch_out)
# RepNCSPELAN4(ch_out,ch_in,partition,n)
# SPPELAN (ch_out,ch_in)

#---------------------------------Backbone------------------------------#
backbone:
  [ 
    # Silence Block                       
    [-1, 1, Silence, []],                      # L0

    # Conv Block            
    [-1, 1, Conv, [64,3,2]],                   # L1  P1/2

    # Conv Block  
    [-1, 1, Conv, [128,3,2]],                  # L2  P2/4

    # CSPELAN Block 1
    [-1, 1, RepNCSPELAN4, [256,128,64,1]],     # L3 

    # ADown 1
    [-1, 1, ADown, [256]],                     # L4  P3/8

    #CSPELAN Block 2
    [-1, 1, RepNCSPELAN4, [512,256,64,1]],     # L5

    # ADown 2
    [-1, 1, ADown, [512]],                     # L6  P4/16

    #CSPELAN Block 3
    [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]], # L7

    # ADown 3
    [-1, 1, ADown, [512]],                     # L8  P5/32

    # CSPELAN Block 4
    [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]]  # L9  
  ]
 
#---------------------------------Heads------------------------------#
head:
  [
    #------------------ Upsample Section ------------------#
    # SPPELAN Block 1
    [-1, 1, SPPELAN, [512,256]],                # L10

    # UP & CONCAT 1
    [-1, 1, nn.Upsample, [None, 2, 'nearest']], # L11 
    [[-1, 7], 1, Concat, [1]],                  # L12  //Concat Backbone L7

    # CSPELAN Block 1
    [-1, 1, RepNCSPELAN4, [512, 512, 256, 1]],  # L13

    # UP & CONCAT 2
    [-1, 1, nn.Upsample, [None, 2, 'nearest']], # L14
    [[-1, 5], 1, Concat, [1]],                  # L15  //Concat backbone L5

    # CSPELAN Block 2 
    [-1, 1, RepNCSPELAN4, [256, 256, 128, 1]],  # L16  //H-small

    #------------------ Downsample Section ------------------#
    # ADown & CONCAT 1
    [-1, 1, ADown, [256]],                      # L17
    [[-1, 13], 1, Concat, [1]],                  # L18  //Concat Upsample L13

    # CSPELAN Block 1 
    [-1, 1, RepNCSPELAN4, [512,512,256,1]],      # L19  //H-medium

    # ADown & CONCAT 2
    [-1, 1, ADown, [512]],                      # L20
    [[-1, 16], 1, Concat, [1]],                  # L21 //Concat Upsample L16

    # CSPELAN Block 2
    [-1, 1, RepNCSPELAN4, [512,512,256,1]],      # L22  //H-large

    #------------------ Segmentation Section ------------------#
    # CSPELAN Block 1
    [17, 1, RepNCSPELAN4, [512,512,256,1]],      # L23

    # UP
    [-1, 1, nn.Upsample, [None, 2, 'nearest']], # L24

    # Conv Block
    [-1, 1, Conv, [256, 3, 1]],                  # L25

    # Segmentation Block
    [[16, 19, 22, 25], 1, DSegment, [nc, 32, 256]]  # Segment(P3, P4, P5)
    
  ]