提交 e33baa2b 编写于 作者: W wangshipeng01

add generate lable list script

上级 1be92714
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)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册