MoyanZitto / keras-cn

Chinese keras documents with more examples, explanations and tips.
1.57k stars 280 forks source link

数据读取速度的问题 #71

Open yifa-n opened 7 years ago

yifa-n commented 7 years ago

背景:我将大约50W张图片做一个vgg的finetune。 使用,fit_generator,这个图片读取速度感觉太慢了。 硬件是K80的GPU,nvidia-smi看GPU的使用率很低,接近0. 是不是很多时间浪费在了数据的IO上面?

caffe之类的框架有lmdb等快速索引的工具。 keras文章中支持h5文件。能不能给出一个使用h5的demo ~~~