I am trying to learn pytorch and can’t understand how the yaml file translates to model description. I am running yolov5 with yolov5n.yaml. here are the first two layers of the model description:
from n params module arguments
0 -1 1 1760 models.common.Conv [3, 16, 6, 2, 2]
1 -1 1 4672 models.common.Conv [16, 32, 3, 2]
and here are the first two layers in yolov5n.yaml:
# YOLOv5 v6.0 backbone
backbone:
# [from, number, module, args]
[[-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
[-1, 1, Conv, [128, 3, 2]], # 1-P2/4
How [64, 6, 2, 2] translates to [64, 6, 2, 2]?
How [16, 32, 3, 2] translates to [128, 3, 2]?