Function ASPPSeparableConv

src/architectures/DeepLabDecoder.cpp:20–25

};

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);
	seq->push_back(torch::nn::BatchNorm2d(torch::nn::BatchNorm2dOptions(out_channels)));
	seq->push_back(torch::nn::ReLU());
	return seq;
}

ASPPPoolingImpl::ASPPPoolingImpl(int in_channels, int out_channels) {
	seq = torch::nn::Sequential(torch::nn::AdaptiveAvgPool2d(torch::nn::AdaptiveAvgPool2dOptions(1)),