前提・実現したいこと
SRGANを実装してみたくて試行錯誤しています。
googleドライブにあるファイルから画像データを読み込む際に以下のエラーメッセージが発生しました。
初めての質問でわかりにくいところがあると思いますが、よろしくお願いいたします。
発生している問題・エラーメッセージ
--------------------------------------------------------------------------- FileNotFoundError Traceback (most recent call last) <ipython-input-9-919ca17efb47> in <module>() ----> 1 train_data = DownSizePairImageFolder('/content/drive/MyDrive/20191030_0', transform=transforms.ToTensor()) 2 test_data = DownSizePairImageFolder('/content/drive/MyDrive/20191030_45_200', transform=transforms.ToTensor()) 3 batch_size = 1 4 train_loader = DataLoader(train_data, batch_size, shuffle=True, num_workers=4) 5 test_loader = DataLoader(test_data, batch_size, shuffle=False, num_workers=4) 4 frames /usr/local/lib/python3.7/dist-packages/torchvision/datasets/folder.py in find_classes(directory) 40 classes = sorted(entry.name for entry in os.scandir(directory) if entry.is_dir()) 41 if not classes: ---> 42 raise FileNotFoundError(f"Couldn't find any class folder in {directory}.") 43 44 class_to_idx = {cls_name: i for i, cls_name in enumerate(classes)} FileNotFoundError: Couldn't find any class folder in /content/drive/MyDrive/20191030_0.
該当のソースコード
python
1class DownSizePairImageFolder(ImageFolder): 2 def __init__(self, root, transform=None, large_size=256, small_size=64, **kwds): 3 super().__init__(root, transform=transform, **kwds) 4 self.large_resizer = transforms.Scale(large_size) 5 self.small_resizer = transforms.Scale(small_size) 6 7 def __getitem__(self, index): 8 path, _ = self.imgs[index] 9 img = self.loader(path) 10 large_img = self.large_resizer(img) 11 small_img = self.small_resizer(img) 12 if self.transform is not None: 13 large_img = self.transform(large_img) 14 small_img = self.transform(small_img) 15 return small_img, large_img 16 17ーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーー 18 19train_data = DownSizePairImageFolder('/content/drive/MyDrive/20191030_0', transform=transforms.ToTensor()) 20test_data = DownSizePairImageFolder('/content/drive/MyDrive/20191030_45_200', transform=transforms.ToTensor()) 21batch_size = 1 22train_loader = DataLoader(train_data, batch_size, shuffle=True, num_workers=4) 23test_loader = DataLoader(test_data, batch_size, shuffle=False, num_workers=4)
試したこと
フォルダ内には画像が入っているのは確認しています。
フォルダのパスをコピーして貼り付けているので、パスの間違いはないと思います。
画像サイズが関係しているのでしょうか?
補足情報(FW/ツールのバージョンなど)
https://github.com/pacifinapacific/SRGAN_pytorch/blob/master/SRGAN.ipynb
をもとにgoogle Colaboratoryでコードを書いています。コード自体は同じものを使用しています。
画像データの読み込みのところでエラーが出てしまいました。