張量流對象檢測訓練模型不起作用

Question

我使用ssd和更快的r-cnn模型訓練了用於張量流對象檢測的數據集。我的數據集中有220個訓練圖像和30個測試圖像。 我對模型進行了20萬步的訓練，結果損失在1以下。但是當我在視頻上測試訓練好的模型時，它幾乎可以檢測並標記視頻中的所有內容。 誰能告訴我為什么會這樣嗎？ 謝謝

Answer 1

您正在使用的類的數量僅為一種，並且您使用屬於同一類的圖像訓練了模型並對其進行了測試。

因此，問題在於模型是偏斜的（對所有圖像預測相同），無論在哪個圖像上進行測試，您都將獲得相同的輸出。

解：

用幾乎相等數量的負像訓練您的模型。

例如：包含要識別的對象的220張圖像（將其標記為1）和不包含對象的另外近220張圖像（將其標記為0）

使用F1分數來檢查您的准確性，因為它可以幫助您了解數據集是否傾斜。

選中此選項可了解不同類型的准確性度量。 參加本課程以了解有關CNN的更多信息。