Gigabyte AORUS GTX 1080官方驱动不支持macOS。我的系统是macOS10.13.4,查过很多资料后,发现解决方案比较伍弯麻烦,需要使用macOS 10.13.3的kext,如下图:
最终还是将系统从10.13.4降到了10.13.2。严格按照 步骤 安装:
安装成功后在Graphics/Displays和NVIDIA Driver Manager可查到闭御外置显卡的信息:
目前无法通过pip安装GPU版本的MXNet,只能通过 源码 )安装:
安装好CUDA9.1后,测试GPU:
安装好cuDNN7.0.5,构建MXNet,最后采用基于MXNet的测试用例验证一下包:
安装过程中需要注意的是:
用install_name_tool将lib指向修改为正确的即可:
Win10下安装CUDA10.2版本Mxnet,咐搭当前只找到 mxnet_cu102-1.7.0-py2.py3-none-win_amd64.whl 。安装后安装gluoncv,通过 git clone https://github.com/dmlc/gluon-cv 获得到了源码,经验瞎简滑如磨腊下:
安装mxnet
Cpu版:pip install mxnet
Gpu版:pip install mxnet-cu80
若cuda版本为cuda-9.0,改为pip install mxnet-90
pip或apt安装sklearn,easydict
制作数据集
将图片分类并放入不同文件夹如纯
运行python im2rec.py train --list ./可生成.list文件,包含图片列表
运行python im2rec.py train ./ train.rec和train.idx训练文件
--train-ratio 0.9生成验证数据集.bin文件参数为训练数据集和验证数据集之比
--resize 128 128指定生成数据集的图片大小
参考 https://github.com/apache/incubator-mxnet
将生成的.rec,.idx,.bin(非必须)文件放入datasets/faces_emore中
新建property文本,写入图片数量,图片长宽 example : 86545 128 128
例:
python -u train.py --network m1 --loss softmax --dataset emore,1
使用softma、nosoftmax、arcface或cosface训练完成后,使用生成模型运行三元组损失局橡雀训练
例:
Python -u train.py --network m1 -loss triplet --lr 0.005 --pretrained ./models/m1-softmax-emore
参数说明
--dataset训练集位置,具体位置查看config.py 108至120行
--network网络模型 候选参数 : r100 r100fc r50 r50v1(基于resnet) d169 d201(基于densenet) y1 y2(基于mobilefacenet) m1 m0.5(基于mobilenet) mnas mnas05 mnas025(基于桐早mnasnet)
--loss损失函数 候选参数 :softmax(标准损失函数) nsoftmax (组合损失函数)arcface cosface combined triplet(三元组损失) atriplet
--ckpt模型存储时间。0: 放弃存储 1:必要时存储(验证集准确率达标时,若无验证集则不存储 3:总是存储)
--lr学习率
--lr-steps学习率改变方法 例:’10000,20000,2200000’即达到图片数量时学习率*0.1
--per-batch-size每次的训练的数量 数量越少,占用显卡内存越少
参考
https://github.com/deepinsight/insightface/tree/master/recognition
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)