TensorFlow与Inception-v3

最新推荐文章于 2022-01-20 18:13:13 发布

原创

最新推荐文章于 2022-01-20 18:13:13 发布 · 326 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#AI #TensorFlow #inception-v3 #图像分类 #计算机视觉CV

本文介绍了如何利用预训练的Inception-v3模型进行图像分类。由于模型是在高端GPU设备上训练的，不适合普通PC训练。内容包括介绍Inception-v3模型，提供数据集下载链接，展示下载模型并进行图像分类的代码，以及如何通过TensorBoard查看模型结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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(