YOLOv5网络修改

[sunshine411]

你好, 我最近两天也在使用tensorflow重新实现网络, 现在还没有完成, 刚看到你的代码, 有两点疑问:

1. 你代码中是否缺少focus层;
2. 根据pytorch源码, 我记得有些层既没有bn, 也没有bias.

`def focus(input_data, trainable, init_width_size):
input_data = tf.concat([input_data[:, ::2, ::2, :], input_data[:, 1::2, ::2, :],
input_data[:, ::2, 1::2, :], input_data[:, 1::2, 1::2, :]], 3)
input_data = conv(input_data, (3, 3, 12, init_width_size), trainable=trainable, name='conv0', act_fun='mish')
return input_data

`

感谢您的分享.

[ycdhqzhiai]

1.你好，readme里面提到过，我去除了focus slice，直接用strides=2取代了
2.完全按照原版里面，没BN的话，训练在第二个epoch loss就为na了，所以我BN统一设置为True