错误训练在增强数据集上更快地进行RCNN v2初始化,但不使用SSD Mobilenet v1 coco

时间:2018-12-11 15:02:49

标签: python tensorflow conv-neural-network object-detection

我扩充了我的数据集,并使用Tensorflow对象检测API https://github.com/tensorflow/models/tree/master/research/object_detection在SSD Mobilenet v1 coco上对其进行了训练。

我最初遇到此错误:

  

ResourceExhaustedError(请参阅上面的回溯):分配时为OOM   形状为[] 24,256,38,38]的张量,类型为float on   / job:localhost / replica:0 / task:0 / device:GPU:0通过分配器GPU_0_bfc

然后我将批次大小从24更改为3,并开始训练。

然后我开始在同一增强数据集上使用Faster RCNN v2 Inception进行训练,并得到以下错误:

  

InvalideArgumentError(请参阅上面的回溯):LossTensor为inf或   南:张量具有NaN值

然后我将批量大小从1更改为32,并得到此错误:

  

InvalidArgumentError(请参阅上面的回溯):断言失败:   [最大框坐标值大于1.100000:]

该如何解决?

0 个答案:

没有答案