13、使用CycleGAN进行图像风格迁移

使用CycleGAN进行图像风格迁移

1. 环境搭建与数据准备

在开始使用CycleGAN进行图像风格迁移之前，我们需要进行一些环境搭建和数据准备工作。

首先，构建Docker镜像。在终端中执行以下命令，使构建脚本可执行并运行：

sudo chmod 775 build.sh && ./build.sh

在执行这些命令之前，请确保你位于 docker 文件夹中。

为了方便后续清理镜像，我们可以创建一个名为 clean.sh 的脚本：

#!/bin/bash
docker rmi ch5

这个脚本可以在需要重新构建镜像时，轻松清理现有的镜像。

接下来，使用新构建的容器下载数据。运行以下脚本：

#!/bin/bash
xhost +
docker run -it \
   --runtime=nvidia \
   --rm \
   -v /home/jk/Desktop/book_repos/Chapter5/data:/data \
   ch5 wget -N https://raw.githubusercontent.com/junyanz/CycleGAN/master/datasets/download_dataset.sh -O /data/download_datas