超可❤的编程知识🌠🌙
最终决定,重新写一下vgg16的分类层(classifier):
en~~~~~~~~后来我觉得人家vgg16的网络写的还真好,就这样吧,就偷了个懒,略略略~~~
结果就很美丽了~
这里还是有点问题的,就是这里的这个有两层Sequential,所以不太好。狮虎建议我改成这样子:
后来,我就去上课,想用这段时间,让这个预训练模型下载一下,结果:
就用上面的这个链接:https://download.pytorch.org/models/vgg16-397923af.pth
(1)New Terminal:
然后就好了~
但是准确率只有0.2,学习率是le-2,调成le-3试试~
原因:因为vgg16没有batchnorm层,所以当学习率大的时候,容易波动~
我又试了一下别的图片,发现无论我输入什么,输出都是right~这是严重的过拟合啊,咋回事啊!!!
而且我发现对于没有睁眼睛的图片,一般都会判断成left。后来发现,它不会就蒙left。
明天需要完成以下几个问题:
(1)这个vgg16还是有问题,做一下数据增强,来解决一下过拟合的问题。做一些数据增强:例如,crop flip colorjiter;
(2)resnet模型:
1)放在电脑上,用cpu跑一下。可以直接调用摄像头实时检测,看一下效果。(重新安装电脑cuda)
2)放在电脑上,数据回传给服务器,判断好了以后,再传回来。