现在的代码:
- import os
- filename_list = []
- filename = os.listdir(r"D:\Work\process\test\polyu-all-train")
- train_img = []
- base_dir = "D:\\Work\\process\\test\\polyu-all-train\"
- train_list=[]
- listlables = []
- for img in filename:
- array = Image.open(base_dir+img)
- array = array.resize((size_m,size_n))
- # array = array.resize(64,64)
- train_img.append(np.asarray(array))
- for i in range(26):
- name = "_p"+str(i+1)+"_"
- if img.find(name)!=-1:
- listlabels = i+1
- train_list.append(listlabels)
复制代码
对应标签结果是:- print(train_list)
- len(train_list)
- [10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12, 12, 13, 13, 14, 14, 15, 15, 16, 16, 17, 17, 18, 18, 19, 19, 1, 1, 20, 20, 21, 21, 22, 22, 23, 23, 24, 24, 25, 25, 26, 26, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9],需要的是里面标签按照顺序,但必须是图片对应的
复制代码
文件名还是这样不变,但顺序就是:
图片:s任意_p1_c任意,s任意_p2_c任意,---> s任意_p26_c任意
对于的标签:1...1...2...2..3..3...4...4...................26....26 |