創建自定義檢測 model - 圖像 Object 檢測

Question

我對圖像/視頻中的 object 檢測有一些疑問。

我成功嘗試了本教程： 使用 6 行代碼訓練 Object 檢測 AI

在本教程中，我們訓練 model 來檢測全息透鏡。 如果我想檢測汽車，我需要訓練一個新的 model。

我的問題是：我可以使用我的實際 model（全息透鏡）並讓它檢測全息透鏡和汽車嗎？

最后，我想要一個文件（模型）來檢測我用來訓練的所有對象。 我將放置一個包含“n”個圖像的目錄，model 將嘗試檢測它經過訓練以檢測的所有內容並返回檢測到的對象的名稱。

對不起我的英語不好！！

Answer 1

是的，您可以在任意數量的 object 課程上訓練您的 model。 本教程基於YOLOV3 ，它是最先進的 object 檢測模型之一，速度足夠快，可用於實時和視頻。 在您提到的頁面上，在第 1 步中，它解釋了如何准備和注釋您的數據集。 您的數據可以包含許多類別的對象（如汽車、hololens 等），但您需要在訓練之前設置此數字。 此外，在注釋步驟（將 label 分配給對象）中，您應該為對象分配正確的名稱。

此頁面上有一個鏈接解釋了如何准備您自己的數據： Object 檢測培訓 — 准備您的自定義數據集

如果數據已經准備好，請確保它是 Pascal Voc 格式，因為它必須與您使用的 model 保持一致。 最后，您只需要一個 model 即可檢測各種物體。 精度取決於迭代次數、每個 class 中的對象數量、對象的形狀、大小、顏色、角度等的變化等。