在上一篇文章中,我們詳細介紹了如何使用LLaMA-Factory框架對Qwen2-VL大模型進行微調,以實現人臉情感識別的功能。本篇文章將聚焦于微調完成后,如何調用這個模型進行人臉情感識別的具體代碼實現,包括詳細的步驟和注釋。
模型調用步驟
- 環境準備:確保安裝了必要的Python庫。
- 加載微調后的模型:使用Hugging Face的
transformers
庫加載微調后的模型和圖片處理器。 - 圖片預處理:對待識別的圖片進行預處理,包括縮放、歸一化等操作,以適應模型的輸入要求。
- 模型預測:將預處理后的圖片輸入到模型中,進行情感預測。
- 結果輸出:輸出模型預測的情感類別。
代碼實現
import os
import torch
from PIL import