Method PANDecoderImpl

src/architectures/PANDecoder.cpp:98–108

}

PANDecoderImpl::PANDecoderImpl(std::vector<int> encoder_channels, int decoder_channels, std::string upscale_mode) {
	fpa = FPABlock(encoder_channels[encoder_channels.size() - 1], decoder_channels);
	gau3 = GAUBlock(encoder_channels[encoder_channels.size() - 2], decoder_channels, upscale_mode);
	gau2 = GAUBlock(encoder_channels[encoder_channels.size() - 3], decoder_channels, upscale_mode);
	gau1 = GAUBlock(encoder_channels[encoder_channels.size() - 4], decoder_channels, upscale_mode);

	register_module("fpa", fpa);
	register_module("gau3", gau3);
	register_module("gau2", gau2);
	register_module("gau1", gau1);
}

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