我正在嘗試從本地檔案夾加載資料集并將其用作 tf 資料集。檔案夾結構為:
../dataset/
class_0/
class_1/
其中 0 類子檔案夾包含所有 0 類影像和 1 類所有 1 類影像。
為了實作這一點,我的代碼是:
images = image_dataset_from_directory('../dataset/',
shuffle=True,
batch_size=32,
image_size=(1080,1920))
所有影像的大小(1080,1920,3)或(1920,1080,3)
我正在嘗試使用以下方式顯示影像:
for image, labels in images.take(1):
img = image[0].numpy() # take first image of batch
print(img.shape)
img = Image.fromarray(img, 'RGB')
img.save('my.png')
img.show()
列印影像形狀=(1080, 1920, 3)
然而,PIL 顯示的影像是扭曲的,看起來像是隨機噪聲。
關于我做錯了什么的任何想法?
uj5u.com熱心網友回復:
問題似乎在于 Pillow 的浮點數
在您的轉換功能中,您有img = Image.fromarray(img, 'RGB').
將其更改為img = Image.fromarray(img.astype('uint8'), 'RGB')應該可以解決此問題。
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/495103.html
上一篇:在InternetExplorer上獲取“localhost拒絕連接”。然后在微軟邊緣頁面出現但不加載影像
下一篇:快速分別拖動多張圖片
