Skip to content

A neural network based on Letnet5 model to fit and predict the CIFAR-10 datasets.

Notifications You must be signed in to change notification settings

guoyanan1g/simple-Letnet5

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

simple-Letnet5

A neural network based on Letnet5 model to fit and predict the CIFAR-10 datasets. Without optimization of hyper-parameters.

CIFAR10数据集共有60000个样本,每个样本都是一张32*32像素的RGB图像(彩色图像),每个RGB图像又必定分为3个通道(R通道、G通道、B通道)。这60000个样本被分成了50000个训练样本和10000个测试样本。 不同类别的物体用不同的标签值,CIFAR10中有10类物体,标签值分别按照0~9来区分,他们分别是飞机( airplane )、汽车( automobile )、鸟( bird )、猫( cat )、鹿( deer )、狗( dog )、青蛙( frog )、马( horse )、船( ship )和卡车( truck )。

input: torch.Size([b,3,32,32]) ,b=BatchSize
output: torch.Size([b,10])

网络结构: 1.卷积层 2,avg池化层 3.卷积层 4.avg池化层 5.线性层 6.线性层
激活函数均为Relu

About

A neural network based on Letnet5 model to fit and predict the CIFAR-10 datasets.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages