17.根據圖像路徑及label生成txt文件

#coding:utf-8
import os
import random
import glob
import numpy as np
import shutil

img_path = "/home/ming/data"
# train val img folder
train_path = img_path + "/sku_train"
val_path = img_path + "/sku_val"


# save txt
train_txt = img_path + "/train.txt"
val_txt = img_path + "/text.txt"

f1 = open(train_txt, "w")
f2 = open(val_txt, "w")

for label in os.listdir(train_path):
        img_list = glob.glob(os.path.join(train_path, label, "*.jpg"))
        for img in img_list:
                f1.write(img + " " + str(label) + "\n")


print("create train.txt done...")
for label in os.listdir(val_path):
        img_list = glob.glob(os.path.join(val_path, label, "*.jpg"))
        for img in img_list:
                f2.write(img + " " + str(label) + "\n")
print("create test.txt done...")
f1.close()
f2.close()

print("shuffle...")
os.system("shuf %s -o %s" %(train_txt, train_txt))
os.system("shuf %s -o %s" %(train_txt, train_txt))
os.system("shuf %s -o %s" %(train_txt, train_txt))
os.system("shuf %s -o %s" %(val_txt, val_txt))
os.system("shuf %s -o %s" %(val_txt, val_txt))
os.system("shuf %s -o %s" %(val_txt, val_txt))