简单记录VGG网络,pytorch+VggNet+CIFAR10

      最后更新:2022-08-02 23:54:28 手机定位技术交流文章

      目录

      数据集:

      Net

      train

      总结

      数据集:

      数据集使用CIFAR10,cifar 10 这个数据集一共有 50000 张训练集,十万张测试套件,两个数据集中的图像是png颜色图像,图片的大小是32x32x3,总共10个分类问题,分别为飞机、汽车、鸟、猫、鹿、狗、青蛙、马、船和卡车。这个数据集是网络性能测试的一个非常重要的指标,我们可以说,如果在这个数据集中一个网络超过另一个网络,因此这个网络的性能必须比其他网络更好,目前,该数据集的最好结果是测试集的大约95%准确性。

      但是这样下载是缓慢的,甚至卡片也不移动,所以我们可以替换其他下载者下载:

      只要下载它,然后将其解凍到目录里

      难点:因为CIFAR10数据集的图片大小为【32,32】,而VGG模型里要求输入的图片大小为【224,224】,所以大小不匹配在进行训练时就报错了,所以需要将其数据集的大小改成适配模型

      可以用到一下代码

      Net

      网络的结构简要地介绍在我的博客里,如果你有兴趣的话就可以看到

      简单记录一些经典的网络结构_subnet博客-CSDN博客

      代码部分:

      train

      #############################################################################

      ##                                上面的代码有些问题

      ##                                也不能说是重大问题吧!至少在我电脑是能运行的

      ##                                但是上传服务器,在服务器跑的话,就出问题了

      ##                                所以看脸了

      #############################################################################

      错误问题为:

      现在如何更改下列服务的代码:

      数据格式的定义已修改为:

      NET网络 removestran = transforms.Resize((224, 224)),x = tran(x)

      换句话说,图像的扩展是在 data_tf 和 OpenCV 中实现的,不要使用变换,不要使用它,并不知道如何修正错误检测

      总结

      上面提到的Vgg网络经过了20轮的训练,而且数据是难以承受的,虽然你不能过度适应,你必须集中力量

      经过20个时代,准确度是零。 如果你不了解,就太高了

      最后,经过修改后,我把 BatchNorm1d添加到网络上,这样我就可以

      不(BatchNorm1d)

      train_loss :  [2.303093245267258, 2.303139756707584, 2.303160000640108,

      2.3031104309174717, 2.303226224296843, 2.303206982210164,

      2.303128406214897, 2.3031576993825187, 2.3029416113558328,

      2.3030955550615744, 2.30320418673708, 2.303175717058694,

      2.3031793052278213, 2.3031632662429224, 2.3030178775567838,

      2.303055908064098, 2.303198173222944, 2.303126417462478]

      train_acc :  [0.10016384271099744, 0.10088315217391304, 0.09906489769820973, 0.09826566496163683, 0.09802589514066497, 0.09950447570332481,

      0.09802589514066497, 0.09906489769820973, 0.10094309462915602,

      0.10068334398976982, 0.09936460997442455, 0.09844549232736573,

      0.09744645140664962, 0.09458919437340153, 0.10124280690537084,

      0.10086317135549872, 0.09736652813299233, 0.09780610613810742]

      有(BatchNorm1d):

      train_loss :  [5.834198918641376, 2.6215473267123524, 2.1093473187492937,

      1.7783519804020367, 1.5179658337017459, 1.3456324152934276,1.2027138713223244, 1.1024912583553577, 1.0052912438769475,0.9160828074378431, 0.8292376793863828, 0.7431533735274048,0.659939240845268, 0.582230176081133, 0.5101518849735065,

      0.45334468538041617, 0.3936309843417019, 0.3504536931433946,0.3098974650549462, 0.2667675401939227]

      train_acc :  [0.18532209079283887, 0.29411764705882354, 0.3796954923273657, 0.4420756074168798, 0.501678388746803, 0.5523897058823529, 0.5899136828644501, 0.6228021099744245, 0.6544916879795396, 0.68130594629156, 0.7115768861892583, 0.7400695332480819, 0.7686421035805626, 0.7947170716112532, 0.8206122122762148, 0.8402533567774936, 0.8641903772378516, 0.8779571611253197, 0.8934622762148338, 0.907608695652174]

      最后,老婆压场

      如果你愿意,就给妻子一个赞美 ☺

      本文由 在线网速测试 整理编辑,转载请注明出处,原文链接:https://www.wangsu123.cn/news/31360.html

          热门文章

          文章分类