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 ·
572 阅读 ·
0 评论