tensorflow 与训练猫狗数据模型做迁移式学习完整流程
一、首先安装anaconda——创建虚拟环境(本人命名为tf1X)——下载安装tensorflow,numpy,pandas等包的准备。一点注意python和tensorflow和其他包的版本匹配,自行去管我查询。具体操作可以参考本人之前的博客。链接赋上:...
经过本人一段学习,终于完成了第一个训练集。记录一下,也为大家一个参考。相关数据集打包到百度网盘评论区留言可取。
推荐一本书

一、首先安装anaconda——创建虚拟环境(本人命名为tf1X)——下载安装tensorflow,numpy,pandas等包的准备。
一点注意python和tensorflow和其他包的版本匹配,自行去管我查询。
具体操作可以参考本人之前的博客。链接赋上:
Anaconda Prompt中装tensorflow_gpu_m0_63172128的博客-CSDN博客
二、开始迁移式学习
测试集图片如下:

首先这是我打标签的过程之一。

用python一键训练模型:

将提供的create_directories.py文件复制到workspaces文件夹下。到cmd输入:
conda activate tf1X
python create_directories.py -n my_training_demo
一键训练:

将one_command_train.py文件复制到my_training_dome,再cmd输入:(本人训练500次)
python one_command_train.py --steps 500 --batch_size 12


训练好之后会在training文件夹生成这些节点文件

最后,导出冻结图文件。完成整个训练过程。
tensorboard --logdir=evaluation\

冻结tensorflow模型
将tf_train\models\research\object_detection下的export_inference_graph.py

复制到自己的E:\tf_train\workspaces\my_training_dome下,启动cmd
python export_inference_graph.py --input_type image_tensor --pipeline_config_path training\ssd_inception_v2_coco.config --trained_checkpoint_prefix training\model.ckpt-500 --output_directory trained_frozen_models\my_training_model
将object_detection_example_1.py 复制到workspaces\my_training_dome下改名:为 object_detection_example_2.py
用编辑器打开object_detection_example_2.py修改代码:

运行看训练结果:
python object_detection_example_2.py

最终的结果还是准确率可以的

以后操作时:只需要两行代码

更多推荐

所有评论(0)