IT编程 > 脚本编程 > Python

3、一键训练模型

50人参与2020-07-07

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

您对本文有任何疑问!!点此进行留言回复

推荐阅读

猜你喜欢

3、一键训练模型

07-07

Python入门03——Python介绍、安装Python解释器、Python程序、变量

07-07

pandas画饼状图,折线图,柱状图

07-07

python百天练习day2

07-07

【剑指 Offer-python】 03. 数组中重复的数字

07-07

python数据分析练习——FoodFacts.csv分析

07-07

大家都在看

3、一键训练模型

07-07

Python3标准库:multiprocessing像线程一样管理进程

03-31

Python 3.7新功能之dataclass装饰器详解

07-23

mac系统安装Python3初体验

03-03

解决Python 命令行执行脚本时,提示导入的包找不到的问题

05-30

Django使用Celery异步任务队列的使用

06-18

C系程序员眼里的Python

05-05

Python3.4 splinter(模拟填写表单)使用方法

11-09

拓展阅读

热门评论