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(...) ：指定されたフォルダー内の複数の画像を処理します。

謝辞

• GPT-4 Vision APIを提供するためのOpenAI。
• このプロジェクトで使用されているライブラリを維持および改善するためのPythonコミュニティ。

---

YOUR_API_KEY実際のOpenAI APIキーに置き換えてください。