pix2pix在keras上的训练以及优化效果过程

2024-01-23 12:20

本文主要是介绍pix2pix在keras上的训练以及优化效果过程,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

pix2pix在keras上训练

  • 1 背景
  • 2 初步训练
  • 3 优化
    • 3.1. 工具类优化:
    • 3.2 算法类优化
  • 4. 优化效果
  • 5. 存在的问题
  • 6. 如果有什么建议可以评论联系我,一起来讨论下.

1 背景

pix2pix提供的示例是使用tensorflow,我我参考着来做了一个keras的实现,原项目中使用的是生成网络加上判别网络来提升生成网络,而我观察实际在训练中判别部分起到的作用比较小,所以我先尝试只使用生成网络来做训练.
具体的项目代码在:pix2pix-keras

2 初步训练

下图为迭代了80次之后在训练下的一个效果,纹理上是逐渐变清晰的.
在这里插入图片描述
但是问题是左上部分的几乎没到训练出效果.

3 优化

此问题的根本是在最后一层时左上部分的输出值过大.

为了定位解决此问题我尝试了几个角度的优化:

3.1. 工具类优化:

- 将所有layer的输出绘制成图像打出,用于观察在那一层开始出现的偏差
- 将所有layer的weight输出,用于观察是如何滚最后一层的时候出现过大数值.
- 将训练过程中每次样本产生梯度更新是的输入输出画出来观察是从那些步骤开始出现问题.
![将没一层输出可视化](https://img-blog.csdnimg.cn/2019122015281159.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2hscG93ZXI=,size_16,color_FFFFFF,t_70)![将所有训练步骤产生的跟新全部打出来](https://img-blog.csdnimg.cn/20191220150910498.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2hscG93ZXI=,size_16,color_FFFFFF,t_70)

工具类代码均更新在git上.

最终定位到是是部分样本上由于光照和裁剪的问题
如遇到以下的样本,对训练引起的偏差较大.
图片中最下层是容易一起训练跑偏的样本,分析如,周边有几个黑色三角型的剪裁.将建筑物遮挡过多,但输入是未处理这些遮挡
在这里插入图片描述

3.2 算法类优化

为了更好训练,我重新审核了一次训练样本,将其中比较差的样本剔除,然后调整loss从mse到mae,之前使用平方是为了加大对误差的灵敏度,后来发现最后输出的数值过小平方反而降低了训练效率.另外为了提高训练速度吧batch该为了2,这个是我的内存能够加载的上限.

4. 优化效果

loss曲线如下:
在这里插入图片描述
不同epho下测试数据上效果:
在这里插入图片描述
总的来说相对于之前的效果好了不少.

5. 存在的问题

我把测试集所有的建筑物全部过了一遍发现以下问题:

  1. 纹理特征还是有一点模糊.
  2. 会丢失一些建筑物特有特征,如屋檐,窗帘,阳台等
  3. 输出墙面基本都是以灰土色为主,原本建筑物可能是红色墙面
    在这里插入图片描述
    在这里插入图片描述
    原因分析:
  4. 细致话的特征样本较少,如阳台虽然都拿绿色标记,但是不同建筑物的阳台各有风格,少量的样本对这种细的纹理特征较难学习到.

6. 如果有什么建议可以评论联系我,一起来讨论下.

这篇关于pix2pix在keras上的训练以及优化效果过程的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/636390

相关文章

Spring Boot + MyBatis Plus 高效开发实战从入门到进阶优化(推荐)

《SpringBoot+MyBatisPlus高效开发实战从入门到进阶优化(推荐)》本文将详细介绍SpringBoot+MyBatisPlus的完整开发流程,并深入剖析分页查询、批量操作、动... 目录Spring Boot + MyBATis Plus 高效开发实战:从入门到进阶优化1. MyBatis

MyBatis 动态 SQL 优化之标签的实战与技巧(常见用法)

《MyBatis动态SQL优化之标签的实战与技巧(常见用法)》本文通过详细的示例和实际应用场景,介绍了如何有效利用这些标签来优化MyBatis配置,提升开发效率,确保SQL的高效执行和安全性,感... 目录动态SQL详解一、动态SQL的核心概念1.1 什么是动态SQL?1.2 动态SQL的优点1.3 动态S

C# WinForms存储过程操作数据库的实例讲解

《C#WinForms存储过程操作数据库的实例讲解》:本文主要介绍C#WinForms存储过程操作数据库的实例,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、存储过程基础二、C# 调用流程1. 数据库连接配置2. 执行存储过程(增删改)3. 查询数据三、事务处

JSON Web Token在登陆中的使用过程

《JSONWebToken在登陆中的使用过程》:本文主要介绍JSONWebToken在登陆中的使用过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录JWT 介绍微服务架构中的 JWT 使用结合微服务网关的 JWT 验证1. 用户登录,生成 JWT2. 自定义过滤

Python如何使用__slots__实现节省内存和性能优化

《Python如何使用__slots__实现节省内存和性能优化》你有想过,一个小小的__slots__能让你的Python类内存消耗直接减半吗,没错,今天咱们要聊的就是这个让人眼前一亮的技巧,感兴趣的... 目录背景:内存吃得满满的类__slots__:你的内存管理小助手举个大概的例子:看看效果如何?1.

一文详解SpringBoot响应压缩功能的配置与优化

《一文详解SpringBoot响应压缩功能的配置与优化》SpringBoot的响应压缩功能基于智能协商机制,需同时满足很多条件,本文主要为大家详细介绍了SpringBoot响应压缩功能的配置与优化,需... 目录一、核心工作机制1.1 自动协商触发条件1.2 压缩处理流程二、配置方案详解2.1 基础YAML

java中使用POI生成Excel并导出过程

《java中使用POI生成Excel并导出过程》:本文主要介绍java中使用POI生成Excel并导出过程,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录需求说明及实现方式需求完成通用代码版本1版本2结果展示type参数为atype参数为b总结注:本文章中代码均为

MySQL中慢SQL优化的不同方式介绍

《MySQL中慢SQL优化的不同方式介绍》慢SQL的优化,主要从两个方面考虑,SQL语句本身的优化,以及数据库设计的优化,下面小编就来给大家介绍一下有哪些方式可以优化慢SQL吧... 目录避免不必要的列分页优化索引优化JOIN 的优化排序优化UNION 优化慢 SQL 的优化,主要从两个方面考虑,SQL 语

MySQL中慢SQL优化方法的完整指南

《MySQL中慢SQL优化方法的完整指南》当数据库响应时间超过500ms时,系统将面临三大灾难链式反应,所以本文将为大家介绍一下MySQL中慢SQL优化的常用方法,有需要的小伙伴可以了解下... 目录一、慢SQL的致命影响二、精准定位问题SQL1. 启用慢查询日志2. 诊断黄金三件套三、六大核心优化方案方案

SpringCloud之LoadBalancer负载均衡服务调用过程

《SpringCloud之LoadBalancer负载均衡服务调用过程》:本文主要介绍SpringCloud之LoadBalancer负载均衡服务调用过程,具有很好的参考价值,希望对大家有所帮助,... 目录前言一、LoadBalancer是什么?二、使用步骤1、启动consul2、客户端加入依赖3、以服务