下载Inception

import tensorflow as tf
import requests
import os
import tarfile

inception_url="http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz";
inception_url_dir="model/";
if not os.path.exists(inception_url_dir):
	os.makedirs(inception_url_dir);
inception_url_dir_file=os.path.join(inception_url_dir,inception_url.split("/")[-1]);
if not os.path.exists(inception_url_dir_file):
	print("Download:",inception_url.split("/")[-1]);
	r=requests.get(inception_url,stream=True);
	with open(inception_url_dir_file,'wb') as f:
		for chunk in r.iter_content(chunk_size=1024):
			if chunk:
				f.write(chunk);

	print("finished!!!");

tarfile.open(inception_url_dir_file,"r:gz").extractall(inception_url_dir);
inception_log="inception_log";
if not os.path.exists(inception_log):
	os.makedirs(inception_log);

model_pb_path=os.path.join(inception_url_dir,"classify_image_graph_def.pb");
with tf.Session() as sess:
	with tf.gfile.GFile(model_pb_path,"rb") as f:
		graph_def=tf.GraphDef();
		graph_def.ParseFromString(f.read());
		tf.import_graph_def(graph_def);

	writer=tf.summary.FileWriter(inception_log,sess.graph);
	writer.close();
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值