深度学习
Smiler_
这个作者很懒,什么都没留下…
展开
-
2020.3.31 miniImagenet数据集处理
miniImagenet数据集的结构为:所有的照片放在一个文件夹中,另外有三个csv文件分别是:train.csv,val.csv,test.csv,三个csv文件都有两列,第一列是文件名,第二列是标签。我在使用数据集时,把train数据,test数据,val数据分别都保存到对应的标签下。处理后的目录结构如下:具体处理代码:import csvimport osfrom ...原创 2020-03-31 18:14:36 · 8135 阅读 · 73 评论 -
Resnet训练 验证自己的数据集
使用resnet神经网络,在预训练模型的基础上,在自己的数据集上进行微调,最后在自己的数据集上验证效果。可以借鉴部分:数据加载resnet模型使用方法预训练模型加载,模型保存训练/验证/测试步骤需要修改部分:预训练模型加载路径自己数据集路径保存路径import torch.nn as nnimport mathimport pickleimport torchimport numpy as npimport torchimport torch.nn as nnimp原创 2021-06-02 16:07:45 · 5758 阅读 · 15 评论