1. 程式人生 > >【影象處理】人臉預處理工具FaceTools

【影象處理】人臉預處理工具FaceTools

前言

在做人臉分析的時候,常常要求人臉資料已經經過檢測、歸一化、對齊等操作,然而我們網上下載的資料庫,大部分是沒有經過這一些處理的,所以也經常需要我們自己寫程式來完成這些資料預處理。

雖然網上都有現成的一些人臉檢測和關鍵點檢測的可執行檔案提供,但是,使用起來都不大方便。

在這裡我提供了一個簡單的工具:用於處理簡單的人臉檢測、關鍵點檢測、基於關鍵點檢測的人臉對齊這三個功能。最大的優點就是你只需要提供相應的待處理的影象檔案路徑和儲存路徑,即可一鍵完成人臉分析的一系列預處理工作。採用的是python語言,人臉檢測方法和關鍵點檢測方法是採用香港中文大學提供的二進位制軟體包,我只是將其封裝起來,便於使用。

使用平臺:Windows

使用方法

比如:我們需要將D:\Test\Val 下的所有的人臉影象都統一檢測和對齊,那麼只需要在在main.py中,修改相對應的路徑即可:

if __name__ == "__main__" :
    '''
    @param: 提供的引數:1,影象的檔名位置,2,需要保留的檔案位置,3,影象的格式列表
    '''
    ImagePath=r'D:\Test\Val' #原始的影象路徑
    savePathDetect=r'D:\Test\Val_detect'#儲存中間檢測到的影象的路徑
    savePathAligned=r'D:\Test\Val_aligned'
#對齊後的人臉影象的儲存路徑 test(ImagePath,savePathDetect,savePathAligned,tag_recover=True,savesize=[128,128])

以上。