人脸识别模型的最后完善

2018-11-12  本文已影响25人  PolarBearWYY

                              超可❤的编程知识🌠🌙

经历了双十一剁手的我,现在在掉这笔敲代码,你信吗?

       我狮虎建议我保存权重,不要保存模型,因为保存模型的模型不能很方便的加载。而保存权重的话,没有什么弊端。有一种方法既保存模型又保存权重,这个方法很复杂,但是需要先定义模型再加载。

      我决定,先看一下我狮虎写的代码,自己摸索一下。(悄咪咪的补充一句,你要是想在服务器上面的notebook上面打开一个.py文件啊,你就,上传,就行了(/≧▽≦)/)

我以为这就是保存参数,其实这是在定义优化器,这个优化器对不同的层,设置不一样的lr等 果然是,我狮虎的模型定义的就是fc层和resnet层 那我猜测,及时这里是在仅保存参数了。

下面请看与我的保存模型的代码对比:

model1是我修改后的模型~ 这一步我将我的模型model1保存为名为“resnet50_face”的.py文件 出现这个问题的原因是因为模型没有用cuda() 重新运行后,并没有什么效果,还是原来的错误 但是还是有问题

我狮虎说:原来是因为数据没有cuda

终于好了。

继续完善vgg16:

这个问题是我多写了一个vgg16.classifier.这个.classifier是多余的 这是原来的classifier

我做完以后却只剩下:

所以决定,重新写一下这个分类层 每天早上叫醒我的不是梦想,而是,我要继续来实验室完善我的代码……

未完,待续……

上一篇 下一篇

猜你喜欢

热点阅读