pip的安装tensorflow【目标检测模块】2-实现目标检测

本文详细介绍如何使用model_main.py脚本进行模型训练,通过设置--model_dir和--pipeline_config_path参数来指定训练模型的路径和配置文件。训练过程中,可通过tensorboard监控训练状态。最后,利用export_inference_graph.py脚本导出训练好的模型,包括.pb和graph.pbtxt文件。

训练模型:

利用脚本文件model_main.py,该脚本文件位于文件夹/research/object_detection下,其中参数--model_dir指明训练出的模型的路径,--pipeline_config_path指定配置文件,该配置文件几乎涵盖了所有信息,包括训练数据、验证数据等 

训练过程中,可以利用tensorboard命令,查看训练过程:

其中参数 --logdir 用于指明模型的路径,即参数--model_dir指明的路径

导出模型:

利用脚本文件export_inference_graph.py文件,导出模型,包括.pb文件 graph.pbtxt。一共四个参数分别指:

--input_type 

--pipeline_config_path 

--trained_chekpoint_prefix

--output_directory

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值