Skip to content

praveenkumarchandaliya/res_ae

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ResAE--Residual Auto Encoder

This repo aims to realize the auto encoder structure with Residual Network (ResNet) by the TensorFlow library. The residual loss strategy in the ResNet definately benefits the training for very deep neural networks and keep it from the saturation and performance decreasing problems.

Which techs will be embedded in the network?

Some tricks will be applied for better training the network to avoide over-fitting, accelerate convergence to local optima, etc.

Construction of the ResAE

Firstly the bottleneck structure, the basis of the ResNet, is realized. Then the block class is built, composed of multiple bottlenecks. By means of them, the ResNet based encoding part can be constructed. As for the decoder part, which is conventionally as the symmetry as the encoder, it can be formed by reversing the encoder. A diagram is illustrated as follow, which is similar to the famous skip connection

Packages to be used

Some python packages should be installed before appying the nets, which are listed as follows,

Also, CUDA is required if you want to run the codes by GPU, a Chinese guide is provided here..

Usage

Author

  • Zhixian MA <zx at mazhixian.me>

License

Unless otherwise declared:

About

Residual Auto Encoder...

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 64.5%
  • Python 35.5%