1

我正在尝试新发布的用于 tensorflow 的对象检测 API。我使用了教程中的示例训练程序,即在宠物数据集上微调FRCNN-Resnet模型。仅使用一个 GPU,当我将批量大小更改为大于 1 的值(默认值为 1)时,总是会显示一条错误消息。错误消息如下所示:

InvalidArgumentError(参见上文的回溯):ConcatOp:输入的尺寸应匹配:shape[0] = [1,750,600,3] 与 shape[1] = [1,600,804,3]

4

1 回答 1

1

您可能会使用keep_aspect_ratio_resizer允许每个图像具有不同大小的 ,这意味着您只能使用批量大小 1 进行训练。要使用更大的批量大小进行训练,当前在 API 中处理此问题的唯一方法是使用fixed_shape_resizer. 有关此示例,请参阅一些 SSD 配置。

于 2017-11-06T16:48:28.923 回答