Question

各位，问个关于caffe生成train.txt的问题

如果目录是多级的如：E:\train\aodi\aodiA4\black\00.jpg的格式，
然后生成的train.txt文件中的标签格式是：
aodi\aodiA4\black\00.jpg 0 0 0
这该怎么弄

Answer 1

def get_file_list(folder):

    file_list = []; 

    for root, dirs, files in os.walk(folder):

        for f in files:

            path=root+os.path.sep+f

            file_list.append(path)

    return file_list

def write2file(a_list,file_path):

    f=open(file_path,'a',encoding='utf-8')

    for line in a_list:

        f.write(line+'\n')

    f.close()

def create_tab(file_list,path_root):

    for i in range(0,len(file_list)):

        file_list[i].replace(file,path_root)

        file_list[i] += " 0 0 0"

    return file_list

        

def main():

    root = r"E:\train"

    train_file_path=r"E:\train.txt"

    write2file(create_tab(get_file_list(root),root),train_file_path)

if __name__ == '__main__'

    main()

追问

这个是哪个语言写的啊？

追答

python啊，你问题挂的标签不是python吗