重新訓練 Tensorflow 對象檢測 API

Question

我已閱讀有關如何使用 TensorFlow 對象檢測 API 訓練新類的教程。 但是我想要做的是在預訓練模型的已經訓練好的類中添加一個新類。

例如：MS-COCO 預訓練模型有 90 個類。 我想再添加一個類並檢測 91 個類的對象。

Answer 1

Tensorflow 對象檢測 API 支持從預訓練的 MS COCO 檢查點開始。 剛設置

fine_tune_checkpoint: "/usr/home/username/tmp/model.ckpt-#####"
from_detection_checkpoint: true

在您的檢測管道中。 （官方參考： https : //github.com/tensorflow/models/blob/master/research/object_detection/g3doc/configuring_jobs.md ）

您應該將帶有額外類的圖像添加到 coco 數據集並對所有 91 個類進行微調，否則您的網絡可能會忘記之前學到的內容而只檢測新對象。