MNIST手写体数字训练/测试数据集(图片)下载及Python分割预处理代码

#!/usr/bin/env python #-*- coding: utf-8 -*- import os import cv2 import ipdb ########################### Global Parameters ################################### trainFolder = 'train' testFolder = 'test' trainPrefix = 'mnist_train' testPrefix = 'mnist_test' postfix = '.jpg' sideLength = 28 ########################### Global Parameters ################################### ############################ Initialization ##################################### ROOT = os.getcwd() # MUST run in the directory of this very script ############################ Initialization ##################################### # Extract & Save each single-number image of MNIST def MNIST_extract(folder, prefix): print 'Category:', folder, prefix # For each number from 0 to 9 for i in xrange(0, 10): print 'Class:', str(i) # Read over-all image of each class allImage = cv2.imread(os.path.join(ROOT, folder, prefix + str(i) + postfix)) print 'Shape:', allImage.shape[0 : 2] count = 0 print 'Processing ...' for corY in xrange(0, allImage.shape[1] / sideLength): for corX in xrange(0, allImage.shape[0] / sideLength): count += 1 # Cropped image of single-number singleImage = allImage[corX * sideLength : (corX + 1) * sideLength, corY * sideLength : (corY + 1) * sideLength, :] ''' Show cv2.imshow('Single Number', singleImage) cv2.waitKey(1) # ''' # ''' Save saveLocation = os.path.join(ROOT, folder, str(i), prefix + str(i) + '_' + str(count) + postfix) cv2.imwrite(saveLocation, singleImage) # ipdb.set_trace() # ''' print 'Count:', count, 'All Saved.' cv2.destroyAllWindows() ########################## if __name__ == '__main__': # ''' \SWITCH/ : train OR test MNIST_extract(trainFolder, trainPrefix) ''' # \Division/ MNIST_extract(testFolder, testPrefix) # ''' # \END/