電腦視覺中的影像標註工具總結

2021 年 9 月 19 日
筆記

本文來自公眾號CV技術指南資源分享系列

創建高品質的數據集是任何機器學習項目的關鍵部分。在實踐中，這通常比實際訓練和超參數優化花費的時間更長。因此，選擇合適的標註工具至關重要。在這裡，我們總結了一些用於電腦視覺任務的最佳影像標註工具：labelme、labelImg、CVAT和hasty.ai。

作者：Dmitrii

編譯：CV技術指南

歡迎關注公眾號CV技術指南，專註於電腦視覺的技術總結、最新技術跟蹤、經典論文解讀。

labelme

地址：//github.com/wkentaro/labelme

你可以用它做什麼

labelme 是一個基於 python 的開源影像多邊形標註工具，可用於手動標註影像以進行對象檢測、分割和分類。它是在線 LabelMe 的離線分支，最近關閉了新用戶註冊選項。所以，在這篇文章中，我們只考慮 labelme（小寫）。

該工具是具有直觀用戶介面的輕量級圖形應用程式。使用 labelme，您可以創建：多邊形、矩形、圓、線、點或線帶。

通常，能夠以眾所周知的格式（例如 COCO、YOLO 或 PASCAL VOL）導出注釋以供後續使用通常很方便。但是，在 labelme 中，標籤只能直接從應用程式保存為 JSON 文件。如果要使用其他格式，可以使用 labelme 存儲庫中的 Python 腳本將注釋轉換為 PASCAL VOL。

儘管如此，它還是一個相當可靠的應用程式，具有用於手動影像標記和廣泛的電腦視覺任務的簡單功能。

安裝和配置

labelme 是一個跨平台的應用程式，可以在多個系統上工作，例如 Windows、Ubuntu 或 macOS。安裝本身非常簡單，這裡有很好的描述。例如，在 macOS 上，您需要在終端中運行以下命令：

安裝依賴：brew install pyqt
安裝labelme：pip install labelme
運行 labelme：labelme

labelImg

地址：//github.com/tzutalin/labelImg

你可以用它做什麼

labelImg 是一種廣泛使用的開源圖形注釋工具。它僅適用於目標定位或檢測任務，並且只能在考慮的對象周圍創建矩形框。

儘管存在這種限制，我們還是建議使用此工具，因為該應用程式僅專註於創建儘可能簡化工具的邊界框。對於此任務，labelImg 具有所有必要的功能和方便的鍵盤快捷鍵。

另一個優點是您可以以 3 種流行的注釋格式保存/載入注釋：PASCAL VOC、YOLO 和 CreateML。

安裝和配置

這裡對安裝進行了很好的描述。還要注意 labelImg 是一個跨平台的應用程式。例如，對於 MacOS，需要在命令行上執行以下操作：

安裝依賴：先 brew install qt，然後 brew install libxml2
選擇要安裝的文件夾的位置。
當你在文件夾中時，運行以下命令：git clone //github.com/tzutalin/labelImg.git, cd labelImg 然後 make qt5py3
運行 labelImg：python3 labelImg.py
開發人員強烈建議使用 Python 3 或更高版本和 PyQt5。

CVAT

地址：//github.com/openvinotoolkit/cvat

你可以用它做什麼

CVAT 是一種用於影像和影片的開源注釋工具，用於對象檢測、分割和分類等任務。

要使用此工具，您無需在電腦上安裝該應用程式。可以在線使用此工具的網路版本。您可以作為一個團隊協作處理標記影像並在用戶之間分配工作。

還有一個很好的選擇，它允許您使用預先訓練的模型來自動標記您的數據，如果您使用 CVAT 儀錶板中現有的可用模型，這可以簡化最流行的類（例如，COCO 中包含的類）的過程。或者，您也可以使用自己的預訓練模型。

CVAT 具有我們已經考慮過的工具中最廣泛的功能集。特別是，它允許您以大約 15 種不同的格式保存標籤。可以在此處找到完整的格式列表。

hasty.ai

地址：//hasty.ai/

你可以用它做什麼

與上述所有工具不同，hasty.ai 不是免費的開源服務，但由於所謂的對象檢測和分割的 AI 助手，它非常方便地標記數據。自動支援允許您顯著加快注釋過程，因為在標記期間輔助模型正在訓練。換句話說，標記的影像越多，助手的工作就越準確。我們將在下面看一個例子來說明它是如何工作的。您也可以免費試用此服務。該試驗提供 3000 積分，足以為一個物體檢測任務自動生成大約 3000 個物體的建議標籤。hasty.ai 允許您以 COCO 或 Pascal VOC 格式導出數據。您還可以作為一個團隊處理單個項目並在項目設置中分配角色。免費積分用完後，hasty.ai 仍然可以免費使用，但標記將完全由手動操作。在這種情況下，最好考慮上述免費工具。

配置