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

时间:2021-01-15 10:55:01

标签: python tensorflow object-detection-api tensorflow-model-garden resuming-training

我试图改进预训练的 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"

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

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

0 个答案:

没有答案