人脸识别模型的最后完善
2018-11-12 本文已影响25人
PolarBearWYY
超可❤的编程知识🌠🌙
经历了双十一剁手的我,现在在掉这笔敲代码,你信吗?
我狮虎建议我保存权重,不要保存模型,因为保存模型的模型不能很方便的加载。而保存权重的话,没有什么弊端。有一种方法既保存模型又保存权重,这个方法很复杂,但是需要先定义模型再加载。
我决定,先看一下我狮虎写的代码,自己摸索一下。(悄咪咪的补充一句,你要是想在服务器上面的notebook上面打开一个.py文件啊,你就,上传,就行了(/≧▽≦)/)
我以为这就是保存参数,其实这是在定义优化器,这个优化器对不同的层,设置不一样的lr等
果然是,我狮虎的模型定义的就是fc层和resnet层
那我猜测,及时这里是在仅保存参数了。
下面请看与我的保存模型的代码对比:
model1是我修改后的模型~
这一步我将我的模型model1保存为名为“resnet50_face”的.py文件
出现这个问题的原因是因为模型没有用cuda()
重新运行后,并没有什么效果,还是原来的错误
但是还是有问题我狮虎说:原来是因为数据没有cuda
终于好了。
继续完善vgg16:
这个问题是我多写了一个vgg16.classifier.这个.classifier是多余的
这是原来的classifier
我做完以后却只剩下:
所以决定,重新写一下这个分类层
每天早上叫醒我的不是梦想,而是,我要继续来实验室完善我的代码……未完,待续……