昇思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()

原文地址：https://blog.csdn.net/beauthy/article/details/140557420

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！