image_converter.py 文件源码

def _texture_loss(self, layers):
        loss_texture = chainer.Variable(self.xp.zeros((), dtype=self.xp.float32))
        for layer_index in range(len(layers)):
            matrix = neural_art.utility.get_matrix(layers[layer_index])
            loss = self.xp.float32(self.model.beta[layer_index]) * chainer.functions.mean_squared_error(
                matrix,
                self.texture_matrices[layer_index]
            ) / self.xp.float32(len(layers))
            loss_texture += loss
        print("loss_texture", loss_texture.data)
        return loss_texture