无法训练张量对象检测API模型

Question

我正在使用tensorflow的对象检测API。我成功地使用了＆ssd_mobilenet_v1_coco_2017_11_17＆＃39;模型，后来我从给定的模型转移到另一个模型，但在训练过程开始时显示错误：     ＆＃34; TypeError：预期int32，得到范围＆lt; 0,3＆gt;类型＆＃39;范围＆＃39;代替＆＃34 ;. 除了ssd_mobilenet_v1_coco_2017_11_17之外的所有其他模型都抛出此错误。

我使用300 * 300尺寸的图像进行所有模型的训练。 这里我附上了显示错误信息的命令提示符窗口的图像。我使用tensorflow版本是1.5和python 3.6。

Answer 1

请修改 第154行：tf.constant（range（num_boundaries），dtype = tf.int32）， to tf.constant（list（range（num_boundaries）），dtype = tf.int32）

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