39 lines
940 B
Python
39 lines
940 B
Python
"""
|
|
.mat files data format
|
|
imagelabel.mat
|
|
jpg_name 1 2 3 ...
|
|
label 32 12 66 ...
|
|
|
|
setid.mat
|
|
jpg_name(10 records in a class) 24 6 100 65 32 ...
|
|
label 4 ...
|
|
"""
|
|
"""
|
|
Usage:
|
|
python generate_flower_list.py prefix_folder mode
|
|
python generate_flower_list.py jpg train > train_list.txt
|
|
python generate_flower_list.py jpg valid > val_list.txt
|
|
"""
|
|
|
|
import scipy.io
|
|
import numpy as np
|
|
import os
|
|
import sys
|
|
|
|
data_path = sys.argv[1]
|
|
imagelabels_path = './imagelabels.mat'
|
|
setid_path = './setid.mat'
|
|
|
|
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])
|
|
|
|
for id in d[sys.argv[2]]:
|
|
message = str(data_path) + "/image_" + str(id).zfill(5) + ".jpg " + str(labels[id - 1] - 1)
|
|
print(message)
|