Log In Sign Up
Models:
Robert Orr
Robert-Orr/
Human-Fall-Detection
0
Downloads: 1
[6c6c05]: / cfg / training / yolov7-tiny.yaml

Download this file

113 lines (94 with data), 4.6 kB 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
# parameters

nc: 80  # number of classes

depth_multiple: 1.0  # model depth multiple

width_multiple: 1.0  # layer channel multiple



# anchors

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



# yolov7-tiny backbone

backbone:

  # [from, number, module, args] c2, k=1, s=1, p=None, g=1, act=True

  [[-1, 1, Conv, [32, 3, 2, None, 1, nn.LeakyReLU(0.1)]],  # 0-P1/2  

  

   [-1, 1, Conv, [64, 3, 2, None, 1, nn.LeakyReLU(0.1)]],  # 1-P2/4    

   

   [-1, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 7

   

   [-1, 1, MP, []],  # 8-P3/8

   [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 14

   

   [-1, 1, MP, []],  # 15-P4/16

   [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 21

   

   [-1, 1, MP, []],  # 22-P5/32

   [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [256, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [256, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [512, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 28

  ]



# yolov7-tiny head

head:

  [[-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, SP, [5]],

   [-2, 1, SP, [9]],

   [-3, 1, SP, [13]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -7], 1, Concat, [1]],

   [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 37

  

   [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

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

   [21, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # route backbone P4

   [[-1, -2], 1, Concat, [1]],

   

   [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 47

  

   [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

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

   [14, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]], # route backbone P3

   [[-1, -2], 1, Concat, [1]],

   

   [-1, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [32, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [32, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 57

   

   [-1, 1, Conv, [128, 3, 2, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, 47], 1, Concat, [1]],

   

   [-1, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [64, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [64, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 65

   

   [-1, 1, Conv, [256, 3, 2, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, 37], 1, Concat, [1]],

   

   [-1, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-2, 1, Conv, [128, 1, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [-1, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [[-1, -2, -3, -4], 1, Concat, [1]],

   [-1, 1, Conv, [256, 1, 1, None, 1, nn.LeakyReLU(0.1)]],  # 73

      

   [57, 1, Conv, [128, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [65, 1, Conv, [256, 3, 1, None, 1, nn.LeakyReLU(0.1)]],

   [73, 1, Conv, [512, 3, 1, None, 1, nn.LeakyReLU(0.1)]],



   [[74,75,76], 1, IDetect, [nc, anchors]],   # Detect(P3, P4, P5)

  ]