前提・実現したいこと
エラーメッセージの解決方法
発生している問題・エラーメッセージ
RuntimeError: Found 0 files in subfolders of: C:/SRGAN/train/image/ Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif"
該当のソースコード
Python
1import torchvision.transforms as transforms 2from torch.utils.data import DataLoader 3from torchvision.datasets import ImageFolder 4import torch.nn as nn 5import torch 6import os 7import torch.utils.data 8import torch.optim as optim 9import matplotlib.pyplot as plt 10from torch.nn import functional as F 11import numpy as np 12from torchvision.utils import save_image 13%matplotlib inline 14from torchvision.models.vgg import vgg16 15cuda = torch.cuda.is_available() 16if cuda: 17 print('cuda is available!') 18 19if not os.path.exists("save_image"): 20 os.mkdir("save_image") 21if not os.path.exists("dataset"): 22 os.mkdir("dataset") 23if not os.path.exists("asset"): 24 os.mkdir("asset") 25 26class DownSizePairImageFolder(ImageFolder): 27 def __init__(self, root, transform=None, large_size=256, small_size=64, **kwds): 28 super().__init__(root, transform=transform, **kwds) 29 self.large_resizer = transforms.Scale(large_size) 30 self.small_resizer = transforms.Scale(small_size) 31 32 def __getitem__(self, index): 33 path, _ = self.imgs[index] 34 img = self.loader(path) 35 large_img = self.large_resizer(img) 36 small_img = self.small_resizer(img) 37 if self.transform is not None: 38 large_img = self.transform(large_img) 39 small_img = self.transform(small_img) 40 return small_img, large_img 41 42train_data = DownSizePairImageFolder('C:/SRGAN/train/image', transform=transforms.ToTensor()) 43test_data = DownSizePairImageFolder('C:/SRGAN/train/image', transform=transforms.ToTensor()) 44batch_size = 1 45train_loader = DataLoader(train_data, batch_size, shuffle=True, num_workers=4) 46test_loader = DataLoader(test_data, batch_size, shuffle=False, num_workers=4)
試したこと
srganのコードを自分のデータセットで試してみたところエラーが出ました. 画像の拡張子はtifファイルを使っています.
print(os.listdir('C:/SRGAN/train/image'))でフォルダ内に画像が含まれていることは確認しました.
補足情報(ツールのバージョンなど)
anaconda JupiterLab 1.2.6
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2021/05/12 01:36
2021/05/12 03:41
2021/05/13 03:11