1. 程式人生 > >提取圖片標籤

提取圖片標籤

import os
def generate(dir, label):
    files = os.listdir(dir)
    files.sort()
    print('****************')
    print('input :', dir)
    print('start...')
    listText = open('all_list.txt', 'a')
    for file in files:
        fileType = os.path.split(file)
        if fileType[1] == '.txt':
            continue
        name = file + ' ' + str(int(label)) + '\n'
        listText.write(name)
    listText.close()
    print('down!')
    print('****************')
outer_path = '/Users/Administrator/code/top/top33'  # 這裡是你的圖片的目錄
if __name__ == '__main__':
    i = 0
    folderlist = os.listdir(outer_path)  # 列舉資料夾
    for folder in folderlist:
        generate(os.path.join(outer_path, folder), i)
        i += 1