python 随机打乱 图片和对应的标签方法
如下所示:
#-*-coding:utf-8-*-
importos
importnumpyasnp
importpandasaspd
importh5py
importpylab
importmatplotlib.pyplotasplt
trainpath=str('C:/Users/49691/Desktop/数据集/train/')
testpath=str('C:/Users/49691/Desktop/数据集/test/')
n_tr=len(os.listdir(trainpath))
print('numoftrainingfiles:',n_tr)
train_labels=pd.read_csv('C:/Users/49691/Desktop/数据集/sample_submission.csv')
train_labels.head()
fromskimageimportio,transform
x=np.empty(shape=(n_tr,224,224,3))
y=np.empty(n_tr)
labels=train_labels.invasive.values
name=train_labels.name.values
permutation=np.random.permutation(name.shape[0])
print(permutation)
print(labels[permutation])
save_data=pd.DataFrame({'name':permutation,'invasive':labels[permutation]})
save_data.to_csv('C:/Users/49691/Desktop/数据集/b.csv')
fork,vinenumerate(np.random.permutation(n_tr)):
print(k,v)
path='{0}{1}.jpg'.format(trainpath,v)
tr_im=io.imread(path)
x[k]=transform.resize(tr_im,output_shape=(224,224,3))
y[k]=float(labels[v-1])
以上这篇python随机打乱图片和对应的标签方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。