昇思25天学习打卡营第18天|Pix2Pix实现图像转换

人工智能mindspore框架下Pix2Pix模型实现真实图到线稿图的转换

（一）dataset_pix2pix数据集下载
（二）Pix2Pix模型构建
（三）Pix2Pix模型训练
（四）模型推理

dataset_pix2pix数据集下载

tips

pip install -i https://pypi.mirrors.ustc.edu.cn/simple mindspore==2.2.14
pip install download

数据下载：

from download import download

url = "https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/notebook/models/application/dataset_pix2pix.tar"

download(url, "./dataset", kind="tar", replace=True)

数据展示

from mindspore import dataset as ds
import matplotlib.pyplot as plt

dataset = ds.MindDataset("./dataset/dataset_pix2pix/train.mindrecord", columns_list=["input_images", "target_images"], shuffle=True)
data_iter = next(dataset.create_dict_iterator(output_numpy=True))
# 可视化部分训练数据
plt.figure(figsize=(10, 3), dpi=140)
for i, image in enumerate(data_iter['input_images'][:10], 1):
    plt.subplot(3, 10, i)
    plt.axis("off")
    plt.imshow((image.transpose(1, 2, 0) + 1) / 2)
plt.show()