Inception-v3
Inception-v3模型是由Google公司在一台配有 8 Tesla K40 GPUs,大概价值$30,000的野兽级计算机上训练了几个星期,因此不可能在一台普通的PC上训练。我们将会下载预训练好的Inception模型,然后用它来做图像分类。
数据集
如果自己没有数据集,可以在www.robots.ox.ac.uk/~vgg/data/上下载。
代码
import tensorflow as tf
import os
import tarfile
import requests
#incption_v3模型下载地址
inception_pretrain_mode_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz'
#模型存放地址
inception_pretrain_mode_dir = "inception_model"
if not os.path.exists(inception_pretrain_mode_dir):
os.makedirs(inception_pretrain_mode_dir)
#获取文件名,以及文件路径
filename = inception_pretrain_mode_url.split('/')[-1]
filepath = os.path.join(inception_pretrain_mode_dir,filename)
#下载模型
if not os.path.exists(filepath):
print("download:",filename)
r = requests.get(inception_pretrain_mode_url)
with open(filepath,'wb') as f:
for chunk in r.iter_content(chunk_size=1024):
if chunk:
f.write(