SRGAN loss部分的mindspore代码实现

2024-01-07 16:38

本文主要是介绍SRGAN loss部分的mindspore代码实现,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

转载地址:https://bbs.huaweicloud.com/forum/thread-137103-1-1.html

作者: 雨丝儿

pytorch代码实现:https://bbs.huaweicloud.com/forum/forum.php?mod=viewthread&tid=137101

接上篇帖子 ,本贴分享mindspore代码的实现:

import mindspore.common.dtype as mstype

import mindspore.nn as nn

import mindspore.ops as ops

from src.vgg19.define import vgg19

from src.loss.Meanshift import MeanShift

对于Discriminator:

class DiscriminatorLoss(nn.Cell):

    """Loss for discriminator"""

    def __init__(self, discriminator, generator):

        super(DiscriminatorLoss, self).__init__()

        self.discriminator = discriminator

        self.generator = generator

        self.adversarial_criterion = nn.BCELoss()

        ones = ops.Ones()

        zeros = ops.Zeros()

        self.real_lable = ones((16, 1), mstype.float32)

        self.fake_lable = zeros((16, 1), mstype.float32)

    def construct(self, HR_img, LR_img):

        """dloss"""

        hr = HR_img

        lr = LR_img

        # Generating fake high resolution images from real low resolution images.

        sr = self.generator(lr)

        # Let the discriminator realize that the sample is real.

        real_output = self.discriminator(hr)

        d_loss_real = self.adversarial_criterion(real_output, self.real_lable)

        # Let the discriminator realize that the sample is false.

        fake_output = self.discriminator(sr)

        d_loss_fake = self.adversarial_criterion(fake_output, self.fake_lable)

        d_loss = d_loss_fake+d_loss_real

        return  d_loss

class GeneratorLoss(nn.Cell):

    """Loss for generator"""

    def __init__(self, discriminator, generator, vgg_ckpt):

        super(GeneratorLoss, self).__init__()

        self.discriminator = discriminator

        self.generator = generator

        self.mse_loss = nn.MSELoss()

        self.adversarial_criterion = nn.BCELoss()

        ones = ops.Ones()

        self.real_lable = ones((16, 1), mstype.float32)

        self.meanshif = MeanShift()

        self.vgg = vgg19(vgg_ckpt)

        for p in self.meanshif.get_parameters():

            p.requires_grad = False

    def construct(self, HR_img, LR_img):

        """gloss"""

        # L2loss

        hr = HR_img

        lr = LR_img

        sr = self.generator(lr)

        L2_loss = self.mse_loss(sr, hr)

        # adversarialloss

        fake_output = self.discriminator(sr)

        adversarial_loss = self.adversarial_criterion(fake_output, self.real_lable)

        # vggloss

        hr = (hr+1.0)/2.0

        sr = (sr+1.0)/2.0

        hr = self.meanshif(hr)

        sr = self.meanshif(sr)

        hr_feat = self.vgg(hr)

        sr_feat = self.vgg(sr)

        percep_loss = self.mse_loss(hr_feat, sr_feat)

        g_loss = 0.006*percep_loss+1e-3*adversarial_loss+L2_loss

        return  g_loss

这篇关于SRGAN loss部分的mindspore代码实现的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!


原文地址:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.chinasem.cn/article/580566

相关文章

springboot循环依赖问题案例代码及解决办法

《springboot循环依赖问题案例代码及解决办法》在SpringBoot中,如果两个或多个Bean之间存在循环依赖(即BeanA依赖BeanB,而BeanB又依赖BeanA),会导致Spring的... 目录1. 什么是循环依赖?2. 循环依赖的场景案例3. 解决循环依赖的常见方法方法 1:使用 @La

Java枚举类实现Key-Value映射的多种实现方式

《Java枚举类实现Key-Value映射的多种实现方式》在Java开发中,枚举(Enum)是一种特殊的类,本文将详细介绍Java枚举类实现key-value映射的多种方式,有需要的小伙伴可以根据需要... 目录前言一、基础实现方式1.1 为枚举添加属性和构造方法二、http://www.cppcns.co

使用Python实现快速搭建本地HTTP服务器

《使用Python实现快速搭建本地HTTP服务器》:本文主要介绍如何使用Python快速搭建本地HTTP服务器,轻松实现一键HTTP文件共享,同时结合二维码技术,让访问更简单,感兴趣的小伙伴可以了... 目录1. 概述2. 快速搭建 HTTP 文件共享服务2.1 核心思路2.2 代码实现2.3 代码解读3.

使用C#代码在PDF文档中添加、删除和替换图片

《使用C#代码在PDF文档中添加、删除和替换图片》在当今数字化文档处理场景中,动态操作PDF文档中的图像已成为企业级应用开发的核心需求之一,本文将介绍如何在.NET平台使用C#代码在PDF文档中添加、... 目录引言用C#添加图片到PDF文档用C#删除PDF文档中的图片用C#替换PDF文档中的图片引言在当

C#使用SQLite进行大数据量高效处理的代码示例

《C#使用SQLite进行大数据量高效处理的代码示例》在软件开发中,高效处理大数据量是一个常见且具有挑战性的任务,SQLite因其零配置、嵌入式、跨平台的特性,成为许多开发者的首选数据库,本文将深入探... 目录前言准备工作数据实体核心技术批量插入:从乌龟到猎豹的蜕变分页查询:加载百万数据异步处理:拒绝界面

MySQL双主搭建+keepalived高可用的实现

《MySQL双主搭建+keepalived高可用的实现》本文主要介绍了MySQL双主搭建+keepalived高可用的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,... 目录一、测试环境准备二、主从搭建1.创建复制用户2.创建复制关系3.开启复制,确认复制是否成功4.同

Java实现文件图片的预览和下载功能

《Java实现文件图片的预览和下载功能》这篇文章主要为大家详细介绍了如何使用Java实现文件图片的预览和下载功能,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... Java实现文件(图片)的预览和下载 @ApiOperation("访问文件") @GetMapping("

用js控制视频播放进度基本示例代码

《用js控制视频播放进度基本示例代码》写前端的时候,很多的时候是需要支持要网页视频播放的功能,下面这篇文章主要给大家介绍了关于用js控制视频播放进度的相关资料,文中通过代码介绍的非常详细,需要的朋友可... 目录前言html部分:JavaScript部分:注意:总结前言在javascript中控制视频播放

使用Sentinel自定义返回和实现区分来源方式

《使用Sentinel自定义返回和实现区分来源方式》:本文主要介绍使用Sentinel自定义返回和实现区分来源方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录Sentinel自定义返回和实现区分来源1. 自定义错误返回2. 实现区分来源总结Sentinel自定

Java实现时间与字符串互相转换详解

《Java实现时间与字符串互相转换详解》这篇文章主要为大家详细介绍了Java中实现时间与字符串互相转换的相关方法,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录一、日期格式化为字符串(一)使用预定义格式(二)自定义格式二、字符串解析为日期(一)解析ISO格式字符串(二)解析自定义