1. 程式人生 > >python提取影象的名字*.jpg到txt文字

python提取影象的名字*.jpg到txt文字

# -*- coding:utf-8 -*-
import sys
sys.path.append('E:\\Anaconda\\libs')
import os #os:作業系統相關的資訊模組
import random #匯入隨機函式
#存放原始圖片地址
data_base_dir = "C:\\Users\\Administrator.MICROSO-1HCAN56\\Desktop\\qingxi"
file_list = [] #建立列表,用於儲存圖片資訊
#讀取圖片檔案,並將圖片地址、圖片名和標籤寫到txt檔案中
write_file_name = 'C:\\Users\\Administrator.MICROSO-1HCAN56\\Desktop\\qingxi\\dir.txt'
write_file = open(write_file_name, "w") #以只寫方式開啟write_file_name檔案
for file in os.listdir(data_base_dir): #file為current_dir當前目錄下圖片名
if file.endswith(".jpg"): #如果file以jpg結尾
write_name = file #圖片路徑 + 圖片名 + 標籤
file_list.append(write_name) #將write_name新增到file_list列表最後
sorted(file_list) #將列表中所有元素隨機排列
number_of_lines = len(file_list) #列表中元素個數
#將圖片資訊寫入txt檔案中,逐行寫入
for current_line in range(number_of_lines): 
write_file.write(file_list[current_line] + '\n')
#關閉檔案
write_file.close()