0

我正在尝试将 TensorFlow 2 对象检测 API 与用于多类的自定义数据集一起使用来训练 SSD,我以文档提供的示例为基础:https ://github.com/tensorflow/models/blob/master/ research/object_detection/colab_tutorials/eager_few_shot_od_training_tf2_colab.ipynb 我目前的问题是当我开始微调时:

InvalidArgumentError: paddings的第一维必须是inputs[2,2] [6] [Op:Pad]的等级

这似乎与 train_step_fn 上的 model.provide_groundtruth 部分有关,正如我提到的,我从 TensorFlow 记录中获取数据,我将其映射到数据集并使用 padded_batches(tf.data.TFRecordDataset) 将其分成批次似乎这用图像进行训练是正确的,但现在我的问题是groundtruth,因为现在这也转换为批次[batch_size,num_detections,coordinate_bbox],这是问题吗?关于如何解决这个问题的任何想法。谢谢

PS 我尝试使用修改后的 pipeline.config 文件的版本并像过去使用 TensorFlow 1 一样运行 model_main_tf2.py,但这种方法有问题。

4

1 回答 1

0

只是为了与大家分享这解决了我的问题是我设法将数据正确地分成图像和地面实况的批次,但我从未将我的标签转换为一种热矢量编码。

于 2020-07-25T01:58:58.810 回答