add generate lable list script
parent
1be9271493
commit
e33baa2b7c
|
@ -0,0 +1,39 @@
|
||||||
|
import scipy.io
|
||||||
|
import numpy as np
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
"""
|
||||||
|
Usage: python generate_flower_list.py ./jpg train > train_list.txt
|
||||||
|
python generate_flower_list.py ./jpg valid > val_list.txt
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
data_path = sys.argv[1]
|
||||||
|
imagelabels_path='./imagelabels.mat'
|
||||||
|
setid_path='./setid.mat'
|
||||||
|
|
||||||
|
"""
|
||||||
|
imagelabel.mat
|
||||||
|
jpg_name 1 2 3 ...
|
||||||
|
label 32 12 66 ...
|
||||||
|
"""
|
||||||
|
labels = scipy.io.loadmat(imagelabels_path)
|
||||||
|
labels = np.array(labels['labels'][0])
|
||||||
|
setid = scipy.io.loadmat(setid_path)
|
||||||
|
|
||||||
|
d = {}
|
||||||
|
d['train'] = np.array(setid['trnid'][0])
|
||||||
|
d['valid'] = np.array(setid['valid'][0])
|
||||||
|
d['test']=np.array(setid['tstid'][0])
|
||||||
|
|
||||||
|
"""
|
||||||
|
setid.mat
|
||||||
|
jpg_name 24 6 100 65 32 ...
|
||||||
|
label 4 ...
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
for id in d[sys.argv[2]]:
|
||||||
|
message = str(data_path)+"/image_"+str(id).zfill(5)+" "+str(labels[id-1])
|
||||||
|
print(message)
|
Loading…
Reference in New Issue