auto labeler

Auto Labeler

Auto Labeler es una herramienta automatizada de anotación de imágenes que aprovecha el poder de la API de visión GPT-4 de OpenAI para detectar objetos en las imágenes y proporcionar anotaciones de caja limitantes.

Características

• Detección de objetos: identifica automáticamente objetos en las imágenes.
• Anotaciones de cuadro delimitador: genera cajas delimitadoras alrededor de los objetos detectados.
• Formatos múltiples: admite anotaciones de ahorro tanto en la detección de objetos TensorFlow como en los formatos JSON.
• Etiquetas personalizadas: permite especificar etiquetas de objetos personalizados para la detección.

Empezando

Requisitos previos

• Python 3.6 o superior

Instalación

Clonar el repositorio de su máquina local:

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

Uso

1. Instalar paquetes requeridos:

 pip install -r requirements.txt

2. Establezca su tecla API OpenAI:

 api_key = "YOUR_API_KEY"

3. Especifique la ruta de la carpeta con las imágenes que desea etiquetar:

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

4. Especifique la ruta de la carpeta donde desea guardar las etiquetas:

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

5. Especifique el formato en el que desea guardar las etiquetas:

 output_format = 'json' # or 'tf'

6. Defina la lista de etiquetas que desea detectar:

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

7. Ejecute el guión principal:

python main.py

Funciones

• encode_image(image_path) : codifica la imagen al formato base64.
• detect_objects(image_path, labels) : detecta objetos en la imagen como se especifica por etiquetas.
• save_tf_annotations(...) : guarda anotaciones en formato XML de detección de objetos TensorFlow.
• save_json_annotations(...) : guarda anotaciones en formato JSON.
• process_images_in_folder(...) : procesa múltiples imágenes en una carpeta especificada.

Expresiones de gratitud

• OpenAi para proporcionar la API de visión GPT-4.
• La comunidad de Python para mantener y mejorar las bibliotecas utilizadas en este proyecto.

---

Asegúrese de reemplazar YOUR_API_KEY con su tecla API OpenAI real.