解析Yolox的 Conv = DWConv if depthwise else BaseConv

在YoloX源码中有这样一句:

        Conv = DWConv if depthwise else BaseConv

其含义如下:

YoloX中存在两种卷积操作 BaseConv 和 DWConv

1、BaseConv 基础卷积操作:

        卷积(nn.BatchNorm2d) -> batch归一化(nn.BatchNorm2d) -> 激活函数

2、DWConv 深度卷积:

        基础卷积(BaseConv) ->  卷积核大小为1,步长为1 的卷积

所以在YoloX中,如果吧标志为 depthwise 设置为 True,则执行深度卷积,否则执行普通卷积。

    

你可能感兴趣的:(深度学习,计算机视觉,人工智能)