【问题标题】：Can't export TensorFlow Object Detection API无法导出 TensorFlow 对象检测 API
【发布时间】：2021-03-26 18:50:52
【问题描述】：
所以，我按照instructions 并在训练完成后运行脚本导出我的模型。现在，我不知道为什么，但我不断收到此错误：
> ValueError: Tensor's shape (3, 3, 256, 12) is not compatible with
> supplied shape (3, 3, 256, 546)
可能是什么问题？
【问题讨论】：
-
您可以尝试按照提到的步骤here 告诉我们更新。谢谢！
标签：
python
tensorflow
image-processing
object-detection
【解决方案1】：
我在尝试使用 Object Detection API 导出模型时遇到了同样的问题，在我的例子中，我使用的是 Model Zoo 中的模型“ssd_resnet50_v1_fpn_640x640_coco17_tpu-8”。
基本上，您必须：
-
保存微调模型的检查点。
-
修改您重新训练的模型的“pipeline.config”文件，以匹配您的自定义数据集的类数；即正确改变“num_classes”的值（如果类的数量没有正确设置，那么图形的形状将不匹配）
-
使用 exporter_main_v2 辅助函数导出模型，以使用您在步骤 1 和 2 中生成的文件进行推理。
我猜你做了第 1 步和第 3 步，但没有第 2 步。
我在this notebook 中提供了此解决方案的详细说明，其中对自定义数据集执行少量镜头微调，并导出、加载和测试生成的模型。