import os
from skimage import io,transform
import glob
import numpy as np
path="flower/";
h=224;
w=224;
c=3;
pathname=[path+x for x in os.listdir(path)];
print(pathname);
count=0;
imgs=[];
labels=[];
for idx,dirname in enumerate(pathname):
for i in glob.glob(dirname+"/*.jpg"):
count+=1;
print("Reading (%s,%d):"%(i,count));
img=io.imread(i);
img=transform.resize(img,(h,w,c));
imgs.append(img);
labels.append(idx);
images=np.asarray(imgs,np.float32);
labels=np.asarray(labels,np.int32);
print(images);
print(labels);
num_samples=images.shape[0];
print(num_samples);
arr=np.arange(num_samples);
print("arr=",arr);
np.random.shuffle(arr);
images=images[arr];
labels=labels[arr];
print(labels);
读取图片生成标签
猜你喜欢
转载自blog.csdn.net/qq_38588316/article/details/82956088
今日推荐
周排行