昇思25天学习打卡营第12天|Pix2Pix实现图像转换
·
Pix2Pix
这里是基于Pytorch的pix2pix实现的github链接GitHub - junyanz/pytorch-CycleGAN-and-pix2pix: Image-to-Image Translation in PyTorch
这次课程是通过华为昇思Mindspore框架来进行pix2pix模型的训练和推理,这里有官方提供的数据集,这次的数据集经过处理,可以直接使用indspore.dataset的方法读取
生成器(使用U-Net结构)和判别器(使用PatchGAN结构),训练的时候 get_lr 函数定义了一个动态调整的学习率调度,开始时使用固定的学习率 lr,在经过 n_epochs_decay 个周期后,学习率会线性衰减到0,然后加载的数据集是官方提供好的,运行jupyterLab的cell就可以训练了,训练完再进行推理
结果

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)