YOLOv5 目標檢測模型

Question

我在訓練 yolov5 時遇到問題。 我可以訓練具有多個類的模型，例如 ['cat','dog','rabbit']。 我想為這個模型再添加一個類，['snake']，這樣我的模型可以一次檢測到所有四個。 我嘗試過模型集成和凍結層，但這些並沒有給我想要的結果。 請誰能幫我弄清楚如何做到這一點。 提前致謝

Answer 1

我想幫助解決您的問題，但信息太少。 出現此問題通常是由於以下原因：

1. 沒有為 ['snake'] 做標簽
2. ['snake'] 沒有添加到 yaml 文件的類中
3. 訓練數據集小或標注不正確。

可能還有其他原因，但是如果沒有運行detect.py的結果圖片等信息，很難提供詳細的幫助。