auto labeler

自動標籤

Auto Labeler是一種自動圖像註釋工具，它利用OpenAI的GPT-4 Vision API的功能來檢測圖像中的對象並提供邊界框註釋。

特徵

• 對象檢測：自動標識圖像中的對象。
• 邊界框註釋：生成周圍對象周圍的邊界框。
• 多種格式：支持在TensorFlow對象檢測和JSON格式中保存註釋。
• 自定義標籤：允許指定自定義對象標籤以進行檢測。

入門

先決條件

• Python 3.6或更高

安裝

將存儲庫克隆到您的本地計算機：

git clone https://github.com/Flode-Labs/auto-labeler.git

用法

1. 安裝所需的軟件包：

 pip install -r requirements.txt

2. 設置您的OpenAI API密鑰：

 api_key = "YOUR_API_KEY"

3. 用要標記的圖像指定文件夾的路徑：

 input_folder = '/path/to/input/folder'

4. 指定要保存標籤的文件夾的路徑：

 output_folder = '/path/to/output/folder'

5. 指定要保存標籤的格式：

 output_format = 'json' # or 'tf'

6. 定義要檢測到的標籤列表：

 labels = [ 'cat' , 'dog' , 'bird' ]

7. 運行主腳本：

python main.py

功能

• encode_image(image_path) ：將圖像編碼為base64格式。
• detect_objects(image_path, labels) ：檢測標籤指定的圖像中的對象。
• save_tf_annotations(...) ：將註釋保存在Tensorflow對象檢測XML格式中。
• save_json_annotations(...) ：以JSON格式保存註釋。
• process_images_in_folder(...) ：指定文件夾中的多個圖像。

致謝

• Openai提供GPT-4 Vision API。
• Python社區，用於維護和改善該項目中使用的庫。

---

確保您用實際的OpenAI API鍵替換YOUR_API_KEY 。