YOLOv5 version 6.1 architecture

[fpearmaine3]

Hello all.
Attached is my interpretation of the architecture that has changed since the first release of YOLOv5, I think it's quite nice and accurate. Please let me know if it's incorrect in any way.
Thank you all

.

[nrupatunga]

Thank you for nice visualisation of the architecture and if I'm not wrong, the following blocks in the head section need to be swapped, and the sequence of upsampling by a factor of 2 indicates the same

[liamh1999]

I think there is a conv layer missing in the neck between the c3 and upsample ?