Function SeparableConv2d

src/architectures/DeepLabDecoder.cpp:12–18

}

torch::nn::Sequential SeparableConv2d(int in_channels, int out_channels, int kernel_size, int stride,
	int padding, int dilation, bool bias) {
	torch::nn::Conv2d dephtwise_conv = torch::nn::Conv2d(conv_options(in_channels, in_channels, kernel_size,
		stride, padding, 1, false, dilation));
	torch::nn::Conv2d pointwise_conv = torch::nn::Conv2d(conv_options(in_channels, out_channels, 1, 1, 0, 1, bias));
	return torch::nn::Sequential(dephtwise_conv, pointwise_conv);
};

torch::nn::Sequential ASPPSeparableConv(int in_channels, int out_channels, int dilation) {
	torch::nn::Sequential seq = SeparableConv2d(in_channels, out_channels, 3, 1, dilation, dilation, false);