使用GPU製作數據集

(1)確保本地環境對GPU的支持,當數據集的製作需要tensorflow裏,需要安裝gpu版本的tensorflow  測試腳本:https://github.com/dualven/insightface/blob/master/src/align/tensorflowTest.py 安裝腳本: conda install tensorflow-gpu (2) 修改align腳本align_gb
相關文章
相關標籤/搜索