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 筆記本。按照筆記本中的說明處理圖像並提取特徵。

重要提示：確保使用腳本中指定的相同檔案和目錄名稱，或修改程式碼中的路徑以符合您的目錄結構。