Docker-tensorflow跑VGG16

标签: VGG docker tensorflow
4661人阅读 评论(7) 收藏 举报
分类:

在我们的实际项目中,一般不会直接从第一层直接开始训练,而是通过在大的数据集上(如ImageNet)训练好的模型,把前面那些层的参数固定,在运用到我们新的问题上,修改最后一到两层,用自己的数据去微调(finetuning),一般效果也很好。

如果要在docker上的tensorfow上跑,我们需要做以下几件事情:

1.下载好VGG16的权重文件

一般是 VGG16_Weights.npy ,或者VGG_Weights.npz, 可以自行百度上搜,文件比较大,我下的是537M,外网比较慢的话,可以从百度网盘上下;

(补充一个VGG下载链接(墙外):https://mega.nz/#!YU1FWJrA!O1ywiCS2IiOlUCtCpI6HTJOMrneN-Qdv3ywQP5poecM


2. 把主机上包含VGG16_Weights.npy权重的文件夹挂载到tensorflow上

由于文件很大,不可能直接上传,docker给镜像提供了挂载机制,具体操作是通过docker -v 主机地址:挂载目录的方式,以我的为例:


docker run -it - p 8888:8888 -v /home/zhu/Downloads:/aa/data tensorflow/tensorflow


通过这种方式,我就把我Downloads文件夹挂载到了tensorflow上,就可以直接引用了;

可以检查以下是否挂载成功:

# cd  /aa/data
# ls
data   data1(1)  input_data.py     rtl8192eu-master.zip   tensorflow-vgg-master.zip  vgg16_weights.h5
data1  data2     rtl8192eu-master  tensorflow-vgg-master  vgg16.py                   vgg16_weights.npz



3. 定义VGG网络结构

github上有很多开源代码,这里给出一个https://github.com/leihe001/tensorflow-vgg


4. 根据需要修改网络结构


5.fine-tuning VGG

finetuning的流程可以参见 http://blog.csdn.net/andrewseu/article/details/70142950 caffe finetuning CaffeNet流程总结


补充:

不fine-tuning的情况下提取VGG特征



参考资料:

1. https://my.oschina.net/piorcn/blog/324202(docker挂载文件目录)

2. http://scikit-image.org/docs/dev/install.html(scikit-image)

3. https://github.com/leihe001/tensorflow-vgg(VGG16 & VGG19)

4. http://www.cnblogs.com/dmir/p/5009075.html(.npy&.npz文件)

查看评论

Tensorflow学习笔记:CNN篇(9)——Finetuning,复用ImageNet的VGGNet进行图像识别

Tensorflow学习笔记:CNN篇(9)——Finetuning,复用在ImageNet已训练好的VGGNet进行图像识别 前序 — 到目前为止,对于模型的设计和训练,读者可能已经较为...
  • Laurenitum0716
  • Laurenitum0716
  • 2018-02-14 17:35:14
  • 478

使用vgg16模型进行图片预测

使用vgg16模型进行图片预测
  • u013473520
  • u013473520
  • 2016-02-24 14:27:36
  • 42772

vgg16 包括网络参数的导入

1. vgg16 含有导入网络参数 2.值得学习的部分为: 2.1 参数移植, 但不太了解assign的用法 sess.run(self.parameters[i].assign(w...
  • wuguangbin1230
  • wuguangbin1230
  • 2017-03-22 21:57:41
  • 1465

vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5

  • 2017年08月05日 08:11
  • 56.16MB
  • 下载

vgg16模型以及用法代码

vgg16的权重文件下载。格式为.npz格式,也包含了实现图片检测的代码
  • 2018年04月11日 09:14

Tensorflow学习笔记:CNN篇(10)——Finetuning,猫狗大战,VGGNet的重新针对训练

Tensorflow学习笔记:CNN篇(10)——Finetuning,猫狗大战,VGGNet的重新针对训练 前序 — 在前面的例子中,对使用已在ImageNet上训练好的VGGNet模型...
  • Laurenitum0716
  • Laurenitum0716
  • 2018-02-14 23:26:40
  • 618

vgg16.npy下载

最近看fcn网络的时候,需要使用vgg16.npy。 而下载它真的是个难受的事情, 下载速度慢而且经常下载失败, 把它存到云盘希望可以帮助到有需要的人。 链接:https://pan.baid...
  • isyiming
  • isyiming
  • 2018-02-02 12:06:52
  • 634

TensorFlow VGG-16 预训练模型

在我们的实际项目中,一般不会直接从第一层直接开始训练,而是通过在大的数据集上(如ImageNet)训练好的模型,把前面那些层的参数固定,在运用到我们新的问题上,修改最后一到两层,用自己的数据去微调(f...
  • daydayup_668819
  • daydayup_668819
  • 2017-04-18 10:14:13
  • 22128

VGG_imagenet.npy

  • 2018年03月14日 13:59
  • 70B
  • 下载

自动下载解压的python logging download os ,一份比较好的python下载解压代码

复制文件 #copy the file "test.py" in now path to the child path mytest/ if not os.path.exists("mytest")...
  • cai13160674275
  • cai13160674275
  • 2017-03-09 14:42:43
  • 1171
    个人资料
    等级:
    访问量: 37万+
    积分: 5383
    排名: 6279
    最新评论