TensorFlow与Inception-v3

本文介绍了如何利用预训练的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(
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值