시각화 데이터 세트 cifar-10

 

가져 NP로 NumPy와
 에서 PIL의 수입 이미지
 가져 오기 피클
 수입 운영 체제의 
 
CHANNEL = 3 
WIDTH = 32 
HEIGHT = 32 
 
데이터 = [] 
라벨 = [] 
분류 = [ ' 비행기 ' , ' 자동차 ' , ' 새 ' , ' 고양이 ' , ' 사슴 ' , ' 개 ' , ' 개구리 ', ' 말 ' , ' 배 ' , ' 트럭 ' ] 
 
에 대한 I 의 범위 (5 ) : 
    (오픈 R은 " ... \ cifar -10- 일} - 피리 \ data_batch_ " + STR (I + 1), 모드 = ' RB ' : 파일) 
        data_dict = pickle.load (파일 인코딩 = ' 바이트 ' ) 
        데이터 [B + = 목록 (data_dict ' 데이터 ' )] 
        라벨 + = 목록 (data_dict [B ' 라벨 '])
 
IMG = np.reshape (데이터 [- 1 , 채널 폭, 높이]) 
 
 
data_path = " 데이터 / 이미지 / " 
만약  하지 os.path.exists (data_path) 
    os.makedirs (data_path) 
에 대한 I 의 범위 (IMG .shape [0]) : 
 
    R = IMG [I] [0] 
    g = IMG [I] [1 ] 
    , B = IMG [I] [2 ] 
 
    , IR = Image.fromarray (R) 
    IG = Image.fromarray (g) 
    IB = Image.fromarray (b) 
    의 RGB = Image.merge ( " RGB "(IR, IG, IB)) 
 
    이름 = " img- " + STR (I) + " - " + 분류 [라벨 [I] + " .png를 " 
    rgb.save (data_path + 이름 " PNG " )

오픈 (R & LT "... \ CIFAR 피라-10-일괄는 \ data_batch _"+ STR (I + 1), MODE = 'RB'. ) 파일로서 : 첫 번째 파라미터는 파일의 문장 전체 경로이다. 의 자체 파일 저장 위치에 따라 
이 매개 변수의 변화.