无法导出 TensorFlow Object 检测 API

Question

因此，我按照说明运行脚本，在训练完成后导出我的 model。 现在，我不知道为什么，但我不断收到此错误：

 > ValueError: Tensor's shape (3, 3, 256, 12) is not compatible with > supplied shape (3, 3, 256, 546)

可能是什么问题？

Answer 1

I had the same issue trying to export a model with the Object Detection API, in my case I was using the model "ssd_resnet50_v1_fpn_640x640_coco17_tpu-8" from Model Zoo.

基本上，您必须：

1. 保存微调后的 model 的检查点。

2. 修改您重新训练的 model 的“pipeline.config”文件，以匹配您自定义数据集的类数； 即，正确更改“num_classes”的值（如果未正确设置类数，则图形的形状将不匹配）

3. 使用exporter_main_v2帮助程序 function 导出 model 以推断您在步骤 1 和 2 中生成的文件。

我猜你做了第 1 步和第 3 步，但没有第 2 步。

我在此笔记本中提供了此解决方案的详细说明，其中对自定义数据集执行了少量镜头微调，并导出、加载和测试了生成的 model。