有沒有辦法限制現有的TensorFlow對象檢測模型？

Question

在GitHub的tensorflow/models回購上，他們提供了五個預先訓練好的模型用於物體檢測。

這些模型是在COCO數據集上訓練的，可以識別90個不同的對象。

我需要一個模型來檢測人，而不是別的。我可以將代碼修改爲僅在人物上打印標籤，但它仍然會查找其他89個對象，這比查找一個對象需要更多的時間。

我可以訓練自己的模型，但我寧願能夠使用預先訓練好的模型，而不是花很多時間來訓練我自己的模型。

那麼有沒有辦法，通過修改模型文件或TensorFlow或Object Detection API代碼，所以它只能查找單個對象？

Answer 1

您可以通過步行數小於幾千步調整您的模型（需要訓練的小數據集就足夠了），或者在標籤定義文件（.pbtx文件）中查找您的模型，搜索人員標籤，並與其他人一起做任何你想要的東西。