Method LinknetDecoderImpl

src/architectures/LinknetDecoder.cpp:43–54

}

LinknetDecoderImpl::LinknetDecoderImpl(std::vector<int> encoder_channels, int prefinal_channels,
	int n_blocks, bool use_batchnorm) {
	encoder_channels = std::vector<int>(encoder_channels.begin()+1, encoder_channels.end());
	std::reverse(std::begin(encoder_channels), std::end(encoder_channels));
	std::vector<int> channels = encoder_channels;
	channels.push_back(prefinal_channels);
	for (int i = 0; i < n_blocks; i++) {
		blocks->push_back(DecoderBlockLink(channels[i], channels[i + 1], use_batchnorm));
	}

	register_module("blocks", blocks);
}

torch::Tensor LinknetDecoderImpl::forward(std::vector< torch::Tensor> features) {
	features = std::vector<torch::Tensor>(features.begin() + 1, features.end());