更改批量大小

Question

我正在尝试新发布的用于张量流的对象检测API。 我使用了教程中的示例训练程序，即在pet数据集上微调FRCNN-Resnet模型。仅使用一个GPU时，当我将批量大小更改为大于1的值（默认值为1）时，始终会显示错误消息。错误消息如下所示：

InvalidArgumentError（参见上面的追溯）：ConcatOp：输入的维度应匹配：shape [0] = [1,750,600,3] vs. shape [1] = [1,600,804,3]

Answer 1

您可能正在使用允许每张图片大小不同的keep_aspect_ratio_resizer，这意味着您只能使用批量大小1进行训练。要使用较大的批量大小进行训练，目前处理此问题的唯一方法是API将使用fixed_shape_resizer。有关此示例，请参阅一些SSD配置。