发新帖

【机器学习初步】基于tensorflow demo进行retrain

[复制链接]
271 1

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
如果你学习了前一篇【机器学习初步】用android studio运行tensorflow demo,那么你就应该已经得到一个
完整的可以运行的demo,下面我们看看如何训练我们自己的模型。


1. 首先准备好自己的图片资料,放在data/pics/下面
2. 开始用这些数据进行训练
python tensorflow/examples/image_retraining/retrain.py \  --bottleneck_dir=/data/pics \  --how_many_training_steps 4000 \  --model_dir=/data/incept \  --output_graph=/data/graph.pb \  --output_labels=/data/labels.txt \  --image_dir /data/dts

3. 得到graph.pb和txt后,替换demo里的对应文件
4. 重新编译app并运行demo观察实验结果
本楼点评(0) 收起

精彩评论1

neverchange  TF豆豆  发表于 2018-7-4 12:39:57 | 显示全部楼层
如果遇到某个op不支持,可能需要优化,因为tf lite并不是完整的op集。
python tensorflow/python/tools/optimize_for_inference.py \
  --input=/data/graph.pb \
  --output=/data/graph2.pb \
  --input_names="Mul" \
  --output_names="final_result"
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表