- 博客(2)
- 收藏
- 关注
原创 YOLO V5颈部网络改进添加BiFPN
[-1, 13, 6], 1, BiFPN_Add3, [256, 256]], # cat P4 <--- BiFPN change 注意v5s通道数是默认参数的一半。# 设置可学习参数 nn.Parameter的作用是:将一个不可训练的类型Tensor转换成可以训练的类型parameter。[-1, 1, Conv, [512, 3, 2]], # 为了BiFPN正确add,调整channel数。四,修改train.py文件的参数:"--cfg"对应的路径改为自己创建的.yaml。
2023-09-07 15:48:36
295
原创 YOLO V5更换主干网络ShuffleNet v2
self.depthwise_conv(branch_features, branch_features, kernel_size=3, stride=self.stride, padding=1),##加的。# self.depthwise_conv(inp, inp, kernel_size=3, stride=1, padding=1),###加的。###############----- ShuffleNetV2主干网络-----#######最后运行train.py 就可以啦!
2023-08-03 22:13:49
550
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人