当前位置: 萬仟网 > IT编程>脚本编程>Python > 3、一键训练模型

3、一键训练模型

2020年07月07日  | 萬仟网IT编程  | 我要评论

0、前言

本文章所需文件下载地址:
链接:https://pan.baidu.com/s/18UP1HfdgX27hmyFB25a5FQ
提取码:1eci
在这里插入图片描述

本文章基于:2、训练模型 https://blog.csdn.net/lemon4869/article/details/107050702,请先实现这个。

1、create_directories.py 一键创建项目目录结构

操作 将下载的 create_directories.py 复制到 tf_train\workspaces 目录下,然后打开处于 tf_gpu 环境的控制台,切换到该目录。
输入命令 python create_directories.py --name=my_training_demo,就会创建相应的目录结构。
在这里插入图片描述
修改 train.py,在 main 函数最前面添加:

# GPU 按需分配
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
config.gpu_options.per_process_gpu_memory_fraction = 0.3
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))

one_command_train.py 一键训练模型

操作 进入 workspaces\my_training_demo\training 目录。
输入命令 python one_command_train.py --steps=100 --batch_size=12

整个文件的执行逻辑:

  1. 检查文件夹是否存在
  2. 检查 train 目录、eval 目录里是否有图片及标注文件
  3. 检查预训练模型 ssd_inception_v2_coco 是否存在,若不存在则下载模型
  4. 提取标签映射文件 label_map.pbtxt 里的信息
  5. 创建 tfrecord 文件
  6. 从命令行读取训练步数(steps)和批样本个数(batch_size)
  7. 修改 ssd_inception_v2_coco.config 文件
  8. 启动训练

训练完成的截图:在这里插入图片描述
导出模型、用模型做目标检测请看前一章:2、训练模型 https://blog.csdn.net/lemon4869/article/details/107050702

本文地址:https://blog.csdn.net/lemon4869/article/details/107145398

如对本文有疑问, 点击进行留言回复!!

相关文章:

验证码:
Copyright © 2017-2020  萬仟网 保留所有权利. 粤ICP备17035492号-1
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com