Method DeepLabV3DecoderImpl

src/architectures/DeepLabDecoder.cpp:81–88

}

DeepLabV3DecoderImpl::DeepLabV3DecoderImpl(int in_channels, int out_channels, std::vector<int> atrous_rates) {
	seq->push_back(ASPP(in_channels, out_channels, atrous_rates));
	seq->push_back(torch::nn::Conv2d(conv_options(out_channels, out_channels, 3, 1, 1, 1, false)));
	seq->push_back(torch::nn::BatchNorm2d(torch::nn::BatchNorm2dOptions(out_channels)));
	seq->push_back(torch::nn::ReLU());

	register_module("seq", seq);
}

torch::Tensor DeepLabV3DecoderImpl::forward(std::vector< torch::Tensor> x_list) {
	auto x = seq->forward(x_list[x_list.size() - 1]);