美文网首页
重新写vgg16的分类层

重新写vgg16的分类层

作者: PolarBearWYY | 来源:发表于2018-11-13 10:09 被阅读37次

                            超可❤的编程知识🌠🌙

 

今早快迟到了,我突然想放歌,就在路上放,wuo~wuo~~~

最终决定,重新写一下vgg16的分类层(classifier):

这是vgg16本来的样子 这是目前我的代码跑出来的样子,还是有些问题的

en~~~~~~~~后来我觉得人家vgg16的网络写的还真好,就这样吧,就偷了个懒,略略略~~~

没错,就是结合了一下,哈哈哈

结果就很美丽了~

这样就ok了

这里还是有点问题的,就是这里的这个有两层Sequential,所以不太好。狮虎建议我改成这样子:

vgg16其实一共有两大层,一层是features,一层是这个classifier 这下,这里就对了,就只有一层Sequential了

后来,我就去上课,想用这段时间,让这个预训练模型下载一下,结果:

于是我狮虎建议我,手动下载一下,再上传到服务器~

就用上面的这个链接:https://download.pytorch.org/models/vgg16-397923af.pth

哼~看起来还挺快的~ 下载的就快多啦~
再上传一下,完美~ 还有几个步骤需要做: 开始吧,kiyomi~

(1)New Terminal:

输入:mv Faces/vgg16-397923af.pth .torch/models/

然后就好了~

我发现我的旺仔小牛奶的取货码是456~

但是准确率只有0.2,学习率是le-2,调成le-3试试~

果然学习率改小了以后,就准确率大了很多啊~

原因:因为vgg16没有batchnorm层,所以当学习率大的时候,容易波动~

但是判断出错,因为过拟合了

我又试了一下别的图片,发现无论我输入什么,输出都是right~这是严重的过拟合啊,咋回事啊!!!

用训练集里面的,结果就是对的,这孩子~ 一定记得改图片的后缀名字啊 还挺准啊 但是对于这种,没脸的,还真是不准~

而且我发现对于没有睁眼睛的图片,一般都会判断成left。后来发现,它不会就蒙left。

明天需要完成以下几个问题:

(1)这个vgg16还是有问题,做一下数据增强,来解决一下过拟合的问题。做一些数据增强:例如,crop flip colorjiter;

(2)resnet模型:

1)放在电脑上,用cpu跑一下。可以直接调用摄像头实时检测,看一下效果。(重新安装电脑cuda)

2)放在电脑上,数据回传给服务器,判断好了以后,再传回来。

今天又是开心的一天呐~

相关文章

网友评论

      本文标题:重新写vgg16的分类层

      本文链接:https://www.haomeiwen.com/subject/qvrkfqtx.html