biao187/ImageProcessing

Image processing by mrlittlepig

ImageProcessing

Image processing by mrlittlepig

数据读取

将MNIST_data/mnist_data.zip解压到跟目录

利用utils/genFileList.py对数据进行处理生成label文件

利用utils/labelFile2Map.py将label文件用字典形式保存在内存里面

在datasets/imnist.py中process_images函数用来读取图片文件夹，返回训练所用到的图片数据和标签数据

训练数据

在classification.py中定义了训练网络，运行开始训练

中文博客