使用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
相关文章
相关标签/搜索