【任意实时风格迁移】AdaAttN的实现(ICCV2021:Revisit Attention Mechanism in Arbitrary Neural S)

在论坛中已有众多文章介绍“AdaAttN：基于注意力机制和归一化来保持内容结构任意实时风格迁移“的原理并已贴出实现代码的github地址，但由于某些众所周知的原因，对于新手小白进入GitHub下载代码并跑通难度较大，耗费时间很多，所以本文将根据原作者在github中上传的方法与代码提供AdaAttN的详细实现方法与步骤。

1.github代码地址：

AdaAttN/README.md at main · Huage001/AdaAttN · GitHub

2.下载/克隆代码到本地

点击网址进入后，可点击网页中部绿色按钮“code”，选中下拉菜单中的“download zip”下载完整代码包，或通过如下命令克隆代码。

git clone https://github.com/Huage001/AdaAttN
cd AdaAttN

3.创建检查目录checkpoints

进入终端，使用cd指令进入刚刚下载的AdaAttN文件目录下，如用命令创建一个检查节点目录文件。

mkdir checkpoints

4.下载预训练模型

预训练模型下载地址：

链接: https://pan.baidu.com/s/1zMT-7sT7bsHkwWYMM3kqUQ?pwd=brmf 提取码: brmf

由于使用Google drive的方式下载较慢所以此处直接提供了下载好的模型在百度云盘中，下载好后使用如下命令将预训练模型文件放在刚刚创建的检查目录checkpoints下。

mv [Download Directory]/AdaAttN_model.zip checkpoints/
unzip checkpoints/AdaAttN_model.zip
rm checkpoints/AdaAttN_model.zip

注意：命令第一行目的是将下载目录“Download Directory”下的AdaAttN_model.zip移动至检查目录下，第二行目的是解压AdaAttN_model.zip，第三行是将压缩包删除，不熟悉Linux指令的同学可自行移动文件操作即可。

该项操作后环境配置完成，正确的文件目录如下：

5.运行模型

修改测试shell文件（test_adaattn.sh）中测试图片的风格图片（style_path）与内容图片(style_path)的地址，将其修改为自己想测试的图片地址并运行test_adaattn.sh，即可开始对图像进行实时风格迁移！