auto labeler

Automodiker

Auto Labeler ist ein automatisiertes Image-Annotation-Tool, das die Leistung der GPT-4-Vision-API von OpenAI nutzt, um Objekte in Bildern zu erkennen und Annotationen für Begrenzungsboxen bereitzustellen.

Merkmale

• Objekterkennung: Identifiziert automatisch Objekte in Bildern.
• Begrenzungsbox -Anmerkungen: Erzeugt Begrenzungsboxen um erkannte Objekte.
• Mehrfachformate: Unterstützt das Speichern von Anmerkungen sowohl in der Erkennung von Tensorflow -Objekten als auch in den JSON -Formaten.
• Benutzerdefinierte Beschriftungen: Ermöglicht das Angeben benutzerdefinierter Objektbezeichnungen zur Erkennung.

Erste Schritte

Voraussetzungen

• Python 3.6 oder höher

Installation

Klonen Sie das Repository in Ihre lokale Maschine:

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

Verwendung

1. Installieren Sie die erforderlichen Pakete:

 pip install -r requirements.txt

2. Stellen Sie Ihren OpenAI -API -Schlüssel ein:

 api_key = "YOUR_API_KEY"

3. Geben Sie den Pfad des Ordners mit den Bildern an, die Sie kennzeichnen möchten:

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

4. Geben Sie den Pfad des Ordners an, in dem Sie die Beschriftungen speichern möchten:

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

5. Geben Sie das Format an, in dem Sie die Etiketten speichern möchten:

 output_format = 'json' # or 'tf'

6. Definieren Sie die Liste der Etiketten, die Sie erkennen möchten:

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

7. Führen Sie das Hauptskript aus:

python main.py

Funktionen

• encode_image(image_path) : Codiert das Bild in das Basis64 -Format.
• detect_objects(image_path, labels) : Erkennt Objekte im Bild gemäß den Beschriftungen.
• save_tf_annotations(...) : Speichern Annotationen im TensorFlow -Objekterkennung XML -Format.
• save_json_annotations(...) : Speichert Annotationen im JSON -Format.
• process_images_in_folder(...) : verarbeitet mehrere Bilder in einem angegebenen Ordner.

Anerkennung

• Openai für die Bereitstellung der GPT-4 Vision API.
• Die Python -Community zur Aufrechterhaltung und Verbesserung der in diesem Projekt verwendeten Bibliotheken.

---

Stellen Sie sicher, dass Sie YOUR_API_KEY durch Ihren tatsächlichen OpenAI -API -Schlüssel ersetzen.