Class PSPDecoderImpl

src/architectures/PSPNetDecoder.h:22–31

}; TORCH_MODULE(PSPModule);

class PSPDecoderImpl : public torch::nn::Module {
public:
	PSPDecoderImpl(std::vector<int> encoder_channels, int out_channels = 512, double dropout = 0.2, bool use_batchnorm = true, int encoder_depth = 3);
	torch::Tensor forward(std::vector<torch::Tensor> x);
private:
	int encoder_depth = 3;
	PSPModule psp{ nullptr };
	torch::nn::Sequential conv{ nullptr };
	torch::nn::Dropout2d dropout{ nullptr };
}; TORCH_MODULE(PSPDecoder);