8000 GitHub - ginnux/cifar10: An deep-learning program of computer vision, which finishes CIFAR-10 pictures classification.
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
/ cifar10 Public

An deep-learning program of computer vision, which finishes CIFAR-10 pictures classification.

Notifications You must be signed in to change notification settings

ginnux/cifar10

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 

Repository files navigation

CIFAR-10 Classification CIFAR-10图像分类任务

A deep-learning program of computer vision, which finishes CIFAR-10 pictures classification.

一个基于计算机视觉的深度学习项目,用于完成CIFAR-10数据集的分类任务。

Necessary Python Lib 依赖库

torch (1.13 for cuda 11.7)
torchvision
matplotlib
numpy

Necessary Environment 依赖环境

python 3.10
NVIDIA cuda 11.7

Parameter 参数

trainMode:

set 0 for checking testset
set 1 for continue training
set 2 for creating a new model to train
set 3 for checking both testset and trainset

REPEAT:

times of training

Notice 注意

The first time to train maybe you can set trainMode 2 and REPEAT 20.
Training with CPU will be VERY VERY SLOW!

首次训练可以将trainmode设置为2,REPEAT设置为20。 如果基于CPU训练会非常缓慢,建议使用NVIDIA显卡训练。

Outcome 结果

The accuracy of training is 95.73%, the accuracy of testing is 88.74%. img

About

An deep-learning program of computer vision, which finishes CIFAR-10 pictures classification.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

0