Image Feature Extraction Using GenAI

使用 GenAI 进行图像特征提取

该项目实现了先进的生成人工智能管道，用于从图像中提取特征并对其进行评级。它结合了 Florence-2（一种最先进的视觉语言模型）的强大功能与 Mistral-v3（一种尖端大型语言模型）的微调版本。

主要特点

• 利用 Florence-2 生成详细的图像描述
• 采用自定义微调的 Mistral-v3 模型进行特征提取和评分
• 以 JSON 格式输出结果，以便于与其他应用程序集成

图像描述生成

• 型号：Florence-2
• 输入：原始图像数据
• 输出：图像的综合文本描述

特征提取和评级

• 型号：微调 Mistral-v3
• 输入：来自 Florence-2 的文本描述
• 输出：包含提取的特征及其分数的 JSON 对象

用法

要运行该项目，请按照下列步骤操作：

1. 下载数据集
2. 运行图像抓取脚本：

 python scrapeImages.py

这将根据数据集从 Airbnb 抓取图像。

3. 打开并运行GenAI_Approach.ipynb Jupyter 笔记本。按照笔记本中的说明处理图像并提取特征。

重要提示：确保使用脚本中指定的相同文件和目录名称，或修改代码中的路径以匹配您的目录结构。