利用inception v3进行物体分类

前言

本文将利用tensorflow官方提供的inception v3模型(基于ImageNet训练)进行迁移学习。目前了解到的模型有两套(推测应该有许多版本),本文采用的这套结构稍显复杂,retrain.py多达1000多行,但是使用起来效率更高,提取完特征之后训练200个周期仅需数秒。

1. 训练和预测流程

1.1 准备初始工作目录

在这里插入图片描述
在这里插入图片描述

  • 如图所示,我们需要的核心文件是retrain.pydata中存放需要训练的数据集,本文将训练5个分类,每个类别500张.jpg图片;.bat.sh文件分别是windows和linux环境下运行retrain.py的脚本文件;最后,predict.py是训练好模型后预测图片分类的程序。文件下载:

链接: https://pan.baidu.com/s/1NfcV1DZkoybtl0L5LujIcg
提取码: gdkx
本地:E:\软件打包\模型和数据集\01.inception v3 迁移学习

<

评论 17
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值