tensorflow 加載本地文件
https://www.cnblogs.com/ypzhai/p/9997856.html
加載圖片也可以用 PIL 模塊下的 Image 函數(shù)秀姐,顯示圖片直接用 img.show()
將圖片轉(zhuǎn)換成灰度圖:img = np.array(img.convert('L'))
其實python加載文件的方法很多纵隔,主要根據(jù)數(shù)據(jù)的格式來選擇合適的方法來加載數(shù)據(jù)技潘。上面的鏈接主要講解 官方數(shù)據(jù)集(.npz )的加載
########這是用 PIL 模塊下的 Image 讀取圖片的代碼,這個方法沒有 cv2 讀取圖片好用感覺#
from PIL import Image
dir =r"./garbage_classify/train_data/"
img_array = []
for i in os.listdir(dir):
? ? print(i)
? ? tmp = i
? ? if tmp.spilt('.')[1] == 'jpg':
? ? ? ?# d ="./garbage_classify/train_data" + i
? ? ? ? img = Image.open(dir+i)
? ? ? ? img.show() ####就這一點很方便!?鹑椤!!援所!
###########這是用 cv2 讀取文件夾下的圖片###########
import cv2
dir =r"./garbage_classify/train_data/"
for iin os.listdir(dir):#雖然 listdir 函數(shù)把目錄下的文件都列出來了,但是你不能通過這種方式讀取圖片欣除,想要讀取住拭,還得絕對路徑
? ? print(i)
????tmp = i
????if tmp.split('.')[1] =='jpg':
? ??????img = cv2.imread(dir + i ,1)
? ??????cv2.imshow('image',img)
????????cv2.waitKey(2500)
????????img_array.append(img)
tensorbord 可視化遇到的問題
設(shè)置斷點續(xù)訓(xùn):
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_save_path,
save_weights_only=True,save_best_only=True)? 和設(shè)置 Tensorbord 類似