训练模型:

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

被折叠的 条评论
为什么被折叠?



