刚开始学习Tensorflow object detection API遇到了不少问题,下面就遇到的问题做下记录。

我是在CPU下训练的模型,然后遇到训练到一百多步的时候会报这个错误,如图。用的训练模型为ssd_mobilenet_v1_coco。

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

然后就各种查资料,最后把训练模型中的 batch_size: 1 改为 batch_size: 5 成功的把困扰好几天的问题解决了。

 

相关文章:

  • 2021-04-04
  • 2021-06-13
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-02-13
猜你喜欢
  • 2021-06-23
  • 2021-11-24
  • 2022-01-29
  • 2021-06-25
  • 2021-12-20
  • 2021-11-03
  • 2021-11-14
相关资源
相似解决方案