Image Feature Extraction Using GenAI

Extracción-de-características-de-imagen-usando-GenAI

Este proyecto implementa un canal de IA generativa avanzada para extraer y calificar características de imágenes. Combina el poder de Florence-2, un modelo de lenguaje de visión de última generación, con una versión mejorada de Mistral-v3, un modelo de lenguaje de gran tamaño de vanguardia.

Características clave

• Utiliza Florence-2 para generar descripciones de imágenes detalladas
• Emplea un modelo Mistral-v3 personalizado y ajustado para extracción y puntuación de características.
• Genera resultados en formato JSON para una fácil integración con otras aplicaciones.

Generación de descripción de imagen

• Modelo: Florencia-2
• Entrada: datos de imagen sin procesar
• Salida: descripción textual completa de la imagen.

Extracción y calificación de características

• Modelo: Mistral-v3 afinado
• Entrada: descripción del texto de Florence-2
• Salida: objeto JSON que contiene características extraídas y sus puntuaciones

Uso

Para ejecutar este proyecto, siga estos pasos:

1. Descargar el conjunto de datos
2. Ejecute el script de extracción de imágenes:

 python scrapeImages.py

Esto extraerá imágenes de Airbnb según el conjunto de datos.

3. Abra y ejecute el cuaderno Jupyter GenAI_Approach.ipynb . Siga las instrucciones del cuaderno para procesar las imágenes y extraer características.

Nota importante: asegúrese de utilizar los mismos nombres de archivos y directorios que se especifican en los scripts, o modifique las rutas en el código para que coincidan con la estructura de su directorio.