【MaskTheFace】給人臉圖片戴口罩！

點擊上方“小白學(xué)視覺”，選擇加"星標(biāo)"或“置頂”

重磅干貨，第一時間送達(dá)

【導(dǎo)讀】MaskTheFace是一個基于計算機(jī)視覺的腳本，用于給人臉數(shù)據(jù)集戴口罩。

https://github.com/aqeelanwar/MaskTheFace

MaskTheFace使用基于dlib的人臉標(biāo)志檢測器來識別人臉傾斜度和應(yīng)用口罩所需的人臉六個關(guān)鍵特征?；诿娌績A斜度，從口罩庫中選擇相應(yīng)的蒙版模板。然后根據(jù)六個關(guān)鍵特征對模板口罩進(jìn)行轉(zhuǎn)換，以使其完美貼合在面部。MaskTheFace提供了許多口罩可供選擇。在各種條件下很難收集遮罩?jǐn)?shù)據(jù)集。MaskTheFace可用于將任何現(xiàn)有的面部數(shù)據(jù)集轉(zhuǎn)換為口罩?jǐn)?shù)據(jù)集。MaskTheFace識別圖像中的所有面部，并考慮到各種限制（例如面部角度，蒙版適合度，光照條件等），將用戶選擇的口罩應(yīng)用于這些口罩。單個圖像或整個圖像目錄可用作代碼的輸入 。

安裝

確保安裝了Python3.6環(huán)境

克隆庫

git clone https://github.com/aqeelanwar/MaskTheFace.git

安裝必需的包

cd MaskTheFacepip install –r requirements.txt

運(yùn)行

cd MaskTheFace# Genericpython mask_the_face.py --path  --mask_type  --verbose --write_original_image
# Examplepython mask_the_face.py --path 'data/office.jpg' --mask_type 'N95' --verbose --write_original_image

參數(shù)概覽

支持的口罩類別

1. Surgical

2. N95

3. KN95

4. Cloth

5. Gas

特色功能

支持戴多種類型的口罩，支持多人圖片