
颜色分类LeetCode-Arbitrary_Style_Transfer: 快速神经风格迁移(采用AdaIN层)- 基于H...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目实现了一种快速的神经艺术风格迁移算法,利用了AdaIN技术进行风格变换,支持多样化的图像处理需求。基于H代码库开发。
颜色分类LeetCode任意式传输Arbitrary-Style-Per-Model快速神经风格迁移方法描述使用Encoder-AdaIN-Decoder架构的深度卷积神经网络作为风格转移网络(STN)。该网络可以接收两个任意图像作为输入(一个为内容,另一个为样式),并输出重新组合的内容和前者的空间结构以及后者的风格(颜色、纹理)。无需对网络进行重新训练。
STN使用MS-COCO数据集和WikiArt数据集进行训练。此代码基于Huang等人在ICCV2017上的系统总览。图片来自黄等人的原论文。
编码器是一个固定的VGG-19模型,最多到relu4_1层,在ImageNet数据集上进行了预训练以用于图像分类任务。我们通过训练解码器将AdaIN输出从特征空间转换回图像空间来完成整个过程。
在工具文件夹中提供了一个转换器,它可以提取来自Torch模型文件的内核和偏差,并将其保存为npz文件,以便于使用NumPy处理。
全部评论 (0)
还没有任何评论哟~


