登录 注册
当前位置:主页 > 资源下载 > 9 > 神经风格转换在PyTorch中的实现:style-transfer-pytorch

神经风格转换在PyTorch中的实现:style-transfer-pytorch

  • 更新:2024-12-12 15:21:38
  • 大小:53KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

风格转移火炬 在PyTorch中实现神经风格转换(),支持CPU和Nvidia GPU。 它会自动进行多尺度(从粗到细)样式化,以生成高质量的高分辨率样式化,如果GPU有足够的内存,甚至可以达到打印分辨率。 如果有两个GPU,则它们都可以用来提高最大分辨率。 (使用两个GPU并不比使用一个GPU快。) 该算法已通过以下方式从文献中进行了修改: 使用PyTorch预训练的VGG-19砝码代替原始的VGG-19砝码 将VGG-19第一层的填充模式更改为“复制”,以减少边缘伪像 使用平均值池或L2池时,请根据经验得出的因子对结果进行缩放,以确保结果的幅度平均保持不变(Gatys等人(2015年)未执行此操作) 使用MSE损失的近似值,使其内容和样式损失的梯度L1范数约为1(以近似梯度归一化的效果,从而产生更好的视觉质量) 通过每个特征图通道中的元素数量而不是元素总数(Johnson等人)