当前位置:
首页 > Python基础教程 >
-
计算机视觉(3):用inception-v3模型重新训练自己的数据模型
背景:
现代的图像识别模型具有数以百万计的参数,从头开始训练(Train from scratch)需要大量的样本数据以及消耗巨大的计算资源(几百个GPU),因此采用迁移学习的方式重训一个模型(Retrain a model)对于学习模型的成本较低,利用Inception-V3作为已经训练好的模型,来实现自己的图像分类识别。
inception模型文件目录介绍:
data目录:需要训练的数据放在该目录: inception_model:把下载inception模型放在该目录下:
test_images:把训练完成之后,测试图片所放目录
数据准备完成之后,按以下步骤执行:
步骤1:
windows:
运行批处理文件retrain.bat
python retrain.py ^ --bottleneck_dir bottleneck ^ #自行生成每个训练图片的数据 --how_many_training_steps 200 ^ #步数 --model_dir inception_model ^ #模型 --output_graph output_graph.pb ^ #训练后输出模型 --output_labels output_labels.txt ^ #训练后输出label --image_dir data pause
Ubuntu:
运行批处理文件retrain.sh
步骤2:
训练完成后,测试如下
Python predict.py
测试结果如下:
结论:如果测试物体不相似,识别率很高,但是如果物体形状相似,识别率则不高
源码:https://pan.baidu.com/s/1qdRmnQsRv5k3QesZxRC9QA 提取码:jipt
栏目列表
最新更新
nodejs爬虫
Python正则表达式完全指南
爬取豆瓣Top250图书数据
shp 地图文件批量添加字段
爬虫小试牛刀(爬取学校通知公告)
【python基础】函数-初识函数
【python基础】函数-返回值
HTTP请求:requests模块基础使用必知必会
Python初学者友好丨详解参数传递类型
如何有效管理爬虫流量?
SQL SERVER中递归
2个场景实例讲解GaussDB(DWS)基表统计信息估
常用的 SQL Server 关键字及其含义
动手分析SQL Server中的事务中使用的锁
openGauss内核分析:SQL by pass & 经典执行
一招教你如何高效批量导入与更新数据
天天写SQL,这些神奇的特性你知道吗?
openGauss内核分析:执行计划生成
[IM002]Navicat ODBC驱动器管理器 未发现数据
初入Sql Server 之 存储过程的简单使用
这是目前我见过最好的跨域解决方案!
减少回流与重绘
减少回流与重绘
如何使用KrpanoToolJS在浏览器切图
performance.now() 与 Date.now() 对比
一款纯 JS 实现的轻量化图片编辑器
关于开发 VS Code 插件遇到的 workbench.scm.
前端设计模式——观察者模式
前端设计模式——中介者模式
创建型-原型模式