Tensorflow對象檢測API未經訓練的Faster-RCNN模型

Question

我目前正在嘗試使用Tensorflow對象檢測API和python構建對象檢測器。 我已按照此處和此處發布的說明設法重新訓練了faster-rcnn model

但是，據我了解，我的訓練時間相當長。 我知道我使用轉移學習，而不是從頭開始訓練一個更快的rcnn模型。 我想知道是否有某種方法可以下載未經訓練的fast-rcnn模型並從頭開始（端對端）對其進行訓練，而不必求助於轉移學習。

我熟悉遷移學習的優點，但是，我的對象檢測器旨在快速培訓，范圍狹窄，並接受相對於對象的字母訓練，因此我認為遷移學習不是最佳途徑。

我相信解決這個問題將與pipeline.config文件有關，特別是在這一部分：

fine_tune_checkpoint: "PATH/TO/PRETRAINED/model.ckpt"
from_detection_checkpoint: true
num_steps: 200000

但是我不確定如何指定沒有fine_tune_checkpoint

Answer 1

要從頭開始訓練自己的模型，請執行以下操作：

1. 注釋掉以下幾行

    # fine_tune_checkpoint: <YOUR PATH>
    # from_detection_checkpoint: true

2. 如果您遵循本教程，請刪除下載的預訓練模型或重命名其路徑。

您不必下載“空”模型。 相反，您可以在配置文件中指定您自己的權重初始化，例如，如下所示： 如何在Tensorflow Object Detection API中為卷積層初始化權重？