diff --git a/dataset/generate_flower_list.py b/dataset/generate_flower_list.py new file mode 100644 index 0000000000000000000000000000000000000000..5496516255e5a82e82acf1ded4fda8c73d207b3a --- /dev/null +++ b/dataset/generate_flower_list.py @@ -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)