pickle

利⽤pickle⽣成数据集⽂件

下⾯展⽰代码。

defget_image(path):

#从⽂件中导⼊图⽚，制作数据集

i=0

j=0

fordir_r(path):

fall_path=h((path,dir_item))

(fall_path):

get_image(fall_path)

el:

ifdir_th('.jpg'):

if('weiqi'indir_item):

image=(fall_path)

image=resize_image(image)

e('E:/python/opencv/data_image/'+str(i)+'weiqi'+'.jpg',image)

i+=1

print('i:',i)

(image)

(dir_item)

el:

image=(fall_path)

image=resize_image(image)

e('E:/python/opencv/data_image/'+'not'+str(j)+'.jpg',image)

j+=1

print('j:',j)

(image)

(dir_item)

returnimages,labels

#读取⽂件夹中的图⽚，返回ndarray数据集

defload_image(path):

#step1：加载图⽚

#step2：处理图⽚

#step3：⽣成数据集

images,labels=get_image(path)

images=(images)

print()

labels=([1if('wq'inlabel)el0forlabelinlabels])

print(labels)

train_dic={'train_datat':images,'labels':labels}

withopen(path+'train_data','wb')asf:

(train_dic,f,2)

()

#参数protocol是序列化模式，默认是0（ASCII协议，表⽰以⽂本的形式进⾏序列化），protocol的值还可以是1和2（1

#和2表⽰以⼆进制的形式进⾏序列化。其中，1

#是⽼式的⼆进制协议；2

#是新⼆进制协议）。

returnimages,labels