Skip to content

fvilmos/ResNet_keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ResNet implementation in Keras

This is an unoffcial implementation of the ResNet in Keras (concept and architectue: https://arxiv.org/pdf/1512.03385.pdf).

This implementation alows in a fexible way to generate the ResNet 18,34,50,101,152 model variants.

Other ResNet configurations (like number of resnet blocks, or filter number) can be easily changed. Identity shortcuts on dimension transition use 1x1 convolution to match the desired output, see the paper for details.

How to use it?

  1. clone the project
  2. select one of the architectures (RESNET_MODELS = {"RESNET_18":0,"RESNET_34":1,"RESNET_50":2,"RESNET_101":3,"RESNET_152":4}) with i.e. SELECTED_ARCHITECTURE = RESNET_MODELS['RESNET_18']
  3. or customize the block / filter lists from the notebook. Default configuration is the ResNet18 configuration block_list = [2,2,2,2] and filter_list = [64,128,256,512]
  4. Train the model on your data and use it in your projects

Resources

  1. Deep Residual Learning for Image Recognition, Kaiming He, et all

/Enjoy.

About

ResNet implementation in Keras, alows a fexible way to generate the ResNet 18/34/50/101/152 or other model variants.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors