0

我正在使用 tensorflow 的对象检测 API。我成功地使用 'ssd_mobilenet_v1_coco_2017_11_17' 模型进行了训练，后来我从给定模型转移到另一个模型，但是在训练过程开始时它显示错误：“TypeError：预期 int32，得到范围 <0,3> 类型为 'range' 而不是”。除 ssd_mobilenet_v1_coco_2017_11_17 之外的所有其他模型都会引发此错误。

我使用 300*300 大小的图像来训练所有模型。在这里，我附上了显示错误消息的命令提示符窗口的图像。我使用的 tensorflow 版本是 1.5 和 python 3.6。 在此处输入图像描述 在此处输入图像描述

4

1 回答 1

1

请修改第 154 行： tf.constant(range(num_boundaries), dtype=tf.int32), 为 tf.constant(list(range(num_boundaries)), dtype=tf.int32)

如需进一步参考： https ://github.com/tensorflow/models/issues/3443

于 2018-03-23T06:18:36.907 回答