深度学习论文: LEDnet: A lightweight encoder-decoder network for real-time semantic segmentation及其PyTorch实现
classHalfSplit(nn.Module):def__init__(self,dim=1):super(HalfSplit,self).__init__()self.dim=dimdefforward(self,input):splits=torch.chunk(input,2,dim=self.dim)returnsplits[0],splits[1]classChannelShuffl