Hilary_0511的博客

原创 Windows下运用Tensorflow object detection API训练出现的问题--爬坑

刚开始学习Tensorflow object detection API遇到了不少问题，下面就遇到的问题做下记录。 我是在CPU下训练的模型，然后遇到训练到一百多步的时候会报这个错误，如图。用的训练模型为ssd_mobilenet_v1_coco。 然后就各种查资料，最后把训练模型中的 batch_size: 1 改为 batch_size: 5 成功的把困扰好几天的问题解决了。  ...