TensorFlow 中的 VGG16
您可以按照 Jupyter 笔记本中的代码ch-12a_VGG16_TensorFlow
。
对于 TensorFlow 中 VGG16 的所有示例,我们首先从 http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz 下载检查点文件并使用以下内容初始化变量码:
model_name='vgg_16'
model_url='http://download.tensorflow.org/models/'
model_files=['vgg_16_2016_08_28.tar.gz']
model_home=os.path.join(models_root,model_name)
dsu.download_dataset(source_url=model_url,
source_files=model_files,
dest_dir = model_home,
force=False,
extract=True)
我们还定义了一些常见的导入和变量:
from tensorflow.contrib import slim
from tensorflow.contrib.slim.nets import vgg
image_height=vgg.vgg_16.default_image_size
image_width=vgg.vgg_16.default_image_size