首页 > 解决方案 > Object_Detection 在张量流中不起作用

问题描述

我是张量流的新手,我训练了面罩模型并在网络摄像头上进行了尝试,但它显示了这样的内容。

在此处输入图像描述 任何人都可以帮我解决这个问题,我使用了大约 100 多张图像和 2-3 小时的训练,我做错了任何人都可以帮助我。

使用的最后 2 个 cmd

python train.py --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config --logtostderr


python export_inference_graph.py --input_type image_tensor --pipeline_config_path training/ssd_mobilenet_v1_pets.config --trained_checkpoint_prefix training/model.ckpt-0 --output_directory new_graph

标签: pythonpython-3.xtensorflowmodelpycharm

解决方案


您使用的training/model.ckpt-0是模型开始训练后的第一个检查点,您需要使用保存在training文件夹中的最后一个检查点。


推荐阅读