unet.py 文件源码

def forward(self, x):

        down_outputs = []
        for i in range(len(self.down)):
            down_output = self.down[i](x)
            down_output = self.dropout2d(down_output)
            down_outputs.append(down_output)

            if i < len(self.pool):
                x = self.pool[i](down_output)

        x = down_outputs[-1]
        for i in reversed(range(len(self.up))):
            x = self.up[i](down_outputs[i], x)
            x = self.dropout2d(x)

        out =  self.classify(x)
        return F.sigmoid(out)