Method PSPDecoderImpl

src/architectures/PSPNetDecoder.cpp:40–50

}

PSPDecoderImpl::PSPDecoderImpl(std::vector<int> encoder_channels, int out_channels, double _dropout,
	bool use_batchnorm, int _encoder_depth) {
	encoder_depth = _encoder_depth;
	std::vector<int> size = { 1, 2, 3, 6 };
	psp = PSPModule(encoder_channels[encoder_depth], size, use_batchnorm);
	conv = Conv2dReLU(encoder_channels[encoder_depth] * 2, out_channels, 1, 0, 1, use_batchnorm);
	dropout = torch::nn::Dropout2d(torch::nn::Dropout2dOptions(_dropout));

	register_module("psp", psp);
	register_module("conv", conv);
}

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