博主好屯掖,我已經(jīng)成功訓(xùn)練了20萬次,把object_detection更換成你發(fā)的最新的文件换帜,然后運(yùn)行命令:python object_detection/export_inference_graph.py --input_type=image_tensor --pipeline_config_path=training/ssdlite_mobilenet_v2_coco.config --trained_checkpoint_prefix=training/model.ckpt-200000 --output_directory=ctree_inference_graph
成功生成了ctree_inference_graph文件夾阵赠,但是文件夾里面沒有任何文件涯塔,在cmd中的錯(cuò)誤提示為:
tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [256] rhs shape= [1280]
[[{{node save/Assign_26}}]]
tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [256] rhs shape= [1280]
[[node save/Assign_26 (defined at Y:\MobileNet-SSD\object_detection\exporter.py:241) ]]
InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [256] rhs shape= [1280]
[[node save/Assign_26 (defined at Y:\MobileNet-SSD\object_detection\exporter.py:241) ]]
tensorflow.python.framework.errors_impl.InvalidArgumentError: Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:
Assign requires shapes of both tensors to match. lhs shape= [256] rhs shape= [1280]
[[node save/Assign_26 (defined at Y:\MobileNet-SSD\object_detection\exporter.py:241) ]]
InvalidArgumentError (see above for traceback): Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:
Assign requires shapes of both tensors to match. lhs shape= [256] rhs shape= [1280]
[[node save/Assign_26 (defined at Y:\MobileNet-SSD\object_detection\exporter.py:241) ]]
看上去是結(jié)構(gòu)不一樣無法保存,請求博主幫助清蚀!
目標(biāo)檢測實(shí)踐_tensorflow版SSD模型測試致謝聲明 本文在學(xué)習(xí)《Tensorflow object detection API 搭建屬于自己的物體識(shí)別模型(2)——訓(xùn)練并使用自己的模型》的基礎(chǔ)上優(yōu)化并總結(jié)匕荸,此博客鏈接...