vgg16 finetune tensorflow实现

vgg

vggnet version1
github: https://github.com/chenlongzhen

代码

环境:python3 keras2(tensorflow backend)

结构

name usage
data_utils 参数文件下载等,vgg16调用
vgg16.py vgg核心代码
vgg16_fintune.py vgg finetune
vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5 vgg 默认参数,需要放在 ~/.keras/models下
cnn_dog_cat_demo.py keras cnn基础例子

调用

python vgg16_fintune.py

dogVScat 例子

  1. 图片文件结构如下
data/
    train/
        dogs/
            dog001.jpg
            dog002.jpg
            ...
        cats/
            cat001.jpg
            cat002.jpg
            ...
    validation/
        dogs/
            dog001.jpg
            dog002.jpg
            ...
        cats/
            cat001.jpg
            cat002.jpg
  1. vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5 | vgg 默认参数,需要放在 ~/.keras/models下
  2. python vgg16_fintune.py
  3. 参数设置python vgg16_fintune.py -h

vgg 结构

这里写图片描述

Alex的结构

image

经过每一层图片的变化

reference

[1]. https://jacobgil.github.io/deeplearning/filter-visualizations
[2]. https://nbviewer.jupyter.org/gist/embanner/6149bba89c174af3bfd69537b72bca74
[3]. https://icmlviz.github.io/assets/papers/4.pdf

你可能感兴趣的:(tensorflow,deeplearning,cnn)