Depth-wise1 [ CNN ] 4. Depth-wise convolution - PyTorch Code Depth-wise convoltuion 단일 채널에 대해서만 수행되는 필터를 이용하여 convolution 하는 것을 말한다. 채널 방향으로 합쳐지지 않고, spatial 방향으로만 convolution을 진행한다. 각 channel 별로 convolution 결과적으로 입력 채널 수 만큼 그룹을 나눈 Grouped Convolution 과 동일하다 장점 : 각 채널 고유의 feature를 뽑아낼 수 있음, parameter 수와 연산을 줄일 수 있음. - parameter 수 : C_in x Kernel_size^2 import torch.nn as nn class ConvBNReLU(nn.Module): def __init__(self, C_in, C_out, kernel_size, stride, .. 2021. 8. 12. 이전 1 다음
