In [10]:

    

import os
import numpy as np
from PIL import Image


    

In [11]:

    

dir = "/media/gui/LENOVO/学习/AMAZON/train-jpg/"


    

In [12]:

    

width = 64
height = 64
l = width*height


    

In [13]:

    

train_jpg_list = os.listdir(dir)


    

In [14]:

    

input_count = len(train_jpg_list)
cut_num = 1000
remain = input_count % cut_num
split = input_count // cut_num


    

In [15]:

    

del train_jpg_list


    

In [16]:

    

def ImageToArray(file,width,height):
    '''3 color channels, 1*length'''
    img_raw = Image.open(file)
    img = img_raw.resize((width, height), Image.ANTIALIAS)
    width,height = img.size
    data = img.convert("RGB").getdata()
    data = np.array(data, dtype='int')
    result = np.reshape(data,(1,width*height*3))
    return result


    

In [19]:

    

for j in range(split):
    if j+1 == split:
        count = cut_num + remain
    else: count = cut_num
    input_images = np.array([[0]*l*3 for i in range(count)])
    for i in range(j*cut_num,j*cut_num+count):
        file = dir + "train_%s.jpg" % i
        data = ImageToArray(file,width,height)
        if i % 1000 == 0: print("Have completed " + str(i))
        input_images[i-j*cut_num] = data
    np.savetxt( "train-jpg-all-"+str(j),input_images,fmt="%d")
print ('Completed!')


    

    




Have completed 0
Have completed 1000
Have completed 2000
Have completed 3000
Have completed 4000
Have completed 5000
Have completed 6000
Have completed 7000
Have completed 8000
Have completed 9000
Have completed 10000
Have completed 11000
Have completed 12000
Have completed 13000
Have completed 14000
Have completed 15000
Have completed 16000
Have completed 17000
Have completed 18000
Have completed 19000
Have completed 20000
Have completed 21000
Have completed 22000
Have completed 23000
Have completed 24000
Have completed 25000
Have completed 26000
Have completed 27000
Have completed 28000
Have completed 29000
Have completed 30000
Have completed 31000
Have completed 32000
Have completed 33000
Have completed 34000
Have completed 35000
Have completed 36000
Have completed 37000
Have completed 38000
Have completed 39000
Have completed 40000
Completed!

In [ ]: