tensorflow学习笔记——使用inception-v3作图像识别

最新推荐文章于 2025-01-02 14:49:38 发布

红鱼鱼

最新推荐文章于 2025-01-02 14:49:38 发布

阅读量921

点赞数 5

分类专栏： tensorflow

本文链接：https://blog.youkuaiyun.com/qq_40692109/article/details/104169424

版权

本文介绍了如何下载并使用预训练的inception-v3模型进行图像识别。首先，从官方下载模型并解压，接着利用TensorBoard查看模型结构，最后展示如何进行图像分类并解释了模型中类别映射的文本文件用途。

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

1.下载inception-v3

从官网下载训练好的inception-v3模型用于之后的训练，代码如下：（也可直接复制网址http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz下载）

import tensorflow as tf
import os
import tarfile
import requests

#inception模型下载地址
inception_pretrain_model_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz'

#模型存放地址
inception_pretrain_model_dir = "inception_model"
if not os.path.exists(inception_pretrain_model_dir):
    os.makedirs(inception_pretrain_model_dir)
    
#获取文件名，以及文件路径
filename = inception_pretrain_model_url.split('/')[-1]
filepath = os.path.join(inception_pretrain_model_dir, filename)

#下载模型
if not os.path.exists(filepath):
    print("download: ", filename)
    r = requests.get(inception_pretrain_model_url, stream=True)
    with open(filepath, 'wb') as f:
        for chunk in r.iter_content(chunk_size=1024):
            if chunk:
                f.write(chunk)
print("finish: ", filename)
#解压文件
tarfile.open(filepath, 'r:gz').extractall(inception_pretrain_model_dir)

解压完后，对应路径文件如下：