我一直在關注這篇文章：https : //github.com/moein-shariatnia/Deep-Learning/tree/main/Image Colorization Tutorial
我有一個問題。我正在嘗試下載已加載到 pytorch 資料加載器中的影像，但由于它們是從較大的資料集中隨機選擇的，因此我無法弄清楚。選擇它們后，如何將已加載到“train_dl”中的確切影像下載到我的 PC（作為檔案夾）？我已經在這些精確的影像上訓練了我的模型，所以我將需要它們用于未來的作業。
uj5u.com熱心網友回復：
資料加載器應該有一個資料集屬性，您可以使用它來查找代碼中的路徑
在代碼中：
class ColorizationDataset(Dataset):
def __init__(self, paths, split='train'):
...
self.split = split
self.size = SIZE
self.paths = paths #<<<<- IMPORTANT LINE HERE
def __getitem__(self, idx):
...
def __len__(self):
return len(self.paths)
def make_dataloaders(batch_size=16, n_workers=4, pin_memory=True, **kwargs): # A handy function to make our dataloaders
dataset = ColorizationDataset(**kwargs)
dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=n_workers,
pin_memory=pin_memory)
return dataloader
因此，要從資料加載器中獲取專案，您可以運行：
train_paths = train_dl.dataset.paths
您可以回圈瀏覽此串列以單獨保存影像（不推薦），或者將其保存為文本，將訓練/驗證/測驗拆分為單獨的檔案。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/311550.html
上一篇：對julia中的匿名函式感到困惑