Class DeepLabV3PlusDecoderImpl

src/architectures/DeepLabDecoder.h:83–94

}; TORCH_MODULE(DeepLabV3Decoder);

class DeepLabV3PlusDecoderImpl :public torch::nn::Module {
public:
	DeepLabV3PlusDecoderImpl(std::vector<int> encoder_channels, int out_channels,
		std::vector<int> atrous_rates, int output_stride = 16);
	torch::Tensor forward(std::vector< torch::Tensor> x);
private:
	ASPP aspp{ nullptr };
	torch::nn::Sequential aspp_seq{ nullptr };
	torch::nn::Upsample up{ nullptr };
	torch::nn::Sequential block1{ nullptr };
	torch::nn::Sequential block2{ nullptr };
}; TORCH_MODULE(DeepLabV3PlusDecoder);