使用tensorflow的retrain.py训练图片分类器
2024-09-05 12:29:33
参考
https://hackernoon.com/creating-insanely-fast-image-classifiers-with-mobilenet-in-tensorflow-f030ce0a2991
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py
https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0
https://www.tensorflow.org/tutorials/image_retraining
准备图片
images/label1/1.jpg
/2.jpg
images/label2/1.jpg
/2.jpg
开始训练，默认Inception v3 model
python retrain.py \
--bottleneck_dir=bottlenecks \
--how_many_training_steps=500 \
--model_dir=inception \
--summaries_dir=training_summaries/basic \
--output_graph=retrained_graph.pb \
--output_labels=retrained_labels.txt \
--image_dir=flower_photos
也可以使用其他模型
详细参数查看
python retrain.py -h
使用训练结果进行预测
参考https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#5
这样在训练好的inception-2015-12-05数据基础上
仅重新训练最后一层
用较短的时间达到分类效果
最新文章
- 廖雪峰js教程笔记13 插入DOM
- IOS开发技巧快速生成二维码
- Python使用QRCode模块生成二维码
- ActiveMQ(七)_伪集群和主从高可用使用
- Android progressbar 详解
- Redis监控方案
- 关于SWT/JFace的事件模型的四种方式
- Pyocr 0.2 发布，Python 的 OCR 库 - 开源中国社区
- 文件描述符和exec() close_on_exec
- Jquery Validate 动态添加校验
- linux文件和目录权限
- 实现基于Keepalived主从高可用集群网站架构
- ES6学习路上的小学生，promise处理异步操作，简易原始起步之用。先能用，再深究！
- 使用Ajax和JSON实现注册时候验证用户名是否存在的功能
- 转：【专题六】UDP编程
- JQ面试问题（转载）
- 20145328《网络对抗技术》Final
- ios runtime简单实用（添加动态属性）
- ZOJ2588：Burning Bridges（无向连通图求割边）
- Javascript变长参数和默认参数