Skip to content

Gerry-Pan/pan-dl4j

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 

Repository files navigation

GRU

2017年12月,在dl4j基础上实现一些算法,根据GRU前向公式推导反向公式,并在dl4j中实现。
https://github.com/Gerry-Pan/pan-dl4j/blob/master/src/main/java/personal/pan/dl4j/nn/layers/recurrent/GRUHelpers.java

Gan示例

结构:
image

1、Generator和Discriminator使用Dense
https://github.com/Gerry-Pan/pan-dl4j/blob/master/src/main/java/personal/pan/dl4j/examples/gan/GanTrainer.java

训练效果:
输入图片说明 输入图片说明

2、Generator使用上采样和Cnn,Discriminator使用Cnn
https://github.com/Gerry-Pan/pan-dl4j/blob/master/src/main/java/personal/pan/dl4j/examples/gan/ConvGanTrainer.java

训练效果:
image

进一步调优训练结果,笔画更清晰:
QQ图片20210518071401 QQ图片20210518071407

CGan示例

1、Generator使用上采样和Cnn,Discriminator使用Cnn
https://github.com/Gerry-Pan/pan-dl4j/blob/master/src/main/java/personal/pan/dl4j/examples/gan/ConditionalConvGanTrainer.java

训练效果:
QQ图片20210518224427 QQ图片20210518224430 QQ图片20210518224436

WGan示例

1、Generator使用上采样和Cnn,Discriminator使用Cnn
https://github.com/Gerry-Pan/pan-dl4j/blob/master/src/main/java/personal/pan/dl4j/examples/gan/WassersteinConvGanTrainer.java

训练效果(效果欠佳,有待改善):
QQ图片20210519232337 QQ图片20210519232351


About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages