mnist 数据集转为 png 的 python代码，包含 mnist 数据集本身

from tensorflow.examples.tutorials.mnist import input_data import numpy as np import cv2 def findLabel( label ): max = 0 iMax = -1 for i in range( len( label ) ) : #print( label[i] , end = ' ' ) if label[i] > max : max = label[i] iMax = i #print( "max=", max , "iMax=" , iMax ) return iMax def main(): mnist = input_data.read_data_sets('data/',one_hot=True) with open("bmp/28x28x8bGray.bmp" , "rb" ) as r : head = r.read( 54 + 1024) procList = [ [ mnist.train.num_examples , mnist.train.labels , 'trainImg/train_' , mnist.train.images ] , [ mnist.validation.num_examples , mnist.validation.labels , 'validationImg/valid_' , mnist.validation.images] , [ mnist.test.num_examples , mnist.test.labels , 'testImg/test_',mnist.test.images ] ] tempBmp = "bmp/mnistTemp.bmp" for ite in procList : for i in range( ite[0] ) : print('=',end='') imax = findLabel ( ite[1][i] ) pngName = ite[2] + str( i ) +'_('+ str( imax) + ').png' image= ite[3][i]*200 image1 = image.astype( np.uint8 ) with open( tempBmp , "wb" ) as f: f.write( head ) for row in range( 27, -1 ,-1 ) : start = row*28 f.write( image1[start:start+28 ]) iBmp = cv2.imread( tempBmp , 0 ) cv2.imwrite( pngName, iBmp ) if i % 100 == 0 : print('\n') main()