Method DecoderBlockLinkImpl

src/architectures/LinknetDecoder.cpp:24–32

}

DecoderBlockLinkImpl::DecoderBlockLinkImpl(int in_channels, int out_channels, bool use_batchnorm) {
	conv2drelu1 = Conv2dReLU(in_channels, in_channels / 4, 1, 0, 1, use_batchnorm);
	transpose = TransposeX2(in_channels / 4, in_channels / 4, use_batchnorm);
	conv2drelu2 = Conv2dReLU(in_channels / 4, out_channels, 1, 0, 1, use_batchnorm);

	register_module("conv2drelu1", conv2drelu1);
	register_module("transpose", transpose);
	register_module("conv2drelu2", conv2drelu2);
}

torch::Tensor DecoderBlockLinkImpl::forward(torch::Tensor x, torch::Tensor skip) {
	x = conv2drelu1->forward(x);