计算机视觉
lai199508
这个作者很懒,什么都没留下…
展开
-
Resnet for Fashion_Mnist(一)
Resnet for Fashion_Mnist目录写作目的Fashion_Mnist制作数据集写Datasets函数模型定义模型训练模型加载及预测数据目录写作目的通过使用resnet 实现 Fashion_Mnist,学会使用pytorch框架。这篇博客全方位介绍了如何使用pytorch,包括数据制作、模型定义、模型训练及验证,模型保存、模型加载、测试集预测。Fashion_Mnist...原创 2018-12-08 17:13:56 · 1470 阅读 · 0 评论 -
Resnet for Fashion_Mnist(四)
Resnet for Fashion_Mnist(五)模型加载及预测数据下面给出模型加载及预测数据的代码,预测结果保存到csv文件中。# -*- coding: utf-8 -*-import torchimport MyDataSetfrom torchvision import transformsfrom torch.autograd import Variableimpor...原创 2018-12-16 19:46:36 · 429 阅读 · 0 评论 -
Resnet for Fashion_Mnist(三)
Resnet for Fashion_Mnist(四)模型训练下面给出模型训练、验证的代码,完成后保存模型。import torchfrom torchvision import datasets, models, transformsimport torch.optim as optimimport modelimport utilsimport timeimport argp...原创 2018-12-16 19:48:32 · 500 阅读 · 0 评论 -
Resnet for Fashion_Mnist(二)
Resnet for Fashion_Mnist(三)模型定义下面给出Resnet的模型定义代码,包含了resnet18。想写其他restnet模型,可以在此基础上扩展。import torchimport torch.nn as nnimport utilsimport mathfrom torch.autograd import Variable__all__ = ['Re...原创 2018-12-16 19:49:56 · 891 阅读 · 0 评论