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