改进预训练的 tensorflow 对象检测 api 模型中的一类

问题描述

我试图改进预训练的 tf 对象检测 API 模型 (efficientdet_d0_coco17_tpu-32) 中的手机移动类。

我收集了训练数据并创建了 tfrecords，其中 label_map.pbtext 如下所示：

item {
 id: 77
 name: 'mobile'
}

并将 pipeline.config 更新如下：

  fine_tune_checkpoint: "checkpoint/ckpt-0" (of the pretrained model)
  fine_tune_checkpoint_version: V2
  fine_tune_checkpoint_type: "detection"

训练后，当我尝试推断新模型时，我总是得到 1 为 detection_classes。

知道我做错了什么吗？或者在这里无法继续学习？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）