Tensorflow 的安装和用InceptionV3训练新的图像分类模型

本文详细介绍了Tensorflow的安装步骤,包括CUDA和CUDNN的安装,以及在Ubuntu14.04上的GPU版本安装。接着,讲解了如何在Tensorflow上使用InceptionV3进行图像分类的模型训练,通过迁移学习方法,提取特征并训练softmax分类器。文章结尾分享了作者在学习Tensorflow过程中的心得和经验。
摘要由CSDN通过智能技术生成

Tensorflow的安装

1.Tensorflow简介

Tensorflow是一个谷歌发布的人工智能开发工具,于2015年年底开源。在开源之前一直是在谷歌内部使用,维护性比较好,里面的很多工具也比较新。Tensorflow是采用C++python写成的,给的接口也是C++python,但是更支持python。下面是Tensorflow和其他几个深度学习或机器学习工具的比较:


 

谷歌称以后的学术文章也大都使用该工具,所以如果要跟进谷歌的学术进展,可能要对这个工作做一定了解的。

 Tensorflow安装

Tensorflow目前只能在Linux或者Mac下使用,其安装还是比较简单的,它提供了CPU版本和GPU版本。如果电脑中有GPU,可以尝试使用GPU版本,这个略微复杂。如果并不想跑太过于复杂的程序,或的电脑GPU也并不是很好,推荐CPU版本,安装简单。下面是在Ubuntu14.04下一个GPU版本的安装方法。安装过程中请联网(fanqiang):

一.在安装Tensorflow之前先安装CUDACUDNN.

1.安装cuda

https://developer.nvidia.com/cuda-downloads中,选择和自己系统相对应的且合适的CUDA版本。我的选择依次是:Linux—x86_64—ubuntu—14.04—deb(local).选择好后开始下载。下载好后进入下载的目录中在命令行输入如下命令:

sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb

sudo apt-get update

sudo apt-get install cuda

2.安装cudnn

https://developer.nvidia.com/rdp/cudnn-download安装cudnn,需要首先注册,随便填填就好,之后选择cuDNN v5.1 Library for Linux 下载后,cd至下载目录,在命令行中输入如下命令:

tar xvzf cudnn-7.5-linux-x64-v5.1-ga.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 

安装好后,在home 路径下的.profile中添加路径:

export CUDA_HOME=/usr/local/cuda

export DYLD_LIBRARY_PATH="$DYLD_LIBRARY_PATH:$CUDA_HOME/lib"

export PATH="$CUDA_HOME/bin:$PATH"

 

安装完成后,可以用编译cuda-sample中的案例来时一下:

$ cp -r /usr/local/cuda/samples ~/cuda-samples

$ pushd ~/cuda-samples

$ make

$ popd

$ ~/cuda-samples/bin/x86_64/darwin/release/deviceQuery

最终会显示GPU的配置等一些信息。编译过程比较缓慢。

最后,在安装tensorflow的根目录下,使用

  • 8
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
Inceptionv3是Google开发的图像分类神经网络模型。该模型采用了一种的计算方法——inception门,能够让模型更有效地利用神经网络的参数。在图像分类领域,Inceptionv3的应用广泛,因为它在分类准确率和计算速度之间实现了很好的平衡。 首先,我们需要准备好图像数据集。对于图像分类问题,数据集通常由几百或几千张图像组成,每张图像都有一个标签,表示它所属的类别。在训练Inceptionv3模型之前,我们需要将这些图像转换成神经网络能够理解的格式。对于Inceptionv3,输入数据格式是299x299像素的RGB图像。 接下来,我们可以使用Python编程语言和深度学习框架TensorFlow来实现Inceptionv3模型。首先,我们需要在TensorFlow中加载Inceptionv3的预训练权重,以便模型能够从中学习图像特征。然后,我们可以将模型的最后一层替换成的全连接层,该层的输出数量与数据集中的类别数量相同,并使用softmax函数进行概率归一化。最后,我们可以使用反向传播算法来训练模型。在训练过程中, 神经网络会不断调整权重,以便能够更好地分类图像。 一旦模型训练好之后,我们可以使用它来对的图像进行分类。通过将图像传递给模型并观察输出的概率,我们可以得知图像属于哪个类别。尽管Inceptionv3是一种强大的图像分类模型,但在实际应用中,人们通常根据具体问题的需求使用不同的深度学习模型
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值