0

我试图在预训练的 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.

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

4

0 回答 0