[英]Displaying images loaded with pytorch dataloader
我正在處理一些由於發布圖像的聲譽限制而無法在此處發布的激光雷達數據圖像。 但是,當使用 pytorch ImageFolder 和 Dataloader 加載相同的圖像時,唯一的轉換是將圖像轉換為張量,似乎存在一些極端的閾值,我似乎無法找到原因。
下面是我顯示第一張圖片的方式:
dataset = gdal.Open(dir)
print(dataset.RasterCount)
img = dataset.GetRasterBand(1).ReadAsArray()
f = plt.figure()
plt.imshow(img)
print(img.shape)
plt.show()
這是我使用數據加載器並顯示閾值圖像的方式:
data_transforms = {
'train': transforms.Compose([
transforms.ToTensor(),
]),
'val': transforms.Compose([
transforms.ToTensor(),
]),
}
image_datasets = {x: datasets.ImageFolder(os.path.join(data_dir, x),
data_transforms[x]) for x in ['train', 'val']}
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
dataloders = {x: torch.utils.data.DataLoader(image_datasets[x],
batch_size=1,
shuffle=True,
num_workers=2) for x in ['train', 'val']}
dataset_sizes = {x: len(image_datasets[x]) for x in ['train', 'val']}
for image in dataloders["train"]:
f = plt.figure()
print(image[0].shape)
plt.imshow(image[0].squeeze()[0,:,:])
plt.show()
break
任何有關顯示圖像的替代方式的幫助或我所犯的任何錯誤將不勝感激。
如果您想可視化 Dataloader 加載的圖像,我建議使用此腳本:
for batch in train_data_loader:
inputs, targets = batch
for img in inputs:
image = img.cpu().numpy()
# transpose image to fit plt input
image = image.T
# normalise image
data_min = np.min(image, axis=(1,2), keepdims=True)
data_max = np.max(image, axis=(1,2), keepdims=True)
scaled_data = (image - data_min) / (data_max - data_min)
# show image
plt.imshow(scaled_data)
plt.show()
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.