Skip to content

llllTnTlll/ImgClassifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ImgClassifier

基于GoogLeNet v1的图像分类器

准备数据集

在开始训练之前请确保拥有足够的用于训练神经网络的图像,并将其分为训练集和测试集,在train_cputrain_gpu中正确配置其路径

进入训练

请在拥有独显的设备上采用train_gpu进行训练,这将成倍的提升训练效率,更快的得到分类模型。

使用gpu训练将会把所有代训练图像分批加载到内存中,请保证你拥有足够的内存空间,否则建议使用cpu版本进行训练,cpu版本采取从硬盘中加载的训练方式,当然,训练速度也会降低

多gpu和显存使用限制

请参照train_gpu中的代码注释进行正确的设置

About

GoogLeNet v1 tensorflow代码实现

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages