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的结构

经过每一层图片的变化

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