Pytorch实现GoogLeNet解决各类数据集(cifar10/mnist/imagenet)分类

有一段时间没有更新了,这次我给大家带来的是大名鼎鼎的GoogleNet模型。也可以称为Inception v3模型。参考了源代码,写出了可读性与性能更优的模型,在模型上有些许微调,输入的图片大小是224 x 224 x 3.!!! 图片大小可以任意调节,甚至用来训练CIFAR10也可以。本文先放出代码。 完整代码点这里 ) import torch from torch import nn NU
相关文章
相关标签/搜索