def _contents_loss(self, layers, content_layers):
"""
calculate content difference between original & processing
"""
loss_contents = chainer.Variable(self.xp.zeros((), dtype=numpy.float32))
for layer_index in range(len(layers)):
loss_contents += numpy.float32(self.model.alpha[layer_index]) * chainer.functions.mean_squared_error(
layers[layer_index],
content_layers[layer_index])
return loss_contents
image_converter.py 文件源码
python
阅读 18
收藏 0
点赞 0
评论 0
评论列表
文章目录